228 lines
9.5 KiB
Go
228 lines
9.5 KiB
Go
// +======================================================================
|
|
// | 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}
|
|
}
|