From 1f193e7850ecd9e65463f3d7d964b6937e7d79ff Mon Sep 17 00:00:00 2001
From: TinyAnts <tinyants@163.com>
Date: Fri, 6 Jan 2023 17:47:21 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BB=A3=E7=A0=81=E7=94=9F=E6=88=90=E5=99=A8?=
 =?UTF-8?q?=E6=9E=84=E5=BB=BA=E8=8F=9C=E5=8D=95=E6=9D=83=E9=99=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/impl/GenerateServiceImpl.java     | 52 +++++++++++++++++++
 1 file changed, 52 insertions(+)

diff --git a/server/like-generator/src/main/java/com/mdd/generator/service/impl/GenerateServiceImpl.java b/server/like-generator/src/main/java/com/mdd/generator/service/impl/GenerateServiceImpl.java
index c29f4040..208db754 100644
--- a/server/like-generator/src/main/java/com/mdd/generator/service/impl/GenerateServiceImpl.java
+++ b/server/like-generator/src/main/java/com/mdd/generator/service/impl/GenerateServiceImpl.java
@@ -7,7 +7,9 @@ import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.github.pagehelper.PageHelper;
 import com.mdd.common.core.PageResult;
 
+import com.mdd.common.entity.system.SystemAuthMenu;
 import com.mdd.common.exception.OperateException;
+import com.mdd.common.mapper.system.SystemAuthMenuMapper;
 import com.mdd.common.util.StringUtils;
 import com.mdd.common.util.TimeUtils;
 import com.mdd.generator.constant.GenConstants;
@@ -60,6 +62,9 @@ public class GenerateServiceImpl implements IGenerateService {
     @Resource
     GenTableColumnMapper genTableColumnMapper;
 
+    @Resource
+    SystemAuthMenuMapper systemAuthMenuMapper;
+
     /**
      * 库列表
      *
@@ -490,6 +495,9 @@ public class GenerateServiceImpl implements IGenerateService {
                 log.error("生成渲染模板失败: " + e.getMessage());
             }
         }
+
+        // 生成菜单
+        this.genAutoMenu(table);
     }
 
     /**
@@ -532,4 +540,48 @@ public class GenerateServiceImpl implements IGenerateService {
         }
     }
 
+    /**
+     * 自动构建菜单
+     */
+    private void genAutoMenu(GenTable table) {
+        if (table.getMenuStatus().equals(1)) {
+            SystemAuthMenu authMenu = new SystemAuthMenu();
+            authMenu.setPid(table.getMenuPid());
+            authMenu.setMenuType("C");
+            authMenu.setMenuName(table.getMenuName());
+            authMenu.setPaths(table.getModuleName());
+            authMenu.setPerms(table.getModuleName() + ":list");
+            authMenu.setComponent(table.getModuleName() + "/index");
+            authMenu.setMenuName(table.getMenuName());
+            authMenu.setCreateTime(System.currentTimeMillis() / 1000);
+            authMenu.setUpdateTime(System.currentTimeMillis() / 1000);
+            systemAuthMenuMapper.insert(authMenu);
+
+            for (String op : Arrays.asList("detail", "add", "edit", "del")) {
+                String menuName = "";
+                switch (op) {
+                    case "detail":
+                        menuName = "详情";
+                        break;
+                    case "add":
+                        menuName = "新增";
+                        break;
+                    case "edit":
+                        menuName = "编辑";
+                        break;
+                    case "del":
+                        menuName = "删除";
+                }
+                String perms = table.getModuleName() + ":" + op;
+                SystemAuthMenu systemAuthMenu = new SystemAuthMenu();
+                systemAuthMenu.setPid(authMenu.getId());
+                systemAuthMenu.setMenuType("A");
+                systemAuthMenu.setPerms(perms);
+                systemAuthMenu.setMenuName(table.getMenuName() + menuName);
+                systemAuthMenu.setCreateTime(System.currentTimeMillis() / 1000);
+                systemAuthMenu.setUpdateTime(System.currentTimeMillis() / 1000);
+                systemAuthMenuMapper.insert(systemAuthMenu);
+            }
+        }
+    }
 }