package main

import (
	"log"
	"os"
	"os/signal"
	"syscall"

	"jd-book-uploader/config"
	"jd-book-uploader/handlers"
	"jd-book-uploader/middleware"
	"jd-book-uploader/services"

	"github.com/gofiber/fiber/v2"
)

func main() {
	// Load configuration
	cfg, err := config.LoadConfig()
	if err != nil {
		log.Fatalf("Failed to load configuration: %v", err)
	}

	// Initialize database connection
	_, err = services.NewDBPool(cfg)
	if err != nil {
		log.Fatalf("Failed to connect to database: %v", err)
	}
	// Note: CloseDB is called explicitly in graceful shutdown, not in defer
	log.Println("Database connected successfully")

	// Initialize Firebase
	_, err = services.InitFirebase(cfg)
	if err != nil {
		log.Fatalf("Failed to initialize Firebase: %v", err)
	}
	log.Println("Firebase initialized successfully")

	// Create Fiber app
	app := fiber.New(fiber.Config{
		AppName: "JD Book Uploader API",
	})

	// Register middleware
	app.Use(middleware.RecoverHandler()) // Recover from panics
	app.Use(middleware.ErrorHandler)     // Global error handler
	app.Use(middleware.SetupCORS(cfg))   // CORS
	app.Use(middleware.SetupLogger())    // Request logging

	// Register routes
	api := app.Group("/api")
	api.Get("/health", handlers.HealthCheck)
	api.Post("/books", handlers.UploadBook)
	api.Post("/stationery", handlers.UploadStationery)

	// Start server
	port := cfg.Port
	if port == "" {
		port = "8080"
	}

	// Graceful shutdown
	c := make(chan os.Signal, 1)
	signal.Notify(c, os.Interrupt, syscall.SIGTERM)

	go func() {
		log.Printf("Server starting on port %s", port)
		if err := app.Listen(":" + port); err != nil {
			log.Fatalf("Server failed to start: %v", err)
		}
	}()

	<-c
	log.Println("Gracefully shutting down...")

	// Close database connection
	if err := services.CloseDB(); err != nil {
		log.Printf("Error closing database: %v", err)
	}

	// Shutdown Fiber app
	if err := app.Shutdown(); err != nil {
		log.Fatalf("Server shutdown failed: %v", err)
	}

	log.Println("Server stopped")
}