From fbc1c1590e7e81a550beb50c7cf65c663c7c5b41 Mon Sep 17 00:00:00 2001
From: zhizhijie <zhizhijie@users.noreply.gitee.com>
Date: 星期二, 01 九月 2026 16:42:06 +0800
Subject: [PATCH] 数据导入与报表生成多项改进

---
 traffic-audit-server/src/main/java/com/trafficaudit/security/controller/LoginController.java |   70 ++++++++++++++++++++++++++++++----
 1 files changed, 61 insertions(+), 9 deletions(-)

diff --git a/traffic-audit-server/src/main/java/com/trafficaudit/security/controller/LoginController.java b/traffic-audit-server/src/main/java/com/trafficaudit/security/controller/LoginController.java
index 92b43c1..18572ae 100644
--- a/traffic-audit-server/src/main/java/com/trafficaudit/security/controller/LoginController.java
+++ b/traffic-audit-server/src/main/java/com/trafficaudit/security/controller/LoginController.java
@@ -9,6 +9,7 @@
 import org.springframework.security.crypto.password.PasswordEncoder;
 import org.springframework.web.bind.annotation.*;
 
+import org.springframework.beans.factory.annotation.Value;
 import javax.annotation.Resource;
 import java.util.Collections;
 import java.util.HashMap;
@@ -31,6 +32,9 @@
     @Resource
     private PasswordEncoder passwordEncoder;
 
+    @Value("${app.captcha.required:true}")
+    private boolean captchaRequired;
+
     @PostMapping("/login")
     public Result<Map<String, Object>> login(@RequestBody Map<String, Object> params) {
         String username = (String) params.get("username");
@@ -38,15 +42,17 @@
         String captchaId = (String) params.get("captchaId");
         List<String> clickIds = strList(params.get("captchaClickIds"));
 
-        CaptchaService.ConsumeResult cr = captchaService.consume(captchaId, clickIds);
-        if (cr == CaptchaService.ConsumeResult.NOT_FOUND) {
-            return Result.error(4002, "楠岃瘉鐮佷笉瀛樺湪鎴栧凡杩囨湡锛岃鍒锋柊閲嶈瘯");
-        }
-        if (cr == CaptchaService.ConsumeResult.NOT_VERIFIED) {
-            return Result.error(4003, "璇峰厛瀹屾垚琛屼负楠岃瘉");
-        }
-        if (cr == CaptchaService.ConsumeResult.WRONG) {
-            return Result.error(4001, "琛屼负楠岃瘉鏈�氳繃锛岃閲嶆柊楠岃瘉");
+        if (captchaRequired) {
+            CaptchaService.ConsumeResult cr = captchaService.consume(captchaId, clickIds);
+            if (cr == CaptchaService.ConsumeResult.NOT_FOUND) {
+                return Result.error(4002, "楠岃瘉鐮佷笉瀛樺湪鎴栧凡杩囨湡锛岃鍒锋柊閲嶈瘯");
+            }
+            if (cr == CaptchaService.ConsumeResult.NOT_VERIFIED) {
+                return Result.error(4003, "璇峰厛瀹屾垚琛屼负楠岃瘉");
+            }
+            if (cr == CaptchaService.ConsumeResult.WRONG) {
+                return Result.error(4001, "琛屼负楠岃瘉鏈�氳繃锛岃閲嶆柊楠岃瘉");
+            }
         }
 
         if (username == null || username.trim().isEmpty() || password == null || password.isEmpty()) {
@@ -77,6 +83,52 @@
         return Result.ok(data);
     }
 
+    /** 鐢ㄦ埛淇敼鑷繁鐨勫瘑鐮侊細闇�鎻愪緵鍘熷瘑鐮侊紝鏂板瘑鐮佽嚦灏� 6 浣� */
+    @PostMapping("/change-password")
+    public Result<Void> changePassword(@RequestHeader(value = "Authorization", required = false) String auth,
+                                       @RequestBody Map<String, Object> params) {
+        String username = jwtUtils.resolveUsername(auth);
+        if (username == null) {
+            return Result.error(401, "鏈櫥褰曟垨鐧诲綍宸茶繃鏈�");
+        }
+        String oldPassword = (String) params.get("oldPassword");
+        String newPassword = (String) params.get("newPassword");
+        if (oldPassword == null || oldPassword.isEmpty()) {
+            return Result.error(400, "璇疯緭鍏ュ師瀵嗙爜");
+        }
+        if (newPassword == null || newPassword.length() < 6) {
+            return Result.error(400, "鏂板瘑鐮侀暱搴︿笉鑳藉皯浜� 6 浣�");
+        }
+        if (newPassword.equals(oldPassword)) {
+            return Result.error(400, "鏂板瘑鐮佷笉鑳戒笌鍘熷瘑鐮佺浉鍚�");
+        }
+        User user = userMapper.selectOne(new LambdaQueryWrapper<User>().eq(User::getUsername, username));
+        if (user == null) {
+            return Result.error(404, "鐢ㄦ埛涓嶅瓨鍦�");
+        }
+        String encoded = user.getPassword();
+        boolean ok;
+        try {
+            ok = encoded != null && passwordEncoder.matches(oldPassword, encoded);
+        } catch (IllegalArgumentException e) {
+            // 搴撲腑瀵嗙爜涓嶆槸鍚堟硶 BCrypt 鏃跺洖閫�涓烘槑鏂囧姣旓紙涓庣櫥褰曚竴鑷达級
+            ok = encoded != null && encoded.equals(oldPassword);
+        }
+        if (!ok) {
+            return Result.error(400, "鍘熷瘑鐮侀敊璇�");
+        }
+        user.setPassword(passwordEncoder.encode(newPassword));
+        userMapper.updateById(user);
+        return Result.ok(null);
+    }
+
+    @GetMapping("/captcha-config")
+    public Result<Map<String, Object>> captchaConfig() {
+        Map<String, Object> data = new HashMap<>();
+        data.put("required", captchaRequired);
+        return Result.ok(data);
+    }
+
     @SuppressWarnings("unchecked")
     private List<String> strList(Object o) {
         if (o instanceof List) {

--
Gitblit v1.9.1