From 3a8ecfd6062adb7f75777656fa7cb1a5a411535f Mon Sep 17 00:00:00 2001
From: zhanzhiqin <895896009@qq.com>
Date: 星期二, 10 五月 2022 17:25:31 +0800
Subject: [PATCH] 一键求助

---
 ximon-admin/src/main/java/com/sandu/ximon/admin/controller/AdminController.java |  142 ++++++++++++++++++++++++++++++++++------------
 1 files changed, 104 insertions(+), 38 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 e42a384..0b69b22 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
@@ -1,33 +1,29 @@
 package com.sandu.ximon.admin.controller;
 
-import cn.hutool.core.bean.BeanUtil;
 import cn.hutool.core.collection.CollectionUtil;
 import cn.hutool.core.map.MapBuilder;
 import cn.hutool.core.map.MapUtil;
-import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import cn.hutool.core.util.StrUtil;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
-import com.github.pagehelper.PageHelper;
 import com.sandu.common.domain.ResponseVO;
 import com.sandu.common.enums.AdminStatusStatus;
 import com.sandu.common.enums.ResponseStatusEnums;
+import com.sandu.common.enums.RoleLevelStatus;
 import com.sandu.common.execption.BusinessException;
 import com.sandu.common.log.Log;
 import com.sandu.common.object.BaseConditionVO;
+import com.sandu.common.redis.RedisService;
 import com.sandu.common.security.LoginUserInfo;
 import com.sandu.common.security.annotation.AnonymousAccess;
 import com.sandu.common.security.config.SecurityProperties;
 import com.sandu.common.security.token.TokenProvider;
 import com.sandu.common.util.IpUtil;
 import com.sandu.common.util.ResponseUtil;
-import com.sandu.ximon.admin.dto.AdminDto;
-import com.sandu.ximon.admin.dto.ClientDto;
-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.param.*;
 import com.sandu.ximon.admin.security.PermissionConfig;
 import com.sandu.ximon.admin.security.SecurityUtils;
 import com.sandu.ximon.admin.service.*;
+import com.sandu.ximon.admin.utils.RedisUtils;
 import com.sandu.ximon.dao.bo.MenuNode;
 import com.sandu.ximon.dao.domain.*;
 import com.sandu.ximon.dao.enums.AdministratorEnums;
@@ -40,8 +36,12 @@
 import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
 
+import javax.servlet.http.HttpServletRequest;
 import java.time.LocalDateTime;
