From d170546ea4498d282c9cfef6258e4498aed7835e Mon Sep 17 00:00:00 2001
From: zhanzhiqin <895896009@qq.com>
Date: 星期五, 22 四月 2022 15:57:22 +0800
Subject: [PATCH] fix

---
 ximon-admin/src/main/java/com/sandu/ximon/admin/controller/ClientController.java |   60 ++++++++++++++++++++++++++++++++++++++++++++++--------------
 1 files changed, 46 insertions(+), 14 deletions(-)

diff --git a/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/ClientController.java b/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/ClientController.java
index efe7faf..287affc 100644
--- a/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/ClientController.java
+++ b/ximon-admin/src/main/java/com/sandu/ximon/admin/controller/ClientController.java
@@ -1,10 +1,10 @@
 package com.sandu.ximon.admin.controller;
 
+import cn.hutool.core.bean.BeanUtil;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 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.object.BaseConditionVO;
 import com.sandu.common.security.LoginUserInfo;
@@ -13,17 +13,20 @@
 import com.sandu.common.security.token.TokenProvider;
 import com.sandu.common.util.IpUtil;
 import com.sandu.common.util.ResponseUtil;
-import com.sandu.ximon.admin.context.CustomerInspectContextHolder;
+import com.sandu.ximon.admin.dto.ClientDto;
 import com.sandu.ximon.admin.param.AddClientPrarm;
 import com.sandu.ximon.admin.param.AdminLoginParam;
-import com.sandu.ximon.admin.param.ClientPrarm;
+import com.sandu.ximon.admin.param.ResetClientPasswordPrarm;
+import com.sandu.ximon.admin.param.UpdateClientPrarm;
+import com.sandu.ximon.admin.security.PermissionConfig;
 import com.sandu.ximon.admin.security.SecurityUtils;
+import com.sandu.ximon.admin.service.ClientRoleRelationService;
 import com.sandu.ximon.admin.service.ClientService;
 import com.sandu.ximon.admin.service.RoleService;
-import com.sandu.ximon.dao.domain.Admin;
 import com.sandu.ximon.dao.domain.Client;
+import com.sandu.ximon.dao.domain.ClientRoleRelation;
 import com.sandu.ximon.dao.enums.AdministratorEnums;
-import jodd.cli.Cli;
+import com.sandu.ximon.dao.enums.MenuEnum;
 import lombok.AllArgsConstructor;
 import org.springframework.security.core.GrantedAuthority;
 import org.springframework.security.crypto.password.PasswordEncoder;
@@ -31,9 +34,7 @@
 import org.springframework.web.bind.annotation.*;
 
 import java.time.LocalDateTime;
