// +====================================================================== // | EasyGoAdmin 企业数字化开发平台 [ 赋能开发者,助力企业发展 ] // +---------------------------------------------------------------------- // | Copyright (c) 2019-2026 半城风雨团队 版权所有 // +---------------------------------------------------------------------- // | 开源协议: Apache-2.0 // | 重要声明: 未经书面许可,不得移除或修改本框架相关版权标识。 // +---------------------------------------------------------------------- // | 官方网站: https://www.easygoadmin.vip // +---------------------------------------------------------------------- // | 研发团队: 小蚂蚁云团队 荣誉出品 // +====================================================================== // | 版权和免责声明: // | 一、知识产权声明 // | 本团队对该软件框架产品(以下简称“本产品”)拥有完整知识产权,包括但不限于商标权、专利权、 // | 著作权、商业秘密等。前述权利均受中华人民共和国及相关国际法律法规保护。未经本团队书面授权, // | 任何个人或单位不得基于本产品本身申请任何形式的知识产权。 // | 二、使用限制与禁止条款 // | 您理解并同意,禁止将本产品用于以下任何行为: // | 1. 任何违反中华人民共和国法律法规的用途; // | 2. 任何侵害他人合法权益(包括但不限于人身权、财产权、知识产权等)的行为; // | 3. 任何危害国家安全、公共利益及社会道德风尚的行为。 // | 三、责任豁免 // | 因使用本产品而产生的任何损失或法律纠纷(包括但不限于直接损失、间接损失、附带损失、衍生损失 // | 以及因违约、侵权等引发的责任),本团队均不承担任何法律责任。若因违规使用本产品导致任何第三 // | 方提起索赔或行政机关进行调查,您应独立承担全部责任。 // | 四、违约与法律责任 // | 如发现任何违反本声明的行为,本团队有权: // | 1. 无条件配合公安机关及其他有权机关的调查取证工作; // | 2. 依法追究相关方的民事责任,并要求赔偿全部损失; // | 3. 保留进一步采取刑事控告等法律措施的权利。 // | 五、许可范围说明 // | 1、被授权方应视本产品源代码为核心机密信息,负有最高标准的保密与保管义务。禁止以任何形式(包 // | 括但不限于直接或间接的分享、复制、传播、分发、转售、出租、开源、反向工程、反编译、反汇编或通 // | 过网络服务提供给任何第三方)泄露、披露或处置源代码的全部或部分内容。任何未经授权的复制、使用、 // | 传播行为均构成对本团队知识产权的严重侵权,本团队有权依法追究其全部法律责任,并要求赔偿因此造 // | 成的一切经济损失及商誉损害。 // | 2、本产品仅限于公司或个人在法律允许范围内,用于内部且合规的软件产品研发。具体权利义务以《软件 // | 许可协议》附件为准。 // +====================================================================== 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} }