-import java.util.*;
+import java.util.Collection;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
 import java.util.stream.Collectors;
 
 /**
@@ -64,15 +64,24 @@
     private final TokenProvider tokenProvider;
     private final AdminRoleRelationService adminRoleRelationService;
     private PermissionConfig permissionConfig;
+    private final RedisService redisService;
 
     @Log("鍚庡彴鐢ㄦ埛鐧诲綍")
     @AnonymousAccess
     @PostMapping(value = "/login")
-    public ResponseVO<Object> login(@RequestBody AdminLoginParam loginParam) {
+    public ResponseVO<Object> login(HttpServletRequest request, @RequestBody AdminLoginParam loginParam) {
         Admin admin = adminService.getOne(Wrappers.lambdaQuery(Admin.class).eq(Admin::getUsername, loginParam.getUsername()).last("limit 1"));
         if (admin == null) {
             return ResponseUtil.error(ResponseStatusEnums.BAD_CREDENTIALS.getCode(), ResponseStatusEnums.BAD_CREDENTIALS.getMessage());
         }
+//        String session_vcode = (String) request.getSession().getAttribute(loginParam.getKey());
+        String s = RedisUtils.getBean().get(loginParam.getKey());
+        System.out.println(s + "===========================");
+        if (!loginParam.getVlue().equals(s)) {
+            log.error(loginParam.getVlue() + "楠岃瘉鐮�");
+            return ResponseUtil.error(ResponseStatusEnums.BAD_AUTHENTICATION.getCode(), ResponseStatusEnums.BAD_AUTHENTICATION.getMessage());
+        }
+        RedisUtils.getBean().delete(loginParam.getKey());
         if (!passwordEncoder.matches(loginParam.getPassword(), admin.getPassword())) {
             return ResponseUtil.error(ResponseStatusEnums.BAD_CREDENTIALS.getCode(), ResponseStatusEnums.BAD_CREDENTIALS.getMessage());
         }
@@ -88,7 +97,16 @@
         loginUserInfo.setAccount(admin.getUsername());
         loginUserInfo.setStatus(admin.getStatus());
         loginUserInfo.setIp(IpUtil.getRealIp());
-        loginUserInfo.setAdministratorType(AdministratorEnums.ADMIN.getCode());
+
+        //纭畾鐢ㄦ埛瑙掕壊鏉冮檺
+        AdminRoleRelation adminRoleRelation = adminRoleRelationService.getOne(Wrappers.lambdaQuery(AdminRoleRelation.class).eq(AdminRoleRelation::getAdminId, loginUserInfo.getUserId()));
+        Role role = roleService.getOne(Wrappers.lambdaQuery(Role.class).eq(Role::getId, adminRoleRelation.getRoleId()));
+        if (role != null && role.getLevel().equals(RoleLevelStatus.SUPER.getCode())) {
+            loginUserInfo.setAdministratorType(AdministratorEnums.ADMIN.getCode());
+        } else {
+            loginUserInfo.setAdministratorType(AdministratorEnums.CUSTOMER.getCode());
+        }
+
         loginUserInfo.setPermission(grantedAuthorities.stream().map(GrantedAuthority::getAuthority).collect(Collectors.joining(",")));
 
         // 鐢熸垚浠ょ墝
@@ -111,6 +129,7 @@
     //    @Log("鍚庡彴鐢ㄦ埛淇℃伅")
     @GetMapping(value = "/info")
     public ResponseVO<Object> getUserInfo() {
+        System.out.println(SecurityUtils.getUserDetails().getAdministratorType() + "         ---------");
         Long userId = SecurityUtils.getUserId();
         List<Role> roles;
         MapBuilder<Object, Object> builder;
@@ -149,6 +168,17 @@
         return ResponseUtil.success(builder.build());
     }
 
+    @GetMapping(value = "/logout")
+    public ResponseVO<Object> LogOut() {
+//        redisService.set(UserModel.USER_INFO.key(SecurityUtils.getUserId().toString() + SecurityUtils.getUserDetails().getAdministratorType()), null);
+//        redisService.set(UserModel.USER_TOKEN.key(UserModel.USER_TOKEN.key(SecurityUtils.getUserDetails().getToken())), null);
+
+
+        String key = String.format("%d_%d", SecurityUtils.getUserDetails().getUserId(), SecurityUtils.getUserDetails().getAdministratorType());
+        redisService.set(key, null);
+        return ResponseUtil.success("閫�鍑虹櫥褰曟垚鍔�");
+    }
+
     private MenuNode covertMenuNode(Menu menu, List<Menu> menuList) {
         MenuNode node = new MenuNode();
         BeanUtils.copyProperties(menu, node);
@@ -165,35 +195,40 @@
         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) {
-            wrapper.like(Admin::getNickName, keyword)
-                    .or(adminLambdaQueryWrapper -> {
-                        adminLambdaQueryWrapper.like(Admin::getUsername, keyword);
-                    });
-        }
-        List<Admin> list = adminService.list(wrapper);
+        return ResponseUtil.successPage(adminService.listAdmin(baseConditionVO, keyword));
 
-        List<AdminDto> adminDtoList = new ArrayList<>();
-        for (Admin demo : list) {
-            AdminDto adminDto = new AdminDto();
-            BeanUtil.copyProperties(demo, adminDto);
-            AdminRoleRelation one = adminRoleRelationService.getOne(Wrappers.lambdaQuery(AdminRoleRelation.class)
-                    .eq(AdminRoleRelation::getAdminId, demo.getId()));
-
-            if (one != null) {
-                adminDto.setRoleId(one.getRoleId());
-            }
-            adminDtoList.add(adminDto);
-
-        }
-        return ResponseUtil.successPage(adminDtoList);
+//        PageHelper.startPage(baseConditionVO.getPageNo(), baseConditionVO.getPageSize());
+//        LambdaQueryWrapper<Admin> wrapper = Wrappers.lambdaQuery(Admin.class);
+//        if (keyword != null) {
+//            wrapper.like(Admin::getNickName, keyword)
+//                    .or(adminLambdaQueryWrapper -> {
+//                        adminLambdaQueryWrapper.like(Admin::getUsername, keyword);
+//                    });
+//        }
+//        List<Admin> list = adminService.list(wrapper);
+//
+//        List<AdminDto> adminDtoList = new ArrayList<>();
+//        for (Admin demo : list) {
+//            AdminDto adminDto = new AdminDto();
+//            BeanUtil.copyProperties(demo, adminDto);
+//            AdminRoleRelation one = adminRoleRelationService.getOne(Wrappers.lambdaQuery(AdminRoleRelation.class)
+//                    .eq(AdminRoleRelation::getAdminId, demo.getId()));
+//
+//            if (one != null) {
+//                adminDto.setRoleId(one.getRoleId());
+//            }
+//            adminDtoList.add(adminDto);
+//
+//        }
+//        return ResponseUtil.successPage(adminDtoList);
     }
 
     @Log("鍚庡彴鐢ㄦ埛娉ㄥ唽")
     @PostMapping("/register")
     public ResponseVO<Object> register(@Validated @RequestBody AdminParam param) {
+        if (!permissionConfig.check(MenuEnum.ADMIN_ADD.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         boolean result = adminService.register(param);
         if (result) {
             return ResponseUtil.success("娉ㄥ唽鐢ㄦ埛鎴愬姛");
@@ -210,6 +245,9 @@
     @Log("鍚庡彴鐢ㄦ埛淇敼淇℃伅")
     @PostMapping("/update/{adminId}")
     public ResponseVO<Object> updateAdmin(@RequestBody AdminParam param, @PathVariable Long adminId) {
+        if (!permissionConfig.check(MenuEnum.ADMIN_INFO_UPDATE.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         boolean result = adminService.updateAdmin(adminId, param);
         if (result) {
             return ResponseUtil.success("淇敼鎴愬姛");
@@ -221,10 +259,35 @@
     /**
      * 淇敼鎴戠殑瀵嗙爜
      */
