freebox-exporter/main.go

package main

import (
	"errors"
	"flag"
	"fmt"
	"io/ioutil"
	"net/http"
	"os"

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

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

func usage(err error) {
	if err != nil {
		fmt.Fprintln(os.Stderr, "Error:", err)
	}
	fmt.Fprintln(os.Stderr, "Usage:", os.Args[0], "[options] <configfile>")
	fmt.Fprintln(os.Stderr)
	fmt.Fprintln(os.Stderr, "Options:")
	flag.PrintDefaults()
	os.Exit(-1)
}

func main() {
	debugPtr := flag.Bool("debug", false, "enable the debug mode")
	listHostsPtr := flag.Bool("listHosts", false, "list the hosts connected to wifi and ethernet")
	listenPtr := flag.String("listen", ":9091", "listen to address")
	flag.Parse()

	args := flag.Args()
	if len(args) < 1 {
		usage(errors.New("configfile not defined"))
	} else if len(args) > 1 {
		usage(errors.New("too many arguments"))
	}
	if *debugPtr {
		log.Init(os.Stdout, os.Stdout, os.Stdout, os.Stderr)
	} else {
		log.Init(ioutil.Discard, os.Stdout, os.Stdout, os.Stderr)
	}
	collector := NewCollector(args[0], *listHostsPtr, *debugPtr)
	defer collector.Close()

	prometheus.MustRegister(collector)

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