Cyrene/backend/gateway/cmd/main.go

package main

import (
    "context"
    "git.yeij.top/AskaEth/Cyrene/pkg/logger"
    "net/http"
    "os"
    "os/signal"
    "syscall"
    "time"

    "github.com/gin-gonic/gin"
    "github.com/joho/godotenv"
    "golang.org/x/crypto/bcrypt"

    "git.yeij.top/AskaEth/Cyrene/gateway/internal/config"
    "git.yeij.top/AskaEth/Cyrene/gateway/internal/engine"
    "git.yeij.top/AskaEth/Cyrene/gateway/internal/handler"
    "git.yeij.top/AskaEth/Cyrene/gateway/internal/middleware"
    "git.yeij.top/AskaEth/Cyrene/gateway/internal/router"
    "git.yeij.top/AskaEth/Cyrene/gateway/internal/store"
    "git.yeij.top/AskaEth/Cyrene/gateway/internal/ws"
)

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

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

    // 确保上传目录存在
    if err := os.MkdirAll("./uploads", 0755); err != nil {
    	logger.Printf("⚠ 创建上传目录失败: %v", err)
    }
   
    // 初始化数据库持久化存储 (降级：连接失败不崩溃)
    var sessionStore *store.SessionStore
    var reminderStore *store.ReminderStore
    var automationStore *store.AutomationStore
    var fileStore *store.FileStore
    var knowledgeStore *store.KnowledgeStore
    var ruleEngine *engine.RuleEngine
    databaseURL := cfg.DatabaseURL()
    if s, err := store.NewSessionStore(databaseURL); err != nil {
    	logger.Printf("⚠ 会话持久化存储初始化失败 (数据库不可用): %v", err)
    	logger.Println("⚠ Gateway 将以仅内存模式运行 — 会话数据在重启后丢失")
    } else {
    	sessionStore = s
    	logger.Println("✅ 会话持久化存储已启用 (PostgreSQL)")
    
    	// 初始化 users 表
    	if err := store.CreateUsersTable(s.DB()); err != nil {
    		logger.Printf("⚠ 创建 users 表失败: %v", err)
    	} else {
    		logger.Println("✅ Users 表已就绪")
    	}

    	// 种子数据：如果没有 admin 用户，创建默认 admin
    	if existingAdmin, err := store.GetUserByUsername(s.DB(), cfg.AdminUsername); err != nil {
    		logger.Printf("⚠ 查询管理员用户失败: %v", err)
    	} else if existingAdmin == nil {
    		logger.Printf("🔧 未找到管理员用户，创建默认 %s (username: %s)...", cfg.AdminUsername, cfg.AdminUsername)
		defaultAdminPassword := cfg.AdminPassword
    		passwordHash, err := bcrypt.GenerateFromPassword([]byte(defaultAdminPassword), bcrypt.DefaultCost)
    		if err != nil {
    			logger.Printf("⚠ 管理员密码哈希生成失败: %v", err)
    		} else {
    			if _, err := store.CreateUser(s.DB(), cfg.AdminUsername, "管理员", string(passwordHash), true); err != nil {
    				logger.Printf("⚠ 创建默认管理员失败: %v", err)
    			} else {
    				logger.Printf("✅ 默认管理员用户已创建 (username: %s)", cfg.AdminUsername)
    			}
    		}
    	} else {
    		logger.Println("✅ 管理员用户已存在")
    	}

    	// 清理旧的管理员用户 (is_admin=true 但 username 与当前 ADMIN_USERNAME 不同)
    	// 当 .env 中 ADMIN_USERNAME 变更时，旧的 admin 用户会成为孤立的会话持有者
    	if allUsers, err := store.ListUsers(s.DB()); err != nil {
    		logger.Printf("⚠ 查询所有用户失败: %v", err)
    	} else {
    		for _, u := range allUsers {
    			if u.IsAdmin && u.Username != cfg.AdminUsername {
    				logger.Printf("🗑 清理旧管理员用户: %s (id=%d)", u.Username, u.ID)
    				if err := store.DeleteUser(s.DB(), u.ID); err != nil {
    					logger.Printf("⚠ 删除旧管理员用户失败: %s, err=%v", u.Username, err)
    				}
    			}
    		}
    	}

    	// 初始化提醒存储（复用同一数据库连接）
    	if rs, err := store.NewReminderStore(s.DB()); err != nil {
    		logger.Printf("⚠ 提醒存储初始化失败: %v", err)
    	} else {
    		reminderStore = rs
    		logger.Println("✅ 提醒持久化存储已启用 (PostgreSQL)")
    	}
    

    	// 初始化自动化存储（复用同一数据库连接）
    	if as, err := store.NewAutomationStore(s.DB()); err != nil {
    		logger.Printf("⚠ 自动化存储初始化失败: %v", err)
    	} else {
    		automationStore = as
    		logger.Println("✅ 自动化持久化存储已启用 (PostgreSQL)")
    	}
    
    	// 初始化文件存储（复用同一数据库连接）
    	if fs, err := store.NewFileStore(s.DB()); err != nil {
    		logger.Printf("⚠ 文件存储初始化失败: %v", err)
    	} else {
    		fileStore = fs
    		logger.Println("✅ 文件持久化存储已启用 (PostgreSQL)")
    	}

    	// 初始化知识库存储（复用同一数据库连接）
    	if ks, err := store.NewKnowledgeStore(s.DB()); err != nil {
    		logger.Printf("⚠ 知识库存储初始化失败: %v", err)
    	} else {
    		knowledgeStore = ks
    		logger.Println("✅ 知识库持久化存储已启用 (PostgreSQL)")
    	}
    }

    // 初始化 WebSocket Hub
    hub := ws.NewHub()
    hub.SetStore(sessionStore)
    hub.SetIdleTimeout(cfg.SessionIdleTimeoutMin)
    hub.SetAICoreConfig(cfg.AICoreURL, cfg.InternalServiceToken)

    // 初始化规则引擎 (需要 Hub)
    if automationStore != nil {
    	ruleEngine = engine.NewRuleEngine(automationStore, hub)
    	ruleEngine.Start()
    	logger.Println("✅ 规则引擎已启动")
    }

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

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

    // 启动 WebSocket Hub
    go hub.Run()

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

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

    // 注册路由
    var db interface{}
    if sessionStore != nil {
    	db = sessionStore.DB()
    }
    // 初始化模型配置存储 (Phase 6)
    modelConfigStore, err := config.NewModelsConfigStore("../models.json")
    if err != nil {
    	logger.Printf("[WARN] 模型配置存储初始化失败 (将仅使用 .env 回退): %v", err)
    	modelConfigStore = nil
    } else if modelConfigStore.HasConfig() {
    	logger.Println("[INFO] 模型配置文件已加载 (models.json)")
    } else {
    	logger.Println("[INFO] 模型配置文件不存在，回退到 .env LLM 配置")
    }

    // 初始化思考调度配置存储
    thinkingScheduleStore, err := config.NewThinkingScheduleStore("../thinking_schedule.json")
    if err != nil {
    	logger.Printf("[WARN] 思考调度配置存储初始化失败: %v", err)
    	thinkingScheduleStore = nil
    } else {
    	logger.Println("[INFO] 思考调度配置文件已加载 (thinking_schedule.json)")
    }

    router.Setup(r, hub, cfg, sessionStore, reminderStore, automationStore, fileStore, ruleEngine, knowledgeStore, nil, db, modelConfigStore, thinkingScheduleStore)

    // 启动提醒调度器
    if reminderStore != nil {
    	handler.StartReminderScheduler(reminderStore, hub)
    }
   
    // 启动服务
    srv := &http.Server{
        Addr:    ":" + cfg.Port,
        Handler: r,
    }

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

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

    hub.StopIoTBroadcast()

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

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