-    @Log("鍚庡彴鐢ㄦ埛淇敼瀵嗙爜")
-    @PutMapping("/updateMyPassword")
+    @Log("淇敼褰撳墠鐢ㄦ埛瀵嗙爜")
+    @PostMapping("/updateMyPassword")
     public ResponseVO<Object> updateMyPassword(@Validated @RequestBody PwdParam param) {
-        boolean result = adminService.updateMyPassword(param);
+        boolean result;
+        if (SecurityUtils.getClientId() == null) {
+            result = adminService.updateMyPassword(param);
+        } else {
+            result = clientService.resetPassword(param);
+        }
+
+        if (result) {
+            return ResponseUtil.success("淇敼鎴愬姛");
+        } else {
+            return ResponseUtil.fail("淇敼澶辫触");
+        }
+    }
+
+    /**
+     * 淇敼绠$悊鍛樼殑瀵嗙爜
+     */
+    @Log("淇敼绠$悊鍛樼殑瀵嗙爜")
+    @PostMapping("/updateAdminPassword")
+    public ResponseVO<Object> updateAdminPassword(@Validated @RequestBody UserPwsParm param) {
+        if (!SecurityUtils.getUserDetails().getAdministratorType().equals(AdministratorEnums.ADMIN.getCode())) {
+            throw new BusinessException("闈炶秴绾х鐞嗗憳鏃犳硶淇敼瀵嗙爜锛�");
+        }
+
+        boolean result = adminService.updateAdminPassword(param);
+
         if (result) {
             return ResponseUtil.success("淇敼鎴愬姛");
         } else {
@@ -245,6 +308,9 @@
     @Log("鍚庡彴鐢ㄦ埛鍒犻櫎")
     @PostMapping("/del/{adminId}")
     public ResponseVO<Object> deleteAdmin(@PathVariable Long adminId) {
+        if (!permissionConfig.check(MenuEnum.ADMIN_DELETE.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
         boolean result = adminService.deleteAdmin(adminId);
         if (result) {
             return ResponseUtil.success("淇敼鎴愬姛");

--
Gitblit v1.9.3