package main

import (
	"bytes"
	"context"
	"encoding/json"
	"flag"
	"io/fs"
	"log/slog"
	"net/http"
	"os"
	"os/signal"
	"time"

	"github.com/alecthomas/chroma/v2/formatters/html"
	"github.com/alecthomas/chroma/v2/lexers"
	"github.com/alecthomas/chroma/v2/styles"

	"git.gabrielgio.me/cerrado/pkg/config"
	"git.gabrielgio.me/cerrado/pkg/worker"
	"git.gabrielgio.me/cerrado/static"
	"git.gabrielgio.me/cerrado/templates"
)

func main() {

	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, os.Kill)
	defer stop()
	if err := run(ctx); err != nil {
		slog.Error("Error", "error", err)
		os.Exit(1)
	}
}

func run(ctx context.Context) error {
	var (
		configPath = flag.String("config", "config.example.scfg", "File path for the configuration file")
	)

	flag.Parse()

	mux := http.NewServeMux()

	staticFs, err := fs.Sub(static.Static, ".")
	if err != nil {
		return err
	}

	mux.Handle("/static/", http.StripPrefix("/static/", http.FileServer(http.FS(staticFs))))
	mux.HandleFunc("/", func(w http.ResponseWriter, _ *http.Request) {
		slog.Info("Handling index")

		f, err := os.Open(*configPath)
		if err != nil {
			slog.Error("Error openning config file json", "error", err, "path", *configPath)
			return
		}

		c, err := config.Parse(f)
		if err != nil {
			slog.Error("Error parsing config", "error", err, "path", *configPath)
			return
		}

		b, err := json.MarshalIndent(c, "", "	")
		if err != nil {
			slog.Error("Error parsing json", "error", err)
			return
		}

		lexer := lexers.Get("json")
		style := styles.Get("monokailight")
		formatter := html.New(
			html.WithLineNumbers(true),
		)
		iterator, err := lexer.Tokenise(nil, string(b))
		if err != nil {
			slog.Error("Error tokenise", "error", err)
			return
		}

		var code bytes.Buffer
		err = formatter.Format(&code, style, iterator)
		if err != nil {
			slog.Error("Error format", "error", err)
			return
		}

		hello := &templates.HelloPage{
			Body: code.String(),
		}

		templates.WritePageTemplate(w, hello)
	})
	serverTask := worker.NewServerTask(&http.Server{Handler: mux, Addr: "0.0.0.0:8080"})

	pool := worker.NewTaskPool()
	pool.AddTask("http-server", 5*time.Second, serverTask)

	return pool.Start(ctx)
}