package migration

import (
	"gofaster/internal/auth/model"

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

// CreateRouteTables 创建路由相关表
func CreateRouteTables(db *gorm.DB, log *zap.Logger) error {
	log.Info("开始创建路由相关表...")

	// 创建菜单分组表
	if err := db.AutoMigrate(&model.MenuGroup{}); err != nil {
		log.Error("创建菜单分组表失败", zap.Error(err))
		return err
	}
	log.Info("✅ 菜单分组表创建完成")

	// 创建前台路由表
	if err := db.AutoMigrate(&model.FrontendRoute{}); err != nil {
		log.Error("创建前台路由表失败", zap.Error(err))
		return err
	}
	log.Info("✅ 前台路由表创建完成")

	// 创建前后台路由关系表
	if err := db.AutoMigrate(&model.FrontendBackendRoute{}); err != nil {
		log.Error("创建前后台路由关系表失败", zap.Error(err))
		return err
	}
	log.Info("✅ 前后台路由关系表创建完成")

	// 为现有Resource表添加菜单相关字段
	if err := addMenuFieldsToResource(db, log); err != nil {
		log.Error("为Resource表添加菜单字段失败", zap.Error(err))
		return err
	}
	log.Info("✅ Resource表菜单字段添加完成")

	log.Info("路由相关表创建完成")
	return nil
}

// addMenuFieldsToResource 为Resource表添加菜单相关字段
func addMenuFieldsToResource(db *gorm.DB, log *zap.Logger) error {
	// 检查IsMenu字段是否存在
	var hasIsMenu bool
	err := db.Raw("SELECT COUNT(*) > 0 FROM information_schema.columns WHERE table_name = 'resources' AND column_name = 'is_menu'").Scan(&hasIsMenu).Error
	if err != nil {
		return err
	}

	if !hasIsMenu {
		// 添加IsMenu字段
		if err := db.Exec("ALTER TABLE resources ADD COLUMN is_menu BOOLEAN DEFAULT FALSE").Error; err != nil {
			return err
		}
		log.Info("添加is_menu字段到resources表")
	}

	// 移除旧的MenuID字段（如果存在）
	var hasMenuID bool
	err = db.Raw("SELECT COUNT(*) > 0 FROM information_schema.columns WHERE table_name = 'resources' AND column_name = 'menu_id'").Scan(&hasMenuID).Error
	if err != nil {
		return err
	}

	if hasMenuID {
		// 删除MenuID字段
		if err := db.Exec("ALTER TABLE resources DROP COLUMN menu_id").Error; err != nil {
			log.Warn("删除menu_id字段失败，可能已被删除", zap.Error(err))
		} else {
			log.Info("删除menu_id字段从resources表")
		}
	}

	return nil
}