package main

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

    "github.com/gin-gonic/gin"
    "github.com/yourname/cyrene-ai/gateway/internal/config"
    "github.com/yourname/cyrene-ai/gateway/internal/middleware"
    "github.com/yourname/cyrene-ai/gateway/internal/router"
    "github.com/yourname/cyrene-ai/gateway/internal/ws"
)

func main() {
    // 加载配置
    cfg := config.Load()

    // 初始化Gin
    if cfg.Env == "production" {
        gin.SetMode(gin.ReleaseMode)
    }
    r := gin.New()

    // 中间件
    r.Use(middleware.CORS())
    r.Use(middleware.RequestLogging())
    r.Use(gin.Recovery())

    // 初始化WebSocket Hub
    hub := ws.NewHub()
    go hub.Run()

    // 注册路由
    router.Setup(r, hub, cfg)

    // 启动服务
    srv := &http.Server{
        Addr:    ":" + cfg.Port,
        Handler: r,
    }

    go func() {
        log.Printf("🚀 Gateway 启动在端口 %s", cfg.Port)
        if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
            log.Fatalf("服务启动失败: %v", err)
        }
    }()

    // 优雅关闭
    quit := make(chan os.Signal, 1)
    signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
    <-quit
    log.Println("正在关闭服务...")

    ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
    defer cancel()
    srv.Shutdown(ctx)
    log.Println("服务已关闭")
}