package migration

import (
	"fmt"

	"go.uber.org/zap"
	"gorm.io/gorm"
)

// RemoveDeleteAtFields 移除相关表的 delete_at 字段
func RemoveDeleteAtFields(db *gorm.DB, log *zap.Logger) error {
	log.Info("开始移除相关表的 delete_at 字段...")

	// 1. 移除 frontend_backend_routes 表的 delete_at 字段
	if err := removeDeleteAtFromFrontendBackendRoutes(db, log); err != nil {
		return fmt.Errorf("移除 frontend_backend_routes 表 delete_at 字段失败: %w", err)
	}

	// 2. 移除 frontend_routes 表的 delete_at 字段
	if err := removeDeleteAtFromFrontendRoutes(db, log); err != nil {
		return fmt.Errorf("移除 frontend_routes 表 delete_at 字段失败: %w", err)
	}

	// 3. 移除 route_mappings 表的 delete_at 字段
	if err := removeDeleteAtFromRouteMappings(db, log); err != nil {
		return fmt.Errorf("移除 route_mappings 表 delete_at 字段失败: %w", err)
	}

	log.Info("✅ 所有表的 delete_at 字段移除完成")
	return nil
}

// removeDeleteAtFromFrontendBackendRoutes 移除 frontend_backend_routes 表的 delete_at 字段
func removeDeleteAtFromFrontendBackendRoutes(db *gorm.DB, log *zap.Logger) error {
	log.Info("移除 frontend_backend_routes 表的 delete_at 字段...")

	// 检查表是否存在
	if !db.Migrator().HasTable("frontend_backend_routes") {
		log.Info("frontend_backend_routes 表不存在，跳过")
		return nil
	}

	// 检查 delete_at 字段是否存在
	if !db.Migrator().HasColumn("frontend_backend_routes", "deleted_at") {
		log.Info("frontend_backend_routes 表没有 deleted_at 字段，跳过")
		return nil
	}

	// 删除 delete_at 字段
	if err := db.Exec("ALTER TABLE frontend_backend_routes DROP COLUMN deleted_at").Error; err != nil {
		return fmt.Errorf("删除 deleted_at 字段失败: %w", err)
	}

	log.Info("✅ frontend_backend_routes 表的 deleted_at 字段移除成功")
	return nil
}

// removeDeleteAtFromFrontendRoutes 移除 frontend_routes 表的 delete_at 字段
func removeDeleteAtFromFrontendRoutes(db *gorm.DB, log *zap.Logger) error {
	log.Info("移除 frontend_routes 表的 delete_at 字段...")

	// 检查表是否存在
	if !db.Migrator().HasTable("frontend_routes") {
		log.Info("frontend_routes 表不存在，跳过")
		return nil
	}

	// 检查 delete_at 字段是否存在
	if !db.Migrator().HasColumn("frontend_routes", "deleted_at") {
		log.Info("frontend_routes 表没有 deleted_at 字段，跳过")
		return nil
	}

	// 删除 delete_at 字段
	if err := db.Exec("ALTER TABLE frontend_routes DROP COLUMN deleted_at").Error; err != nil {
		return fmt.Errorf("删除 deleted_at 字段失败: %w", err)
	}

	log.Info("✅ frontend_routes 表的 deleted_at 字段移除成功")
	return nil
}

// removeDeleteAtFromRouteMappings 移除 route_mappings 表的 delete_at 字段
func removeDeleteAtFromRouteMappings(db *gorm.DB, log *zap.Logger) error {
	log.Info("移除 route_mappings 表的 delete_at 字段...")

	// 检查表是否存在
	if !db.Migrator().HasTable("route_mappings") {
		log.Info("route_mappings 表不存在，跳过")
		return nil
	}

	// 检查 delete_at 字段是否存在
	if !db.Migrator().HasColumn("route_mappings", "deleted_at") {
		log.Info("route_mappings 表没有 deleted_at 字段，跳过")
		return nil
	}

	// 删除 delete_at 字段
	if err := db.Exec("ALTER TABLE route_mappings DROP COLUMN deleted_at").Error; err != nil {
		return fmt.Errorf("删除 deleted_at 字段失败: %w", err)
	}

	log.Info("✅ route_mappings 表的 deleted_at 字段移除成功")
	return nil
}