// Package router configures the chi router with all routes, middleware,
// and handler wiring for the Control Center API.
package router

import (
	"net/http"
	"time"

	"code.cubecraftcreations.com/CubeCraft-Creations/Control-Center/go-backend/internal/db"
	"code.cubecraftcreations.com/CubeCraft-Creations/Control-Center/go-backend/internal/handler"
	"github.com/go-chi/chi/v5"
	"github.com/go-chi/chi/v5/middleware"
	"github.com/go-chi/cors"
)

// Dependencies carries the handler and database pool into the router.
type Dependencies struct {
	Handler   *handler.Handler
	DB        *db.Pool
	CORSOrigin string
}

// New creates a fully-configured chi router with all API routes mounted.
func New(deps *Dependencies) *chi.Mux {
	r := chi.NewRouter()

	// ── Global middleware ──────────────────────────────────────────────────
	r.Use(middleware.RequestID)
	r.Use(middleware.RealIP)
	r.Use(middleware.Logger)
	r.Use(middleware.Recoverer)
	r.Use(middleware.Timeout(30 * time.Second))

	// ── CORS ───────────────────────────────────────────────────────────────
	corsOrigin := deps.CORSOrigin
	if corsOrigin == "" {
		corsOrigin = "*"
	}
	r.Use(cors.Handler(cors.Options{
		AllowedOrigins:   []string{corsOrigin},
		AllowedMethods:   []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"},
		AllowedHeaders:   []string{"Accept", "Authorization", "Content-Type"},
		ExposedHeaders:   []string{"Link", "X-Total-Count"},
		AllowCredentials: false,
		MaxAge:           300,
	}))

	// ── Health check (with DB connectivity probe) ──────────────────────────
	r.Get("/health", func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Content-Type", "application/json")
		status := "ok"
		if deps.DB != nil {
			if err := deps.DB.Health(r.Context()); err != nil {
				w.WriteHeader(http.StatusServiceUnavailable)
				status = "db_unhealthy"
			}
		}
		w.Write([]byte(`{"status":"` + status + `"}`))
	})

	// ── API v1 routes ──────────────────────────────────────────────────────
	r.Route("/api", func(api chi.Router) {
		// Agents CRUD
		api.Route("/agents", func(agents chi.Router) {
			agents.Get("/", deps.Handler.ListAgents)       // GET /api/agents
			agents.Post("/", deps.Handler.CreateAgent)      // POST /api/agents
			agents.Get("/{id}", deps.Handler.GetAgent)       // GET /api/agents/{id}
			agents.Put("/{id}", deps.Handler.UpdateAgent)    // PUT /api/agents/{id}
			agents.Delete("/{id}", deps.Handler.DeleteAgent) // DELETE /api/agents/{id}
			agents.Get("/{id}/history", deps.Handler.AgentHistory) // GET /api/agents/{id}/history
		})

		// Sessions
		api.Get("/sessions", deps.Handler.ListSessions)

		// Tasks
		api.Get("/tasks", deps.Handler.ListTasks)

		// Projects
		api.Get("/projects", deps.Handler.ListProjects)
	})

	return r
}