package database

import (
	"context"
	"database/sql"
	"embed"
	"fmt"
	"strings"
)

//go:embed migrations/sqlite/0001_initial.sql migrations/sqlite/0002_domain_options.sql migrations/mysql/0001_initial.sql migrations/mysql/0002_domain_options.sql
var migrationFiles embed.FS

func Migrate(ctx context.Context, db *sql.DB, driver string) error {
	if db == nil {
		return fmt.Errorf("数据库连接不能为空")
	}
	if driver != "sqlite" && driver != "mysql" {
		return fmt.Errorf("数据库驱动无效：%s", driver)
	}
	if _, err := db.ExecContext(ctx, migrationTableSQL(driver)); err != nil {
		return fmt.Errorf("创建迁移表失败：%w", err)
	}
	var version int
	err := db.QueryRowContext(ctx, "SELECT version FROM schema_migrations ORDER BY version DESC LIMIT 1").Scan(&version)
	if err != nil && err != sql.ErrNoRows {
		return fmt.Errorf("读取迁移版本失败：%w", err)
	}
	if err == sql.ErrNoRows {
		version = 0
	}
	for nextVersion := version + 1; nextVersion <= latestMigrationVersion; nextVersion++ {
		if err := applyMigration(ctx, db, driver, nextVersion); err != nil {
			return err
		}
	}
	return nil
}

const latestMigrationVersion = 2

func applyMigration(ctx context.Context, db *sql.DB, driver string, version int) error {
	path := fmt.Sprintf("migrations/%s/%04d_%s.sql", driver, version, migrationName(version))
	content, err := migrationFiles.ReadFile(path)
	if err != nil {
		return fmt.Errorf("读取迁移文件失败：%w", err)
	}
	statements := migrationStatements(content)
	if driver == "mysql" {
		for _, statement := range statements {
			if _, err := db.ExecContext(ctx, statement); err != nil {
				return fmt.Errorf("执行数据库迁移失败：%w", err)
			}
		}
		if _, err := db.ExecContext(ctx, "INSERT INTO schema_migrations (version) VALUES (?)", version); err != nil {
			return fmt.Errorf("记录迁移版本失败：%w", err)
		}
		return nil
	}
	tx, err := db.BeginTx(ctx, nil)
	if err != nil {
		return fmt.Errorf("开始迁移事务失败：%w", err)
	}
	for _, statement := range statements {
		if _, err := tx.ExecContext(ctx, statement); err != nil {
			_ = tx.Rollback()
			return fmt.Errorf("执行数据库迁移失败：%w", err)
		}
	}
	if _, err := tx.ExecContext(ctx, "INSERT INTO schema_migrations (version) VALUES (?)", version); err != nil {
		_ = tx.Rollback()
		return fmt.Errorf("记录迁移版本失败：%w", err)
	}
	if err := tx.Commit(); err != nil {
		return fmt.Errorf("提交数据库迁移失败：%w", err)
	}
	return nil
}

func migrationName(version int) string {
	switch version {
	case 1:
		return "initial"
	case 2:
		return "domain_options"
	default:
		return "unknown"
	}
}

func migrationStatements(content []byte) []string {
	statements := make([]string, 0)
	for _, statement := range strings.Split(string(content), ";") {
		statement = strings.TrimSpace(statement)
		if statement != "" {
			statements = append(statements, statement)
		}
	}
	return statements
}

func migrationTableSQL(driver string) string {
	if driver == "mysql" {
		return `CREATE TABLE IF NOT EXISTS schema_migrations (
			version INT NOT NULL PRIMARY KEY,
			applied_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP
		)`
	}
	return `CREATE TABLE IF NOT EXISTS schema_migrations (
		version INTEGER NOT NULL PRIMARY KEY,
		applied_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP
	)`
}