From e714f2e3d41ae67a868be875028f62ac57811ede Mon Sep 17 00:00:00 2001
From: "yang.yongquan" <3395816735@qq.com>
Date: Wed, 6 Jul 2022 16:45:32 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E6=9F=A5=E8=AF=A2=E9=A1=B9?=
 =?UTF-8?q?=E7=9B=AE=E8=81=8C=E4=BD=8D=E7=9A=84=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../controller/ProjectGroupController.java    | 30 ++++++++++++++++---
 1 file changed, 26 insertions(+), 4 deletions(-)

diff --git a/src/main/java/cn/edu/hfut/rmdjzz/projectmanagement/controller/ProjectGroupController.java b/src/main/java/cn/edu/hfut/rmdjzz/projectmanagement/controller/ProjectGroupController.java
index 3141609..5519066 100644
--- a/src/main/java/cn/edu/hfut/rmdjzz/projectmanagement/controller/ProjectGroupController.java
+++ b/src/main/java/cn/edu/hfut/rmdjzz/projectmanagement/controller/ProjectGroupController.java
@@ -1,14 +1,16 @@
 package cn.edu.hfut.rmdjzz.projectmanagement.controller;
 
+import cn.edu.hfut.rmdjzz.projectmanagement.entity.ProjectGroup;
 import cn.edu.hfut.rmdjzz.projectmanagement.exception.BadRequestException;
+import cn.edu.hfut.rmdjzz.projectmanagement.exception.UnauthorizedException;
 import cn.edu.hfut.rmdjzz.projectmanagement.service.IProjectGroupService;
+import cn.edu.hfut.rmdjzz.projectmanagement.utils.TokenUtils;
 import cn.edu.hfut.rmdjzz.projectmanagement.utils.http.ResponseList;
+import cn.edu.hfut.rmdjzz.projectmanagement.utils.http.ResponseMap;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import lombok.SneakyThrows;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.PathVariable;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.bind.annotation.*;
 
 import java.util.List;
 
@@ -29,4 +31,24 @@ public class ProjectGroupController {
         List<Integer> res = projectGroupService.findAllProjectNumber(projectId);
         return ResponseList.ofSuccess("查询成功", res);
     }
+
+    @SneakyThrows
+    @GetMapping("/{staffId}")
+    public ResponseMap getStaffPostion(
+            @RequestHeader("Token") String token,
+            @PathVariable Integer staffId,
+            @PathVariable Integer projectId
+    ) {
+        if (TokenUtils.getStaffId(token) != staffId) {
+            throw new BadRequestException("用户访问错误");
+        }
+        if (projectGroupService.getUserLevelInGroup(token, projectId) == 0) {
+            throw new UnauthorizedException("无该项目访问权限");
+        }
+        return ResponseMap.ofSuccess("查询成功", projectGroupService.getOne(
+                Wrappers.<ProjectGroup>lambdaQuery()
+                        .eq(ProjectGroup::getStaffId, staffId)
+                        .eq(ProjectGroup::getProjectId, projectId)
+        ));
+    }
 }