From 09ee5added9d59e90310a2586e846137ea597b19 Mon Sep 17 00:00:00 2001
From: LSH
Date: 星期五, 01 十二月 2023 13:15:42 +0800
Subject: [PATCH] #

---
 src/main/java/com/zy/crm/common/web/BaseController.java |   62 +++++++++++++++++++++++++++++++
 1 files changed, 62 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/zy/crm/common/web/BaseController.java b/src/main/java/com/zy/crm/common/web/BaseController.java
index ca4ec88..8855433 100644
--- a/src/main/java/com/zy/crm/common/web/BaseController.java
+++ b/src/main/java/com/zy/crm/common/web/BaseController.java
@@ -8,7 +8,9 @@
 import com.core.controller.AbstractBaseController;
 import com.core.exception.CoolException;
 import com.zy.crm.system.entity.Dept;
+import com.zy.crm.system.entity.Role;
 import com.zy.crm.system.service.DeptService;
+import com.zy.crm.system.service.RoleService;
 import io.swagger.annotations.ApiModelProperty;
 import org.springframework.beans.factory.annotation.Autowired;
 import com.zy.crm.manager.entity.Node;
@@ -44,6 +46,8 @@
     private DeptService deptService;
     @Autowired
     private UserLoginService userLoginService;
+    @Autowired
+    private RoleService roleService;
 
     protected Long getHostId(){
         if (getUserId() == 9527) {
@@ -76,6 +80,11 @@
         return Long.parseLong(String.valueOf(request.getAttribute("userId")));
     }
 
+    protected Long getUserId(String token){
+        UserLogin userLogin = userLoginService.selectOne(new EntityWrapper<UserLogin>().eq("token", token));
+        return userLogin.getUserId();
+    }
+
     protected User getUser(){
         User user = userService.selectById(getUserId());
         if (null == user) {
@@ -84,8 +93,24 @@
         return user;
     }
 
+    protected User getUser(String token){
+        User user = userService.selectById(getUserId(token));
+        if (null == user) {
+            throw new CoolException(BaseRes.DENIED);
+        }
+        return user;
+    }
+
     protected Long getDeptId() {
         return getUser().getDeptId();
+    }
+
+    protected Role getRole(){
+        Role role = roleService.selectById(getUser().getRoleId());
+        if (null == role) {
+            throw new CoolException(BaseRes.DENIED);
+        }
+        return role;
     }
 
     protected String getComment(Class<?> cls, String fieldName){
@@ -211,4 +236,41 @@
             wrapper.like(columns.get(i), condition);
         }
     }
+
+    //鑾峰彇褰撳墠鐧诲綍鐢ㄦ埛锛屽搴旀潈闄愪笅鎵�鏈夌敤鎴穒d銆傦紙鏅�氳亴鍛樿幏鍙栬嚜宸憋紝棰嗗鑾峰彇鑷繁鍙婁笅灞烇紝瓒呯骇绠＄悊鍛樺叏閮級
+    protected ArrayList<Long> getUserRoleBelongsToUserId() {
+        User user = getUser();
+        ArrayList<Long> userIds = new ArrayList<>();
+        if (user.getRoleCode().equals("boss")) {
+            //鍏ㄩ儴鏉冮檺
+            userIds = null;
+        } else if (user.getRoleCode().equals("manager")) {
+            //鑷繁鍙婁笅灞炴潈闄�
+            userIds.add(getUserId());
+            EntityWrapper<User> entityWrapper = new EntityWrapper<>();
+            entityWrapper.eq("dept_id", user.getDeptId());
+            List<User> userList = userService.selectList(entityWrapper);
+            for (User item : userList) {
+                userIds.add(item.getId());
+            }
+        }else{
+            //鑷繁鏉冮檺
+            userIds.add(getUserId());
+        }
+
+        return userIds;
+    }
+
+    protected ArrayList<Long> getUserRoleBelongsToUserId(String permission) {
+        User user = getUser();
+        if (user.getDeptName().equals("骞冲簱瑙勫垝缁�") || user.getDeptName().equals("鏍镐环缁�")) {
+            if (permission.equals("allopen")) {
+                //鏉冮檺鍏ㄩ儴寮�鏀�
+                return null;
+            }
+        }
+
+        //杩斿洖榛樿閫夋嫨鏂规
+        return getUserRoleBelongsToUserId();
+    }
 }

--
Gitblit v1.9.1