package main

import (
	"log"
	"net/http"
	"time"

	"github.com/prometheus/client_golang/prometheus/promhttp"
	"github.com/user/nanometrics/server/config"
	"github.com/user/nanometrics/server/db"
	"github.com/user/nanometrics/server/handlers"
	"github.com/user/nanometrics/server/models"
	prom "github.com/user/nanometrics/server/prometheus"
	"github.com/user/nanometrics/server/transport"
	ws "github.com/user/nanometrics/server/websocket"
)

func main() {
	cfg := config.Load()

	database, err := db.Open(cfg.DBPath)
	if err != nil {
		log.Fatalf("DB: %v", err)
	}

	hub := ws.NewHub()

	onMetrics := func(m *models.AgentMetrics) {
		if err := database.UpsertAgent(m); err != nil {
			log.Printf("[ingest] upsert agent: %v", err)
		}
		if err := database.InsertMetrics(m); err != nil {
			log.Printf("[ingest] insert metrics: %v", err)
		}
		prom.Update(m)
		hub.Broadcast(models.WSMessage{
			Type:    "metrics_update",
			AgentID: m.Hostname,
			Data:    m,
		})
	}

	if err := transport.StartUDP(cfg.UDPAddr, onMetrics); err != nil {
		log.Fatalf("UDP: %v", err)
	}

	var mqttClient *transport.MQTTClient
	if mc, err := transport.StartMQTT(cfg.MQTTBroker, cfg.MQTTTopicBase, onMetrics); err != nil {
		log.Printf("[mqtt] non disponible: %v", err)
	} else {
		mqttClient = mc
	}

	pushConfig := func(agentID string, agentCfg *models.AgentConfig) {
		if mqttClient != nil {
			if err := mqttClient.PushConfig(agentID, agentCfg); err != nil {
				log.Printf("[mqtt] push config to %s: %v", agentID, err)
			}
		}
	}

	go func() {
		ticker := time.NewTicker(time.Minute)
		defer ticker.Stop()
		for range ticker.C {
			srvCfg, _ := database.GetServerConfig()
			_ = database.PruneOldMetrics(srvCfg.RetentionDays)
			_ = database.MarkOffline(30)
		}
	}()

	mux := http.NewServeMux()
	mux.Handle("/metrics", promhttp.Handler())
	mux.Handle("/ws", ws.Handler(hub))
	mux.HandleFunc("/api/agents", handlers.AgentsHandler(database))
	mux.HandleFunc("/api/agents/", func(w http.ResponseWriter, r *http.Request) {
		switch {
		case endsWith(r.URL.Path, "/history"):
			handlers.MetricsHistoryHandler(database)(w, r)
		case endsWith(r.URL.Path, "/config"):
			handlers.AgentConfigHandler(database, pushConfig)(w, r)
		case endsWith(r.URL.Path, "/icon") && r.Method == http.MethodPost:
			handlers.IconUploadHandler(database)(w, r)
		case endsWith(r.URL.Path, "/icon") && r.Method == http.MethodGet:
			handlers.IconGetHandler(database)(w, r)
		default:
			http.NotFound(w, r)
		}
	})
	mux.HandleFunc("/api/config", handlers.ServerConfigHandler(database))

	log.Printf("[http] écoute sur %s", cfg.HTTPAddr)
	log.Fatal(http.ListenAndServe(cfg.HTTPAddr, mux))
}

func endsWith(path, suffix string) bool {
	return len(path) >= len(suffix) && path[len(path)-len(suffix):] == suffix
}