freebox-exporter/main.go

package main

import (
	"flag"
	"fmt"
	"net/http"
	"os"

	"github.com/prometheus/client_golang/prometheus"
	"github.com/prometheus/client_golang/prometheus/promhttp"

	"github.com/trazfr/freebox-exporter/fbx"
	"github.com/trazfr/freebox-exporter/log"
)

func usage() {
	fmt.Fprintf(flag.CommandLine.Output(),
		"Usage: %s [options] <api_token_file>\n"+
			"\n"+
			"api_token_file: file to store the token for the API\n"+
			"\n"+
			"options:\n",
		os.Args[0])
	flag.PrintDefaults()
}

func main() {
	flag.Usage = usage
	debugPtr := flag.Bool("debug", false, "enable the debug mode")
	hostDetailsPtr := flag.Bool("hostDetails", false, "get details about the hosts connected to wifi and ethernet. This increases the number of metrics")
	httpDiscoveryPtr := flag.Bool("httpDiscovery", false, "use http://mafreebox.freebox.fr/api_version to discover the Freebox at the first run (by default: use mDNS)")
	listenPtr := flag.String("listen", ":9091", "listen to address")
	flag.Parse()

	args := flag.Args()
	if len(args) < 1 {
		fmt.Fprintf(flag.CommandLine.Output(), "ERROR: api_token_file not defined\n")
		usage()
		os.Exit(1)
	} else if len(args) > 1 {
		fmt.Fprintf(flag.CommandLine.Output(), "ERROR: too many arguments\n")
		usage()
		os.Exit(1)
	}
	if *debugPtr {
		log.InitDebug()
	} else {
		log.Init()
	}
	discovery := fbx.FreeboxDiscoveryMDNS
	if *httpDiscoveryPtr {
		discovery = fbx.FreeboxDiscoveryHTTP
	}

	collector := NewCollector(args[0], discovery, *hostDetailsPtr, *debugPtr)
	defer collector.Close()

	prometheus.MustRegister(collector)

	http.Handle("/metrics", promhttp.Handler())
	log.Info.Println("Listen to", *listenPtr)
	log.Error.Println(http.ListenAndServe(*listenPtr, nil))
}