// @title       NBA_Go API
// @version     1.0
// @description Stats service, now with public access!
// @schemes     https
// @BasePath    /
//
// @tag.name    PlayerTotals
// @tag.name    PlayerStats
// @tag.name    PlayerShotChart
// To re-enable security docs:
// 1. Uncomment the securityDefinitions block below.
// 2. Uncomment the @Security annotation in each controller.
// 3. Run 'swag init'.
//
// //@securityDefinitions.apikey ApiKeyAuth
// //@in   header
// //@name X-API-Key
package main

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

	"github.com/gofiber/fiber/v2"
	"github.com/gofiber/fiber/v2/middleware/adaptor"
	"github.com/gofiber/fiber/v2/middleware/cors"
	"github.com/gofiber/fiber/v2/middleware/logger"
	"github.com/prometheus/client_golang/prometheus/promhttp"
	fiberswagger "github.com/swaggo/fiber-swagger"
	// "gorm.io/gorm"

	"github.com/nprasad2077/NBA_Go/config"
	"github.com/nprasad2077/NBA_Go/controllers"
	"github.com/nprasad2077/NBA_Go/routes"
	"github.com/nprasad2077/NBA_Go/utils/middleware"
	_ "github.com/nprasad2077/NBA_Go/docs"
)

func main() {
	// ——— One-off import-data mode ———
	if len(os.Args) > 1 && os.Args[1] == "import-data" {
		// Run all migrations + import steps exactly once
		db := config.InitDB(true)

		importPlayerAdvanced(db)
		log.Println("🎉 Player Advanced Import completed successfully")

		importPlayerAdvancedPlayoffs(db)
		log.Println("🎉 Player Advanced Playoffs Import completed successfully")

		importPlayerTotalsScrape(db)
		log.Println("🎉 Player Totals (scraped) Import completed successfully")

		importPlayerTotalsPlayoffsScrape(db)
		log.Println("🎉 Player Playoffs (scraped) Import completed successfully")

		log.Println("🏀 ALL Imports completed successfully ✅ 🙌")
		return
	}

	// ——— Normal API startup: skip AutoMigrate ———
	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
	defer stop()

	app := fiber.New(fiber.Config{
		ReadTimeout:  15 * time.Second,
		WriteTimeout: 15 * time.Second,
		ErrorHandler: func(c *fiber.Ctx, err error) error {
			code := fiber.StatusInternalServerError
			if e, ok := err.(*fiber.Error); ok {
				code = e.Code
			}
			return c.Status(code).JSON(fiber.Map{"error": err.Error()})
		},
	})

	// — CORS Allow ALL origins (development) —
	app.Use(cors.New())

	// middlewares
	app.Use(logger.New())
	app.Use(middleware.MetricsMiddleware())

	// DB connection (no migrations on API startup)
	db := config.InitDB(false)

	/* ---------- PUBLIC ROUTES (no API key) ---------- */
	// Redirect root to swagger docs
	app.Get("/", func(c *fiber.Ctx) error {
		return c.Redirect("/swagger/index.html")
	})

	app.Get("/metrics", adaptor.HTTPHandler(promhttp.Handler()))
	app.Get("/swagger/*", fiberswagger.WrapHandler)
	controllers.RegisterKeyAdminRoutes(app, db)

	/* ---------- PROTECTED ROUTES ---------- */
	// Remove Comment to re-enable api key middleware.
	// app.Use(middleware.APIKeyAuth(db))
	routes.RegisterPlayerAdvancedRoutes(app, db)
	routes.RegisterPlayerTotalRoutes(app, db)
	routes.RegisterPlayerShotChartRoutes(app, db)

	/* ---------- START & SHUTDOWN ---------- */
	go func() {
		if err := app.Listen(":5000"); err != nil && !errors.Is(err, http.ErrServerClosed) {
			log.Fatalf("listen: %v", err)
		}
	}()

	<-ctx.Done()
	stop()
	log.Println("shutting down…")
	_ = app.Shutdown()
	log.Println("bye")
}