package main

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

    "github.com/gin-gonic/gin"
    "github.com/joho/godotenv"
    "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/store"
    "github.com/yourname/cyrene-ai/gateway/internal/ws"
)

func main() {
    // 自动加载 .env 文件（来自 backend/.env）
    if err := godotenv.Load("../.env"); err != nil {
        log.Println("ℹ 未找到 .env 文件，将使用环境变量或默认值")
    }

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

    // 初始化数据库持久化存储 (降级：连接失败不崩溃)
    var sessionStore *store.SessionStore
    databaseURL := cfg.DatabaseURL()
    if s, err := store.NewSessionStore(databaseURL); err != nil {
        log.Printf("⚠ 会话持久化存储初始化失败 (数据库不可用): %v", err)
        log.Println("⚠ Gateway 将以仅内存模式运行 — 会话数据在重启后丢失")
    } else {
        sessionStore = s
        log.Println("✅ 会话持久化存储已启用 (PostgreSQL)")
    }

    // 初始化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()
    hub.SetStore(sessionStore)
    hub.SetIdleTimeout(cfg.SessionIdleTimeoutMin)
    go hub.Run()

    // 启动闲置会话清理 (标记超时会话为 idle，不删除)
    hub.StartIdleCleanup()

    // 启动 IoT 设备状态广播（每10秒向所有WebSocket客户端推送设备状态）
    hub.StartIoTBroadcast(cfg.IoTDebugServiceURL)

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

    // 启动服务
    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("正在关闭服务...")

    hub.StopIoTBroadcast()

    // 关闭数据库连接
    if sessionStore != nil {
        if err := sessionStore.Close(); err != nil {
            log.Printf("⚠ 关闭数据库连接失败: %v", err)
        }
    }

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