package main

import (
	"context"
	"log/slog"
	"net/http"
	"os"

	"github.com/mathiasbq/supervisor/internal/config"
	iexec "github.com/mathiasbq/supervisor/internal/exec"
	"github.com/mathiasbq/supervisor/internal/mcp"
	"github.com/mathiasbq/supervisor/internal/registry"
	"github.com/mathiasbq/supervisor/internal/skills/brain"
	"github.com/mathiasbq/supervisor/internal/skills/org"
	"github.com/mathiasbq/supervisor/internal/skills/retrospective"
	skilldebug "github.com/mathiasbq/supervisor/internal/skills/debug"
	"github.com/mathiasbq/supervisor/internal/skills/review"
	"github.com/mathiasbq/supervisor/internal/skills/sessionlog"
	"github.com/mathiasbq/supervisor/internal/skills/tdd"
	"github.com/mathiasbq/supervisor/internal/tier"
)

func main() {
	logger := slog.New(slog.NewJSONHandler(os.Stdout, nil))

	cfg, err := config.Load()
	if err != nil {
		logger.Error("load config", "err", err)
		os.Exit(1)
	}

	models, err := config.LoadModels(cfg.ModelsFile)
	if err != nil {
		logger.Error("load models", "err", err)
		os.Exit(1)
	}

	systemPrompt, err := os.ReadFile(cfg.ConfigDir + "/CLAUDE.md")
	if err != nil {
		logger.Error("read supervisor CLAUDE.md", "path", cfg.ConfigDir+"/CLAUDE.md", "err", err)
		os.Exit(1)
	}

	tddPrompt, err := os.ReadFile(cfg.ConfigDir + "/tdd.md")
	if err != nil {
		logger.Error("read tdd.md", "path", cfg.ConfigDir+"/tdd.md", "err", err)
		os.Exit(1)
	}

	retroPrompt, err := os.ReadFile(cfg.ConfigDir + "/retrospective.md")
	if err != nil {
		logger.Error("read retrospective.md", "path", cfg.ConfigDir+"/retrospective.md", "err", err)
		os.Exit(1)
	}

	reviewPrompt, err := os.ReadFile(cfg.ConfigDir + "/review.md")
	if err != nil {
		logger.Error("read review.md", "path", cfg.ConfigDir+"/review.md", "err", err)
		os.Exit(1)
	}

	debugPrompt, err := os.ReadFile(cfg.ConfigDir + "/debug.md")
	if err != nil {
		logger.Error("read debug.md", "path", cfg.ConfigDir+"/debug.md", "err", err)
		os.Exit(1)
	}

	executor := iexec.New(iexec.Config{
		SystemPrompt:   string(systemPrompt),
		LiteLLMBaseURL: cfg.LiteLLMBaseURL,
		LiteLLMAPIKey:  cfg.LiteLLMAPIKey,
	})

	tierFn := func(ctx context.Context) tier.Info {
		return tier.Detect(ctx, "https://api.anthropic.com", cfg.LiteLLMBaseURL)
	}

	reg := registry.New()
	reg.Register(tdd.New(tdd.Config{
		SystemPrompt: string(systemPrompt),
		SkillPrompt:  string(tddPrompt),
		DefaultModel: models.Resolve("tdd", ""),
		ExecutorFn:   executor.Run,
		SessionsDir:  cfg.SessionsDir,
	}))
	reg.Register(brain.New(brain.Config{
		IngestBaseURL: cfg.IngestBaseURL,
	}))
	reg.Register(org.New(org.Config{
		TierFn: tierFn,
	}))
	reg.Register(sessionlog.New(sessionlog.Config{
		SessionsDir: cfg.SessionsDir,
	}))
	reg.Register(retrospective.New(retrospective.Config{
		SkillPrompt:  string(retroPrompt),
		DefaultModel: models.Resolve("retrospective", ""),
		SessionsDir:  cfg.SessionsDir,
		ExecutorFn:   executor.Run,
	}))
	reg.Register(review.New(review.Config{
		SkillPrompt:  string(reviewPrompt),
		DefaultModel: models.Resolve("review", ""),
		ExecutorFn:   executor.Run,
		SessionsDir:  cfg.SessionsDir,
	}))
	reg.Register(skilldebug.New(skilldebug.Config{
		SkillPrompt:  string(debugPrompt),
		DefaultModel: models.Resolve("debug", ""),
		ExecutorFn:   executor.Run,
		SessionsDir:  cfg.SessionsDir,
	}))

	srv := mcp.NewServer(reg)
	mux := http.NewServeMux()
	mux.Handle("/mcp", srv)

	addr := ":" + cfg.Port
	logger.Info("supervisor starting", "addr", addr)
	if err := http.ListenAndServe(addr, mux); err != nil {
		logger.Error("server stopped", "err", err)
		os.Exit(1)
	}
}