-import java.util.Collection;
-import java.util.HashMap;
-import java.util.Map;
+import java.util.*;
 import java.util.stream.Collectors;
 
 /**
@@ -48,7 +49,9 @@
     private final ClientService clientService;
     private final PasswordEncoder passwordEncoder;
     private final RoleService roleService;
+    private final ClientRoleRelationService clientRoleRelationService;
     private final TokenProvider tokenProvider;
+    private PermissionConfig permissionConfig;
 
     @PostMapping("/add")
     public ResponseVO<Object> addClient(@RequestBody @Validated AddClientPrarm addClientPrarm) {
@@ -62,12 +65,22 @@
 
 
     @PostMapping("/update/{id}")
-    public ResponseVO<Object> updateClient(@PathVariable Long id, @RequestBody @Validated ClientPrarm clientPrarm) {
-        boolean b = clientService.updateClient(id, clientPrarm);
+    public ResponseVO<Object> updateClient(@PathVariable Long id, @RequestBody @Validated UpdateClientPrarm updateClientPrarm) {
+        boolean b = clientService.updateClient(id, updateClientPrarm);
         if (b) {
             return ResponseUtil.success("鏇存柊鎴愬姛");
         } else {
             return ResponseUtil.fail("鏇存柊澶辫触");
+        }
+    }
+
+    @PostMapping("/resetClientPassword")
+    public ResponseVO<Object> resetClientPassword(@RequestBody @Validated ResetClientPasswordPrarm resetClientPasswordPrarm) {
+        boolean b = clientService.resetPassword(resetClientPasswordPrarm);
+        if (b) {
+            return ResponseUtil.success("淇敼鎴愬姛");
+        } else {
+            return ResponseUtil.fail("淇敼澶辫触");
         }
     }
 
@@ -83,17 +96,36 @@
 
     @GetMapping("/list")
     public ResponseVO<Object> listLikeClient(BaseConditionVO baseConditionVO, @RequestParam(value = "keyword", required = false) String keyword) {
+        if (!permissionConfig.check(MenuEnum.CLIENT_LIST.getCode())) {
+            return ResponseUtil.fail("缂哄皯瀵瑰簲鐢ㄦ埛鏉冮檺");
+        }
+
         PageHelper.startPage(baseConditionVO.getPageNo(), baseConditionVO.getPageSize());
         LambdaQueryWrapper<Client> wrapper = Wrappers.lambdaQuery(Client.class);
         if (AdministratorEnums.CUSTOMER.getCode().equals(SecurityUtils.getAdministratorIdentity())) {
             wrapper.eq(Client::getSuperiorId, SecurityUtils.getUserId());
         }
-        if (null != keyword) {
+        if (keyword != null && !keyword.isEmpty()) {
             wrapper.like(Client::getClientName, keyword)
                     .or(clientLambdaQueryWrapper -> clientLambdaQueryWrapper.like(Client::getMobile, keyword))
                     .or(clientLambdaQueryWrapper -> clientLambdaQueryWrapper.like(Client::getLinkMan, keyword));
         }
-        return ResponseUtil.successPage(clientService.list(wrapper));
+        List<Client> clientList = clientService.list(wrapper);
+        List<ClientDto> clientDtoList = new ArrayList<>();
+
+        for (Client demo : clientList) {
+            ClientDto clientDto = new ClientDto();
+            BeanUtil.copyProperties(demo, clientDto);
+            ClientRoleRelation one = clientRoleRelationService.getOne(Wrappers.lambdaQuery(ClientRoleRelation.class)
+                    .eq(ClientRoleRelation::getClientId, demo.getId()));
+
+            if (one != null) {
+                clientDto.setRoleId(one.getRoleId());
+            }
+            clientDtoList.add(clientDto);
+
+        }
+        return ResponseUtil.successPage(clientDtoList);
     }
 
     @AnonymousAccess
@@ -107,14 +139,14 @@
             return ResponseUtil.error(ResponseStatusEnums.BAD_CREDENTIALS.getCode(), ResponseStatusEnums.BAD_CREDENTIALS.getMessage());
         }
 
-//        Collection<GrantedAuthority> grantedAuthorities = roleService.mapToGrantedAuthorities(client.getId());
+        Collection<GrantedAuthority> grantedAuthorities = roleService.mapToGrantedAuthorities(client.getId(), false);
 
         LoginUserInfo loginUserInfo = new LoginUserInfo();
         loginUserInfo.setUserId(client.getId());
         loginUserInfo.setAccount(client.getClientName());
         loginUserInfo.setIp(IpUtil.getRealIp());
         loginUserInfo.setAdministratorType(AdministratorEnums.NORMAL.getCode());
-//        loginUserInfo.setPermission(grantedAuthorities.stream().map(GrantedAuthority::getAuthority).collect(Collectors.joining(",")));
+        loginUserInfo.setPermission(grantedAuthorities.stream().map(GrantedAuthority::getAuthority).collect(Collectors.joining(",")));
 
         // 鐢熸垚浠ょ墝
         String token = tokenProvider.createToken(loginUserInfo);

--
Gitblit v1.9.3