From 25e3153ff1256c321d4985070b5f07db930026ec Mon Sep 17 00:00:00 2001
From: zhanzhiqin <895896009@qq.com>
Date: 星期三, 20 四月 2022 11:48:47 +0800
Subject: [PATCH] fix

---
 ximon-admin/src/main/java/com/sandu/ximon/admin/controller/AdminController.java |   71 +++++++++++++++++++++++++++++++++--
 1 files changed, 66 insertions(+), 5 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 7988077..e42a384 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
@@ -10,6 +10,8 @@
 import com.sandu.common.domain.ResponseVO;
 import com.sandu.common.enums.AdminStatusStatus;
 import com.sandu.common.enums.ResponseStatusEnums;
+import com.sandu.common.execption.BusinessException;
+import com.sandu.common.log.Log;
 import com.sandu.common.object.BaseConditionVO;
 import com.sandu.common.security.LoginUserInfo;
 import com.sandu.common.security.annotation.AnonymousAccess;
@@ -22,12 +24,17 @@
 import com.sandu.ximon.admin.param.AdminLoginParam;
 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;
@@ -56,8 +63,9 @@
     private final RoleService roleService;
     private final TokenProvider tokenProvider;
     private final AdminRoleRelationService adminRoleRelationService;
+    private PermissionConfig permissionConfig;
 
-
+    @Log("鍚庡彴鐢ㄦ埛鐧诲綍")
     @AnonymousAccess
     @PostMapping(value = "/login")
     public ResponseVO<Object> login(@RequestBody AdminLoginParam loginParam) {
@@ -100,13 +108,14 @@
         return ResponseUtil.success(authInfo);
     }
 
+    //    @Log("鍚庡彴鐢ㄦ埛淇℃伅")
     @GetMapping(value = "/info")
     public ResponseVO<Object> getUserInfo() {
         Long userId = SecurityUtils.getUserId();
         List<Role> roles;
         MapBuilder<Object, Object> builder;
-        if (AdministratorEnums.ADMIN.equals(SecurityUtils.getAdministratorIdentity())
-                || AdministratorEnums.CUSTOMER.equals(SecurityUtils.getAdministratorIdentity())) {
+        if (AdministratorEnums.ADMIN.getCode().equals(SecurityUtils.getAdministratorIdentity())
+                || AdministratorEnums.CUSTOMER.getCode().equals(SecurityUtils.getAdministratorIdentity())) {
             Admin admin = adminService.getById(userId);
             roles = roleService.listByAdminId(admin.getId());
             builder = MapUtil.builder().put("admin", admin);
@@ -114,7 +123,15 @@
         } else {
             Client client = clientService.getById(userId);
             roles = roleService.listByClientId(client.getId());
-            builder = MapUtil.builder().put("admin", client);
+            //涓轰簡淇濇寔杩斿洖鏁版嵁鏍煎紡涓�鑷�
+            Admin admin = new Admin();
+            admin.setId(client.getId());
+            admin.setIcon(client.getIcon());
+            admin.setNickName(client.getLinkMan());
+            admin.setMobile(client.getMobile());
+            admin.setLoginTime(client.getLoginTime());
+            admin.setCreateTime(client.getCreateTime());
+            builder = MapUtil.builder().put("admin", admin);
 
         }
         if (CollectionUtil.isEmpty(roles)) {
@@ -123,13 +140,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) {
@@ -156,6 +191,7 @@
         return ResponseUtil.successPage(adminDtoList);
     }
 
+    @Log("鍚庡彴鐢ㄦ埛娉ㄥ唽")
     @PostMapping("/register")
     public ResponseVO<Object> register(@Validated @RequestBody AdminParam param) {
         boolean result = adminService.register(param);
@@ -171,6 +207,7 @@
     /**
      * 淇敼绠$悊鍛樹俊鎭�
      */
+    @Log("鍚庡彴鐢ㄦ埛淇敼淇℃伅")
     @PostMapping("/update/{adminId}")
     public ResponseVO<Object> updateAdmin(@RequestBody AdminParam param, @PathVariable Long adminId) {
         boolean result = adminService.updateAdmin(adminId, param);
@@ -184,6 +221,7 @@
     /**
      * 淇敼鎴戠殑瀵嗙爜
      */
+    @Log("鍚庡彴鐢ㄦ埛淇敼瀵嗙爜")
     @PutMapping("/updateMyPassword")
     public ResponseVO<Object> updateMyPassword(@Validated @RequestBody PwdParam param) {
         boolean result = adminService.updateMyPassword(param);
@@ -204,6 +242,7 @@
 //        return ResponseUtil.success(null);
 //    }
 
+    @Log("鍚庡彴鐢ㄦ埛鍒犻櫎")
     @PostMapping("/del/{adminId}")
     public ResponseVO<Object> deleteAdmin(@PathVariable Long adminId) {
         boolean result = adminService.deleteAdmin(adminId);
@@ -214,6 +253,28 @@
         }
     }
 
+    @Log("鐢ㄦ埛淇敼澶村儚")
+    @PostMapping("/updateIcon")
+    public ResponseVO<Object> updateIcon(@RequestBody ReceiveParam receiveParam) {
+        if (receiveParam.getIconUrl().trim() == null) {
+            throw new BusinessException("澶村儚URL涓嶈兘涓虹┖锛�");
+        }
+        boolean result;
+        //鏅�氱敤鎴蜂慨鏀瑰ご鍍�
+        if (SecurityUtils.getAdministratorIdentity().equals(AdministratorEnums.NORMAL.getCode())) {
+            result = clientService.updateIcon(SecurityUtils.getUserId(), receiveParam.getIconUrl());
+        } else {
+            //瓒呯骇绠$悊鍛樸�佺鐞嗗憳淇敼澶村儚
+            result = adminService.updateIcon(SecurityUtils.getUserId(), receiveParam.getIconUrl());
+        }
+
+        if (result) {
+            return ResponseUtil.success("淇敼鎴愬姛");
+        } else {
+            return ResponseUtil.fail("淇敼澶辫触");
+        }
+    }
+
 //    /**
 //     * 缁欑鐞嗗憳鍒嗛厤瑙掕壊
 //     *

--
Gitblit v1.9.3