From f86fd019e49a4331c18bd5d736096af09f2a73ad Mon Sep 17 00:00:00 2001
From: liuhaonan <konodioda2333@vip.qq.com>
Date: 星期一, 18 四月 2022 15:09:47 +0800
Subject: [PATCH] 节目/文件归属问题

---
 ximon-admin/src/main/java/com/sandu/ximon/admin/controller/AdminController.java |   24 +++++++++++++++++++++++-
 1 files changed, 23 insertions(+), 1 deletions(-)

diff --git a/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/AdminController.java b/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/AdminController.java
index 49b96c0..cf3af06 100644
--- a/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/AdminController.java
+++ b/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/AdminController.java
@@ -25,12 +25,16 @@
 import com.sandu.ximon.admin.param.AdminParam;
 import com.sandu.ximon.admin.param.PwdParam;
 import com.sandu.ximon.admin.param.ReceiveParam;
+import com.sandu.ximon.admin.security.PermissionConfig;
 import com.sandu.ximon.admin.security.SecurityUtils;
 import com.sandu.ximon.admin.service.*;
+import com.sandu.ximon.dao.bo.MenuNode;
 import com.sandu.ximon.dao.domain.*;
 import com.sandu.ximon.dao.enums.AdministratorEnums;
+import com.sandu.ximon.dao.enums.MenuEnum;
 import lombok.AllArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.BeanUtils;
 import org.springframework.security.core.GrantedAuthority;
 import org.springframework.security.crypto.password.PasswordEncoder;
 import org.springframework.validation.annotation.Validated;
@@ -59,6 +63,7 @@
     private final RoleService roleService;
     private final TokenProvider tokenProvider;
     private final AdminRoleRelationService adminRoleRelationService;
+    private PermissionConfig permissionConfig;
 
     @Log("鍚庡彴鐢ㄦ埛鐧诲綍")
     @AnonymousAccess
@@ -127,14 +132,31 @@
         List<Menu> menus = menuService.listByRoleIds(roles.stream().map(Role::getId).collect(Collectors.toList()));
         if (CollectionUtil.isNotEmpty(menus)) {
             List<Menu> collect = menus.stream().filter(menu -> menu.getType() == 0 || menu.getType() == 1).collect(Collectors.toList());
-            builder.put("menus", collect);
+
+            List<MenuNode> menuNodeList = collect.stream()
+                    .filter(menu -> menu.getPid().equals(0L))
+                    .map(menu -> covertMenuNode(menu, collect)).collect(Collectors.toList());
+            builder.put("menus", menuNodeList);
         }
         return ResponseUtil.success(builder.build());
+    }
+
+    private MenuNode covertMenuNode(Menu menu, List<Menu> menuList) {
+        MenuNode node = new MenuNode();
+        BeanUtils.copyProperties(menu, node);
+        List<MenuNode> children = menuList.stream()
+                .filter(subMenu -> subMenu.getPid().equals(menu.getId()))
+                .map(subMenu -> covertMenuNode(subMenu, menuList)).collect(Collectors.toList());
+        node.setChildren(children);
+        return node;
     }
 
     @Log("鍚庡彴鐢ㄦ埛鍒楄〃")
     @GetMapping(value = "/list")
     public ResponseVO<Object> listAdmin(BaseConditionVO baseConditionVO, @RequestParam(value = "keyword", required = false) String keyword) {
+        if (!permissionConfig.check(MenuEnum.ADMIN_LIST.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         PageHelper.startPage(baseConditionVO.getPageNo(), baseConditionVO.getPageSize());
         LambdaQueryWrapper<Admin> wrapper = Wrappers.lambdaQuery(Admin.class);
         if (keyword != null) {

--
Gitblit v1.9.3