package main

import (
	"context"
	"fmt"
	"log"
	"net/http"
	"os"
	"os/signal"
	"syscall"
	"time"

	"github.com/fujin/anthropic-proxy/internal/auth"
	"github.com/fujin/anthropic-proxy/internal/config"
	"github.com/fujin/anthropic-proxy/internal/proxy"
	"github.com/fujin/anthropic-proxy/internal/server"
)

func run() error {
	log.SetFlags(log.LstdFlags)

	cfg, err := config.Load("config.yaml")
	if err != nil {
		return fmt.Errorf("load config: %w", err)
	}

	creds, err := config.LoadCredentials(cfg)
	if err != nil {
		return fmt.Errorf("load credentials: %w", err)
	}

	if len(creds) == 0 {
		return fmt.Errorf("no credentials found")
	}

	log.Printf("loaded %d credentials", len(creds))

	pool := auth.NewPool(creds)

	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

	pool.RefreshExpiring(context.Background())
	auth.StartBackgroundRefresh(ctx, pool)

	var profile *proxy.SniffedProfile
	if cfg.ClaudeBinary != "" {
		log.Printf("sniffing claude-code at %s...", cfg.ClaudeBinary)
		profile, err = proxy.SniffClaudeCode(cfg.ClaudeBinary)
		if err != nil {
			log.Printf("warning: sniff failed, using defaults: %v", err)
		}
	}

	log.Printf("starting server on port %d", cfg.Port)
	srv := server.New(cfg, pool, profile)

	quit := make(chan os.Signal, 1)
	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)

	go func() {
		<-quit
		log.Printf("shutting down...")
		cancel()

		shutdownCtx, shutdownCancel := context.WithTimeout(context.Background(), 10*time.Second)
		defer shutdownCancel()

		if err := srv.Shutdown(shutdownCtx); err != nil {
			log.Printf("shutdown error: %v", err)
		}
	}()

	if err := srv.Start(); err != nil && err != http.ErrServerClosed {
		return err
	}

	return nil
}

func main() {
	if err := run(); err != nil {
		log.Printf("error: %v", err)
		os.Exit(1)
	}
}