manta/gofaster/backend/internal/auth/migration/create_route_tables.go

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.Menu{}); err != nil {
		log.Error("创建菜单表失败", zap.Error(err))
		return err
	}
	log.Info("✅ 菜单表创建完成")

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

	// 创建菜单路由关联表
	if err := db.AutoMigrate(&model.MenuRoute{}); 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表菜单字段添加完成")

	// 处理RouteMapping表的字段变更
	if err := updateRouteMappingFields(db, log); err != nil {
		log.Error("更新RouteMapping表字段失败", zap.Error(err))
		return err
	}
	log.Info("✅ RouteMapping表字段更新完成")

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

// updateRouteMappingFields 更新RouteMapping表的字段
func updateRouteMappingFields(db *gorm.DB, log *zap.Logger) error {
	// 移除旧的MenuID字段（如果存在）
	var hasMenuID bool
	err := db.Raw("SELECT COUNT(*) > 0 FROM information_schema.columns WHERE table_name = 'route_mappings' AND column_name = 'menu_id'").Scan(&hasMenuID).Error
	if err != nil {
		return err
	}

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

	// 添加AuthGroup字段（如果不存在）
	var hasAuthGroup bool
	err = db.Raw("SELECT COUNT(*) > 0 FROM information_schema.columns WHERE table_name = 'route_mappings' AND column_name = 'auth_group'").Scan(&hasAuthGroup).Error
	if err != nil {
		return err
	}

	if !hasAuthGroup {
		// 添加AuthGroup字段
		if err := db.Exec("ALTER TABLE route_mappings ADD COLUMN auth_group VARCHAR(20) DEFAULT 'Read'").Error; err != nil {
			log.Error("添加auth_group字段失败", zap.Error(err))
			return err
		}
		log.Info("添加auth_group字段到route_mappings表")

		// 根据HTTP方法更新现有记录的AuthGroup
		if err := updateAuthGroupByMethod(db, log); err != nil {
			log.Error("更新AuthGroup失败", zap.Error(err))
			return err
		}
	}

	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
}

// updateAuthGroupByMethod 根据HTTP方法更新AuthGroup
func updateAuthGroupByMethod(db *gorm.DB, log *zap.Logger) error {
	// 更新修改型操作为Edit分组
	editMethods := []string{"POST", "PUT", "PATCH", "DELETE"}
	for _, method := range editMethods {
		if err := db.Exec("UPDATE route_mappings SET auth_group = 'Edit' WHERE http_method = ?", method).Error; err != nil {
			log.Error("更新Edit分组失败", zap.String("method", method), zap.Error(err))
			return err
		}
	}

	// 更新读取型操作为Read分组
	readMethods := []string{"GET", "HEAD", "OPTIONS"}
	for _, method := range readMethods {
		if err := db.Exec("UPDATE route_mappings SET auth_group = 'Read' WHERE http_method = ?", method).Error; err != nil {
			log.Error("更新Read分组失败", zap.String("method", method), zap.Error(err))
			return err
		}
	}

	log.Info("AuthGroup更新完成")
	return nil
}