EasyGoAdmin_GoFrame_EleVue/internal/logic/generator_side.go

package logic

import (
	"context"
	"fmt"
	"os"
	"strings"

	"github.com/gogf/gf/v2/frame/g"

	"easygoadmin/api/v1"
	"easygoadmin/internal/dao"
)

// ============================================================
// 代码生成器副作用同步：模块菜单与权限节点、字典数据、router.go 路由注册
// 所有副作用均保证幂等
// ============================================================

// createMenu 生成模块主菜单与权限节点
//
// 菜单挂在 generator.parentMenuId 指定的父菜单下；
// 按 path 查重保证幂等，已存在时直接返回。
func (s *sGenerator) createMenu(ctx context.Context, vars *genVars) v1.GeneratorMenuResult {
	path := "/" + vars.Domain + "/" + vars.ModuleCamel
	component := path + "/index"

	exists, err := dao.Menu.Ctx(ctx).Where("path", path).Where("type", 0).One()
	if err == nil && len(exists) > 0 {
		return v1.GeneratorMenuResult{Success: false, MenuId: exists["id"].Int64(), Message: "菜单已存在"}
	}

	parentId := g.Cfg().MustGet(ctx, "generator.parentMenuId").Int()
	// 同级最大 sort + 5
	sortBase := 5
	if one, e := dao.Menu.Ctx(ctx).Where("parent_id", parentId).Fields("MAX(sort) AS max_sort").One(); e == nil && !one["max_sort"].IsNil() {
		sortBase = one["max_sort"].Int() + 5
	}

	result, err := dao.Menu.Ctx(ctx).Data(map[string]any{
		"parent_id":  parentId,
		"name":       vars.ModuleComment,
		"icon":       "el-icon-Document",
		"icon2":      "GlobalOutlined",
		"path":       path,
		"component":  component,
		"permission": "",
		"target":     0,
		"type":       0,
		"status":     0,
		"hide":       0,
		"sort":       sortBase,
		"note":       "代码生成器自动创建",
	}).Insert()
	if err != nil {
		return v1.GeneratorMenuResult{Success: false, Message: "菜单创建失败: " + err.Error()}
	}
	menuId, _ := result.LastInsertId()

	// 权限节点（与生成模块实际路由一一对应）
	type perm struct{ name, code string }
	perms := []perm{
		{vars.ModuleComment + "-分页查询", ":page"},
		{vars.ModuleComment + "-列表查询", ":list"},
		{vars.ModuleComment + "-详情查询", ":detail"},
		{vars.ModuleComment + "-新增", ":add"},
		{vars.ModuleComment + "-编辑", ":update"},
		{vars.ModuleComment + "-删除", ":delete"},
		{vars.ModuleComment + "-批量删除", ":batchDelete"},
	}
	if vars.HasStatus {
		perms = append(perms, perm{vars.ModuleComment + "-状态管理", ":status"})
	}
	if vars.HasImport {
		perms = append(perms, perm{vars.ModuleComment + "-导入", ":import"})
	}
	if vars.HasExport {
		perms = append(perms, perm{vars.ModuleComment + "-导出", ":export"})
	}
	nodeSort := 0
	for _, pm := range perms {
		nodeSort += 5
		if _, e := dao.Menu.Ctx(ctx).Data(map[string]any{
			"parent_id":  menuId,
			"name":       pm.name,
			"icon":       "",
			"path":       "",
			"component":  "",
			"permission": vars.PermissionPrefix + pm.code,
			"target":     0,
			"type":       1,
			"status":     0,
			"hide":       0,
			"sort":       nodeSort,
			"note":       "代码生成器自动创建",
		}).Insert(); e != nil {
			return v1.GeneratorMenuResult{Success: false, MenuId: menuId, Message: "权限节点创建失败: " + e.Error()}
		}
	}
	return v1.GeneratorMenuResult{Success: true, MenuId: menuId,
		Message: fmt.Sprintf("菜单创建成功（含 %d 个权限节点）", len(perms))}
}

// createDict 为带枚举注释的表单字段创建字典
//
// 字典编码形如 {module_snake}_{field_snake}，按编码查重保证幂等
func (s *sGenerator) createDict(ctx context.Context, vars *genVars) []v1.GeneratorDictResult {
	results := []v1.GeneratorDictResult{}
	for _, f := range vars.FormFields {
		if !f.HasChoices {
			continue
		}
		code := vars.Module + "_" + f.Name
		exists, err := dao.Dict.Ctx(ctx).Where("code", code).One()
		if err == nil && len(exists) > 0 {
			results = append(results, v1.GeneratorDictResult{Dict: code, Status: "已存在"})
			continue
		}
		dResult, err := dao.Dict.Ctx(ctx).Data(map[string]any{
			"name": vars.ModuleComment + "-" + f.Label,
			"code": code,
			"sort": 0,
			"note": "代码生成器自动创建",
		}).Insert()
		if err != nil {
			results = append(results, v1.GeneratorDictResult{Dict: code, Status: "创建失败: " + err.Error()})
			continue
		}
		dictId, _ := dResult.LastInsertId()
		itemSort := 0
		for _, c := range f.Choices {
			itemSort += 5
			if _, e := dao.DictItem.Ctx(ctx).Data(map[string]any{
				"dict_id": dictId,
				"name":    c.Label,
				"value":   c.Value,
				"sort":    itemSort,
				"note":    "",
			}).Insert(); e != nil {
				results = append(results, v1.GeneratorDictResult{Dict: code, Status: "字典项失败: " + e.Error()})
				break
			}
		}
		results = append(results, v1.GeneratorDictResult{
			Dict: code, Name: vars.ModuleComment + "-" + f.Label, Items: len(f.Choices), Status: "创建成功",
		})
	}
	return results
}

// registerRouter 将路由注册行写入 internal/router/router.go
//
// 按控制器是否已存在保证幂等，已注册时直接返回；
// 在最后一个 Bind 行之后插入新注册行。
func (s *sGenerator) registerRouter(vars *genVars) v1.GeneratorRouteResult {
	const routerFile = "internal/router/router.go"
	content, err := os.ReadFile(routerFile)
	if err != nil {
		return v1.GeneratorRouteResult{Success: false, Message: "路由文件不存在: " + err.Error()}
	}
	text := string(content)

	if strings.Contains(text, "new(controller.C"+vars.ModelClass+")") {
		return v1.GeneratorRouteResult{Success: false, Message: "路由已存在: " + vars.ModelClass}
	}

	// 1) 控制器为单一 controller 包（扁平结构），确认已导入
	if !strings.Contains(text, `"easygoadmin/internal/controller"`) {
		return v1.GeneratorRouteResult{Success: false, Message: "路由缺少 controller 导入"}
	}

	// 2) 在最后一个 Bind 行之后插入注册行
	idx := strings.LastIndex(text, "\n\tBind(s, new(")
	if idx < 0 {
		return v1.GeneratorRouteResult{Success: false, Message: "未找到路由注册锚点"}
	}
	lineEnd := strings.Index(text[idx:], "\n")
	insertAt := idx + lineEnd
	newLine := "\n\tBind(s, new(controller.C" + vars.ModelClass + ")) // 代码生成器注册"
	text = text[:insertAt] + newLine + text[insertAt:]

	if err = os.WriteFile(routerFile, []byte(text), 0o644); err != nil {
		return v1.GeneratorRouteResult{Success: false, Message: "路由写入失败: " + err.Error()}
	}
	return v1.GeneratorRouteResult{Success: true, Message: "路由注册成功: " + vars.ModelClass}
}