package main

import (
	"context"
	"fmt"
	"log"
	"os"
	"os/signal"
	"syscall"
	"time"

	"xiaji-go/config"
	"xiaji-go/internal/bot"
	"xiaji-go/internal/qq"
	"xiaji-go/internal/service"
	"xiaji-go/internal/web"
	"xiaji-go/models"
	"xiaji-go/version"

	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
)

func main() {
	// 版本信息
	if len(os.Args) > 1 && (os.Args[1] == "-v" || os.Args[1] == "--version" || os.Args[1] == "version") {
		fmt.Println(version.Info())
		return
	}

	log.Printf("🦞 %s", version.Info())

	// 1. 加载配置
	cfgPath := "config.yaml"
	if len(os.Args) > 1 {
		cfgPath = os.Args[1]
	}

	cfg, err := config.LoadConfig(cfgPath)
	if err != nil {
		log.Fatalf("无法加载配置: %v", err)
	}

	// 2. 初始化数据库
	db, err := gorm.Open(sqlite.Open(cfg.Database.Path), &gorm.Config{})
	if err != nil {
		log.Fatalf("无法连接数据库: %v", err)
	}

	// 3. 自动迁移表结构
	if err := models.Migrate(db); err != nil {
		log.Fatalf("数据库迁移失败: %v", err)
	}

	// 4. 初始化核心服务
	finance := service.NewFinanceService(db)
	defer finance.Close()

	// 全局 context，用于优雅退出
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

	// 5. 启动 Telegram Bot
	if cfg.Telegram.Enabled {
		tgBot, err := bot.NewTGBot(cfg.Telegram.Token, finance)
		if err != nil {
			log.Printf("⚠️ TG Bot 启动失败: %v", err)
		} else {
			go tgBot.Start(ctx)
		}
	}

	// 6. 启动 QQ Bot
	if cfg.QQBot.Enabled {
		qqBot := qq.NewQQBot(cfg.QQBot.AppID, cfg.QQBot.Secret, finance)
		go qqBot.Start(ctx)
	}

	// 7. 启动 Web 后台
	webServer := web.NewWebServer(db, cfg.Server.Port, cfg.Admin.Username, cfg.Admin.Password)
	go webServer.Start()

	// 8. 优雅关闭
	log.Println("🦞 Xiaji-Go 已全面启动")
	sig := make(chan os.Signal, 1)
	signal.Notify(sig, syscall.SIGINT, syscall.SIGTERM)
	<-sig

	log.Println("⏳ 正在关闭服务...")
	cancel() // 通知所有 goroutine 停止

	// 等待一点时间让 goroutine 退出
	time.Sleep(2 * time.Second)

	// 关闭数据库连接
	sqlDB, err := db.DB()
	if err == nil {
		sqlDB.Close()
	}

	log.Println("👋 Xiaji-Go 已关闭")
}