netatmo-exporter/main.go

package main

import (
	"errors"
	"log"
	"net/http"
	"os"

	netatmo "github.com/exzz/netatmo-api-go"
	"github.com/prometheus/client_golang/prometheus"
	"github.com/spf13/pflag"
)

type config struct {
	Addr    string
	Netatmo netatmo.Config
}

func parseConfig() (config, error) {
	cfg := config{}

	// Set default port to bind to
	addr := ":9210"
	if envAddr := os.Getenv("NETATMO_EXPORTER_ADDR"); envAddr != "" {
		addr = envAddr
	}

	pflag.StringVarP(&cfg.Addr, "addr", "a", addr, "Address to listen on.")
	pflag.StringVarP(&cfg.Netatmo.ClientID, "client-id", "i", os.Getenv("NETATMO_CLIENT_ID"), "Client ID for NetAtmo app.")
	pflag.StringVarP(&cfg.Netatmo.ClientSecret, "client-secret", "s", os.Getenv("NETATMO_CLIENT_SECRET"), "Client secret for NetAtmo app.")
	pflag.StringVarP(&cfg.Netatmo.Username, "username", "u", os.Getenv("NETATMO_CLIENT_USERNAME"), "Username of NetAtmo account.")
	pflag.StringVarP(&cfg.Netatmo.Password, "password", "p", os.Getenv("NETATMO_CLIENT_PASSWORD"), "Password of NetAtmo account.")
	pflag.Parse()

	if len(cfg.Addr) == 0 {
		return cfg, errors.New("no listen address")
	}

	if len(cfg.Netatmo.ClientID) == 0 {
		return cfg, errors.New("need a NetAtmo client ID")
	}

	if len(cfg.Netatmo.ClientSecret) == 0 {
		return cfg, errors.New("need a NetAtmo client secret")
	}

	if len(cfg.Netatmo.Username) == 0 {
		return cfg, errors.New("username can not be blank")
	}

	if len(cfg.Netatmo.Password) == 0 {
		return cfg, errors.New("password can not be blank")
	}

	return cfg, nil
}

func main() {
	cfg, err := parseConfig()
	if err != nil {
		log.Fatalf("Error in configuration: %s", err)
	}

	log.Printf("Login as %s", cfg.Netatmo.Username)
	client, err := netatmo.NewClient(cfg.Netatmo)
	if err != nil {
		log.Fatalf("Error creating client: %s", err)
	}

	metrics := &netatmoCollector{
		client: client,
	}
	prometheus.MustRegister(metrics)

	http.Handle("/metrics", prometheus.UninstrumentedHandler())
	http.Handle("/", http.RedirectHandler("/metrics", http.StatusFound))

	log.Printf("Listen on %s...", cfg.Addr)
	log.Fatal(http.ListenAndServe(cfg.Addr, nil))
}