From b7ede5652c724de351df457e57f413c4cdd89b09 Mon Sep 17 00:00:00 2001
From: heshunme <heshunme@outlook.com>
Date: Tue, 2 Jul 2024 15:41:05 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E7=89=B9=E6=9D=83=E7=94=A8?=
 =?UTF-8?q?=E6=88=B7:superAdmin?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../controller/AuthenticationController.java           | 10 +++++++---
 .../org/cmh/backend/authentication/model/UserHS.java   |  1 +
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/main/java/org/cmh/backend/authentication/controller/AuthenticationController.java b/src/main/java/org/cmh/backend/authentication/controller/AuthenticationController.java
index 8d2fdb4..e58b5f1 100644
--- a/src/main/java/org/cmh/backend/authentication/controller/AuthenticationController.java
+++ b/src/main/java/org/cmh/backend/authentication/controller/AuthenticationController.java
@@ -59,8 +59,7 @@ class AuthenticationController {
         if (JwtUtil.isTokenValid(token)) {
             UserHS user = userService.getUserByUsername(JwtUtil.extractUsername(token));
             if (user != null) {
-
-                return new ResponseEntity<>(new UserProfileResponse(
+                UserProfileResponse response = new UserProfileResponse(
                         user.getUsername(),
                         user.getNickname(),
                         user.getGender(),
@@ -69,7 +68,12 @@ class AuthenticationController {
                         user.getDepartment(),
                         user.getRole(),
                         user.getCreatedAt()
-                ), HttpStatus.OK);
+                );
+                if (user.getSuperAdmin()) {
+                    response.setDepartment("超级管理员");
+                    response.setRole("超级管理员");
+                }
+                return new ResponseEntity<>(response, HttpStatus.OK);
             }
         }
         return new ResponseEntity<>(HttpStatus.NOT_FOUND);
diff --git a/src/main/java/org/cmh/backend/authentication/model/UserHS.java b/src/main/java/org/cmh/backend/authentication/model/UserHS.java
index 995186d..0b86a00 100644
--- a/src/main/java/org/cmh/backend/authentication/model/UserHS.java
+++ b/src/main/java/org/cmh/backend/authentication/model/UserHS.java
@@ -25,4 +25,5 @@ public class UserHS {
     private String department;
     private String role;
     private LocalDateTime createdAt;
+    private Boolean superAdmin = false;
 }