main.go

package main

import (
	"bufio"
	"context"
	"log"
	"net/http"
	"os"
	"os/signal"
	"syscall"
	"tplink_router_exporter/config"
	"tplink_router_exporter/internal/collector"
	"tplink_router_exporter/pkg/tplink"

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

func init() {
	if config.ROUTER_ENDPOINT == "" || config.PASSWORD == "" {
		log.Fatal("ROUTER_ENDPOINT and PASSWORD environment variables are required")
	}
}

func main() {
	token, err := tplink.Login()
	if err != nil {
		log.Fatalf("login failed: %v", err)
	}
	tplink.GlobalToken = token

	networkCollector := collector.NewNetworkCollector()
	hostsCollector := collector.NewHostsCollector()

	prometheus.MustRegister(networkCollector)
	prometheus.MustRegister(hostsCollector)

	go networkCollector.StartBackgroundCollection()
	go hostsCollector.StartBackgroundCollection()

	mux := http.NewServeMux()
	mux.Handle("/metrics", promhttp.Handler())

	server := &http.Server{
		Addr:    ":9100",
		Handler: mux,
	}

	go func() {
		log.Println("Exporter started: http://localhost:9100/metrics")
		if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
			log.Fatalf("http server: %v", err)
		}
	}()

	sigCh := make(chan os.Signal, 1)
	signal.Notify(sigCh, os.Interrupt, syscall.SIGTERM)

	enterCh := make(chan struct{})
	go func() {
		_, _ = bufio.NewReader(os.Stdin).ReadBytes('\n')
		close(enterCh)
	}()

	log.Println("Press Enter or Ctrl+C to exit.")

	select {
	case <-sigCh:
		log.Println("Received signal, shutting down...")
	case <-enterCh:
		log.Println("Received Enter, shutting down...")
	}

	if err := server.Shutdown(context.Background()); err != nil {
		log.Printf("shutdown http server: %v", err)
	}

	log.Println("Bye.")
}