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

package migration

import (
	"gofaster/internal/auth/model"
	"gofaster/internal/auth/repository"

	"gorm.io/gorm"
)

// RunMigrations 运行数据库迁移
func RunMigrations(db *gorm.DB) error {
	// 自动迁移用户表
	if err := db.AutoMigrate(&model.User{}); err != nil {
		return err
	}

	// 自动迁移角色表
	if err := db.AutoMigrate(&model.Role{}); err != nil {
		return err
	}

	// 自动迁移用户角色关联表
	if err := db.AutoMigrate(&model.UserRole{}); err != nil {
		return err
	}

	// 自动迁移验证码表
	if err := db.AutoMigrate(&repository.Captcha{}); err != nil {
		return err
	}

	// 创建默认角色
	if err := createDefaultRoles(db); err != nil {
		return err
	}

	// 创建默认管理员用户
	if err := createDefaultAdmin(db); err != nil {
		return err
	}

	return nil
}

// createDefaultRoles 创建默认角色
func createDefaultRoles(db *gorm.DB) error {
	// 检查是否已存在角色
	var count int64
	db.Model(&model.Role{}).Count(&count)
	if count > 0 {
		return nil // 已存在角色，跳过
	}

	roles := []model.Role{
		{
			Name: "超级管理员",
			Code: "SUPER_ADMIN",
			Description: "系统超级管理员，拥有所有权限",
		},
		{
			Name: "管理员",
			Code: "ADMIN",
			Description: "系统管理员，拥有大部分权限",
		},
		{
			Name: "普通用户",
			Code: "USER",
			Description: "普通用户，拥有基本权限",
		},
	}

	for _, role := range roles {
		if err := db.Create(&role).Error; err != nil {
			return err
		}
	}

	return nil
}

// createDefaultAdmin 创建默认管理员用户
func createDefaultAdmin(db *gorm.DB) error {
	// 检查是否已存在管理员用户
	var count int64
	db.Model(&model.User{}).Where("username = ?", "admin").Count(&count)
	if count > 0 {
		return nil // 已存在管理员用户，跳过
	}

	// 创建默认管理员用户
	adminUser := &model.User{
		Username: "admin",
		Password: "$2a$10$92IXUNpkjO0rOQ5byMi.Ye4oKoEa3Ro9llC/.og/at2.uheWG/igi", // "password"
		Email:    "admin@gofaster.com",
		Phone:    "13800138000",
		Status:   1, // 正常状态
	}

	if err := db.Create(adminUser).Error; err != nil {
		return err
	}

	// 获取超级管理员角色
	var superAdminRole model.Role
	if err := db.Where("code = ?", "SUPER_ADMIN").First(&superAdminRole).Error; err != nil {
		return err
	}

	// 关联超级管理员角色
	if err := db.Model(adminUser).Association("Roles").Append(&superAdminRole); err != nil {
		return err
	}

	return nil
}