添加验证码图片和图片路径

src/main/java/org/cmh/backend/authentication/controller/AuthenticationController.java

@@ -6,20 +6,21 @@ import org.cmh.backend.Utils.JwtVerify;
 import org.cmh.backend.authentication.dto.*;
 import org.cmh.backend.authentication.model.UserHS;
 import org.cmh.backend.authentication.service.UserService;
+import org.cmh.backend.authentication.service.VerificationCodeService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.*;
 
-import java.util.HashMap;
-import java.util.Map;
-
 @RestController
 class AuthenticationController {
 
     @Autowired
     private UserService userService;
 
+    @Autowired
+    private VerificationCodeService verificationCodeService;
+
     @GetMapping("/hello")
     public String hello() {
         return "Hello SpringBoot!";
@@ -59,8 +60,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 +69,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);
@@ -95,18 +100,9 @@ class AuthenticationController {
         return new ResponseEntity<>(HttpStatus.BAD_REQUEST);
     }
 
-    @PostMapping("/sendVerificationCode")
-    public ResponseEntity<Map<String, Integer>> sendVerificationCode(@RequestBody Map<String, String> request) {
-        String contactInfo = request.get("contactInfo");
-//      TODO:发送验证码功能有待添加。  boolean isSent = userService.sendVerificationCode(contactInfo);
-        boolean isSent = true;
-        if (isSent) {
-            HashMap<String, Integer> response = new HashMap<>();
-            response.put("code", 1234);
-            return new ResponseEntity<>(response, HttpStatus.OK);
-        } else {
-            return new ResponseEntity<>(HttpStatus.INTERNAL_SERVER_ERROR);
-        }
+    @GetMapping("/getVerificationCode")
+    public ResponseEntity<VerificationCodeResponse> getVerificationCode() {
+        return new ResponseEntity<>(verificationCodeService.provideVerificationCode(), HttpStatus.OK);
     }
 
 

src/main/java/org/cmh/backend/authentication/dto/VerificationCodeResponse.java

@@ -0,0 +1,13 @@
+package org.cmh.backend.authentication.dto;
+
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+import lombok.Setter;
+
+@Getter
+@Setter
+@AllArgsConstructor
+public class VerificationCodeResponse {
+    private String code;
+    private String path;
+}

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;
 }

src/main/java/org/cmh/backend/authentication/service/VerificationCodeService.java

@@ -0,0 +1,51 @@
+package org.cmh.backend.authentication.service;
+
+import jakarta.annotation.PostConstruct;
+import lombok.Getter;
+import org.cmh.backend.authentication.dto.VerificationCodeResponse;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Service;
+
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Random;
+import java.util.stream.Stream;
+
+@Service
+public class VerificationCodeService {
+
+    @Value("${verification.code.images.path}")
+    private String verificationCodeImagesPath;
+
+    @Getter
+    private List<String> verificationCodeList = new ArrayList<>();
+
+    @PostConstruct
+    public void init() {
+        loadVerificationCodeImages();
+    }
+
+    private void loadVerificationCodeImages() {
+        try (Stream<Path> paths = Files.list(Paths.get(verificationCodeImagesPath))) {
+            paths.filter(path -> path.toString().endsWith(".png"))
+                    .forEach(path -> verificationCodeList.add(path.getFileName().toString()));
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+    }
+
+    public VerificationCodeResponse provideVerificationCode() {
+        if (verificationCodeList.isEmpty()) {
+            return null;
+        }
+        Random random = new Random();
+        int code = random.nextInt(random.nextInt(verificationCodeList.size()));
+        String selectedCode = verificationCodeList.get(code);
+        return new VerificationCodeResponse(selectedCode.split("\\.")[0], "/verificationCodeImages/" + selectedCode);
+    }
+
+}

src/main/resources/application.properties

@@ -20,5 +20,7 @@ spring.datasource.hikari.connection-timeout=30000
 server.servlet.encoding.enabled=true
 server.servlet.encoding.force=true
 server.servlet.encoding.charset=utf-8
+# verificationCode
+verification.code.images.path=src/main/resources/static/verificationCodeImages