package main

import (
	"fmt"
	"log"
	"net/http"
	"os"
	"strconv"
	"time"

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

type serverSettings struct {
	port                                  int
	plexAddr                              string
	plexToken                             string
	plexTimeout                           int
	metricsPrefix                         string
	metricsMediaCollectingIntervalSeconds int
}

func envGetter(key, fallback string) string {
	if value, ok := os.LookupEnv(key); ok {
		return value
	}
	return fallback
}

func setupServerSettings() serverSettings {
	port, err := strconv.Atoi(envGetter("PORT", "9545"))
	if err != nil {
		log.Fatal("Unable to parse port string to int, failing....")
	}
	plexToken, value := os.LookupEnv("PLEX_TOKEN")
	if !value {
		log.Fatal("No plex token provided, failing....")
	}
	plexTimeout, err := strconv.Atoi(envGetter("PLEX_TIMEOUT", "10"))
	if err != nil {
		log.Fatal("Unable to parse plextimeout string to int, failing....")
	}
	metricsMediaCollectingIntervalSeconds, err := strconv.Atoi(envGetter("METRICS_MEDIA_COLLECTING_INTERVAL_SECONDS", "300"))
	if err != nil {
		log.Fatal("Unable to parse METRICS_MEDIA_COLLECTING_INTERVAL_SECONDS string to int, failing....")
	}

	return serverSettings{
		port:                                  port,
		plexAddr:                              envGetter("PLEX_ADDR", "http://localhost:32400"),
		plexToken:                             plexToken,
		plexTimeout:                           plexTimeout,
		metricsPrefix:                         envGetter("METRICS_PREFIX", "PLEX"),
		metricsMediaCollectingIntervalSeconds: metricsMediaCollectingIntervalSeconds,
	}
}

func main() {

	os.Setenv("PORT", "9545")
	os.Setenv("PLEX_ADDR", "https://plex.derajnet.duckdns.org:32400")
	os.Setenv("PLEX_TOKEN", "5ezJu5cjnhoAbPJRKngs")
	os.Setenv("PLEX_TIMEOUT", "10")
	os.Setenv("METRICS_PREFIX", "PLEX")
	os.Setenv("METRICS_MEDIA_COLLECTING_INTERVAL_SECONDS", "300")

	settings := setupServerSettings()
	// Setup plex client
	pc := setupClient(&settings)

	prometheus.MustRegister(plexActiveSessions)
	prometheus.MustRegister(plexNumMovies)
	prometheus.MustRegister(plexNumTVShows)
	prometheus.MustRegister(plexTranscodeSessions)
	prometheus.MustRegister(plexAllSessions)

	go func() {
		for {
			time.Sleep(time.Second * 60)
			fmt.Println("Collecting info...")
			stats := pc.gatherAllStats()
			plexActiveSessions.Set(stats.currentSessions)
			plexNumMovies.Set(stats.numMovies)
			plexNumTVShows.Set(stats.numTV)
			plexTranscodeSessions.Set(stats.numTranscodes)
			plexAllSessions.Set(stats.numAllSessions)
		}
	}()

	// m := NewMetrics(reg)
	http.Handle("/metrics", promhttp.Handler())
	http.ListenAndServe(fmt.Sprintf(":%d", settings.port), nil)

}