package database

import (
	"backend/config"
	"backend/internal/model"
	"fmt"

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

// Init 初始化数据库连接并执行迁移 // Logan.Li
func Init(cfg *config.Config) (*gorm.DB, error) {
	logLevel := getLogLevel(cfg.Mode)

	fmt.Printf("[数据库] 使用 SQLite: %s\n", cfg.DBName)
	db, err := gorm.Open(sqlite.Open(cfg.DBName), &gorm.Config{
		Logger: logger.Default.LogMode(logLevel),
	})
	if err != nil {
		return nil, fmt.Errorf("SQLite 连接失败: %v", err)
	}

	return migrateDatabase(db)
}

// getLogLevel 根据运行模式返回日志级别 // Logan.Li
func getLogLevel(mode string) logger.LogLevel {
	if mode == "release" {
		return logger.Warn
	}
	return logger.Info
}

// migrateDatabase 执行数据库迁移 // Logan.Li
func migrateDatabase(db *gorm.DB) (*gorm.DB, error) {
	err := db.AutoMigrate(
		// 认证与系统
		&model.User{},                  // 管理员用户 // Logan.Li
		&model.Setting{},               // 系统设置 // Logan.Li
		&model.AuditLog{},              // 操作日志 // Logan.Li
		&model.LicenseTemplate{},       // License 模板 // Logan.Li
		&model.PaymentGatewayConfig{},  // 支付网关配置 // Logan.Li
		// 核心授权域
		&model.Product{},               // 产品 // Logan.Li
		&model.Customer{},              // 客户 // Logan.Li
		&model.License{},               // License // Logan.Li
		&model.Activation{},            // 激活记录 // Logan.Li
		// 定价与订单
		&model.Plan{},                  // 定价计划 // Logan.Li
		&model.Order{},                 // 订单 // Logan.Li
		// 防重复购买
		&model.CustomerEntitlement{},   // 客户权益 // Logan.Li
	)
	if err != nil {
		return nil, fmt.Errorf("数据库迁移失败: %v", err)
	}

	// 初始化种子数据（默认管理员、默认设置、示例数据） // Logan.Li
	if err := SeedDefaults(db); err != nil {
		return nil, fmt.Errorf("种子数据初始化失败: %v", err)
	}

	return db, nil
}