package database

import (
	"fmt"
	"time"

	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
	"gorm.io/gorm/schema"

	"github.com/sundynix/pets-be/internal/config"
	"github.com/sundynix/pets-be/internal/model"
)

// New 建立 GORM 连接，所有表名统一加 sundynix_ 前缀
func New(cfg *config.Config) (*gorm.DB, error) {
	logLevel := logger.Info
	if cfg.Server.Mode == "release" {
		logLevel = logger.Warn
	}

	db, err := gorm.Open(mysql.Open(cfg.MySQL.DSN()), &gorm.Config{
		NamingStrategy: schema.NamingStrategy{
			TablePrefix:   "sundynix_",
			SingularTable: false,
		},
		// 关联在 Go 层管，不靠库级外键。迁移期建外键时 MySQL 会为外键列自动
		// 补一个索引，和字段上的 `index` tag 撞成同名索引，父子表同迁时第二次
		// 建索引报 1061 duplicate。关掉外键约束创建，索引就只由 index tag 建一次
		DisableForeignKeyConstraintWhenMigrating: true,
		Logger: logger.Default.LogMode(logLevel),
	})
	if err != nil {
		return nil, fmt.Errorf("gorm open: %w", err)
	}

	sqlDB, err := db.DB()
	if err != nil {
		return nil, err
	}
	sqlDB.SetMaxOpenConns(50)
	sqlDB.SetMaxIdleConns(10)
	sqlDB.SetConnMaxLifetime(time.Hour)

	return db, nil
}

// Migrate 自动建表
func Migrate(db *gorm.DB) error {
	return db.AutoMigrate(model.AllModels()...)
}