package main

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

	"github.com/gin-gonic/gin"
	"go.uber.org/zap"

	"github.com/genex/trading-service/internal/interface/http/handler"
	"github.com/genex/trading-service/internal/interface/http/middleware"
	"github.com/genex/trading-service/internal/matching"
)

func main() {
	logger, _ := zap.NewProduction()
	defer logger.Sync()

	port := os.Getenv("PORT")
	if port == "" {
		port = "3003"
	}

	// Initialize matching engine
	engine := matching.NewEngine()

	// Setup Gin router
	r := gin.New()
	r.Use(gin.Recovery())

	// Health endpoints
	r.GET("/health", func(c *gin.Context) {
		c.JSON(200, gin.H{"status": "ok", "service": "trading-service", "timestamp": time.Now().UTC().Format(time.RFC3339)})
	})
	r.GET("/health/ready", func(c *gin.Context) { c.JSON(200, gin.H{"status": "ready"}) })
	r.GET("/health/live", func(c *gin.Context) { c.JSON(200, gin.H{"status": "alive"}) })

	// API routes
	api := r.Group("/api/v1")

	tradeHandler := handler.NewTradeHandler(engine)

	trades := api.Group("/trades")
	trades.Use(middleware.JWTAuth())
	{
		trades.POST("/orders", tradeHandler.PlaceOrder)
		trades.DELETE("/orders/:id", tradeHandler.CancelOrder)
	}

	// Public orderbook
	api.GET("/trades/orderbook/:couponId", tradeHandler.GetOrderBook)

	// Admin routes (require JWT + admin role)
	adminTradeHandler := handler.NewAdminTradeHandler(engine)
	adminMMHandler := handler.NewAdminMMHandler(engine)

	admin := api.Group("/admin")
	admin.Use(middleware.JWTAuth(), middleware.RequireAdmin())
	{
		// Trade administration
		adminTrades := admin.Group("/trades")
		adminTrades.GET("/stats", adminTradeHandler.GetTradingStats)
		adminTrades.GET("/orders", adminTradeHandler.ListOrders)
		adminTrades.GET("/volume-trend", adminTradeHandler.GetVolumeTrend)
		adminTrades.POST("/orders/:id/investigate", adminTradeHandler.InvestigateOrder)

		// Market maker administration
		mm := admin.Group("/mm")
		mm.GET("/list", adminMMHandler.ListMarketMakers)
		mm.GET("/:id/details", adminMMHandler.GetMarketMakerDetails)
		mm.POST("/:id/suspend", adminMMHandler.SuspendMarketMaker)
		mm.POST("/:id/resume", adminMMHandler.ResumeMarketMaker)
		mm.GET("/liquidity-pools", adminMMHandler.GetLiquidityPools)
		mm.GET("/order-book-depth", adminMMHandler.GetOrderBookDepth)
		mm.GET("/health-indicators", adminMMHandler.GetHealthIndicators)
	}

	server := &http.Server{
		Addr:         ":" + port,
		Handler:      r,
		ReadTimeout:  15 * time.Second,
		WriteTimeout: 15 * time.Second,
		IdleTimeout:  60 * time.Second,
	}

	go func() {
		logger.Info("Trading Service starting", zap.String("port", port))
		if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
			logger.Fatal("Server failed", zap.Error(err))
		}
	}()

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

	logger.Info("Shutting down gracefully...")
	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
	defer cancel()

	if err := server.Shutdown(ctx); err != nil {
		logger.Fatal("Server forced to shutdown", zap.Error(err))
	}
	logger.Info("Trading Service stopped")
}