From df05fc5ccd9302ccca1e92c2c8ca24c272a658e4 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期三, 13 十一月 2024 16:41:56 +0800
Subject: [PATCH] 1

---
 se-auth/src/main/java/com/se/auth/controller/TokenController.java |   52 ++++++++++++++++++++++++++++++++++------------------
 1 files changed, 34 insertions(+), 18 deletions(-)

diff --git a/se-auth/src/main/java/com/se/auth/controller/TokenController.java b/se-auth/src/main/java/com/se/auth/controller/TokenController.java
index 843cd7e..3f34d1f 100644
--- a/se-auth/src/main/java/com/se/auth/controller/TokenController.java
+++ b/se-auth/src/main/java/com/se/auth/controller/TokenController.java
@@ -5,11 +5,10 @@
 import com.se.auth.form.LoginBody;
 import com.se.auth.form.RegisterBody;
 import com.se.auth.service.SysLoginService;
+import com.se.common.core.utils.AesUtils;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.DeleteMapping;
-import org.springframework.web.bind.annotation.PostMapping;
-import org.springframework.web.bind.annotation.RequestBody;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.web.bind.annotation.*;
 import com.se.common.core.domain.R;
 import com.se.common.core.utils.JwtUtils;
 import com.se.common.core.utils.StringUtils;
@@ -24,29 +23,48 @@
  * @author admin
  */
 @RestController
-public class TokenController
-{
+public class TokenController {
     @Autowired
     private TokenService tokenService;
 
     @Autowired
     private SysLoginService sysLoginService;
 
+    @Value("${enableEncrypt}")
+    boolean enableEncrypt;
+
     @PostMapping("login")
-    public R<?> login(@RequestBody LoginBody form)
-    {
+    public R<?> login(@RequestBody LoginBody form) throws Exception {
+        if (enableEncrypt && !StringUtils.isEmpty(form.getPassword())) {
+            form.setPassword(AesUtils.decrypt(form.getPassword()));
+        }
         // 鐢ㄦ埛鐧诲綍
         LoginUser userInfo = sysLoginService.login(form.getUsername(), form.getPassword());
         // 鑾峰彇鐧诲綍token
         return R.ok(tokenService.createToken(userInfo));
     }
 
+    @GetMapping("validate")
+    @PostMapping("validate")
+    public R<Object> validate(HttpServletRequest request) {
+        try {
+            boolean flag = false;
+            String token = SecurityUtils.getToken(request);
+            if (!StringUtils.isNotEmpty(token)) {
+                String userName = JwtUtils.getUserName(token);
+                flag = !StringUtils.isEmpty(userName);
+            }
+
+            return R.ok(flag);
+        } catch (Exception ex) {
+            return R.fail(ex.getMessage());
+        }
+    }
+
     @DeleteMapping("logout")
-    public R<?> logout(HttpServletRequest request)
-    {
+    public R<?> logout(HttpServletRequest request) {
         String token = SecurityUtils.getToken(request);
-        if (StringUtils.isNotEmpty(token))
-        {
+        if (StringUtils.isNotEmpty(token)) {
             String username = JwtUtils.getUserName(token);
             // 鍒犻櫎鐢ㄦ埛缂撳瓨璁板綍
             AuthUtil.logoutByToken(token);
@@ -57,11 +75,9 @@
     }
 
     @PostMapping("refresh")
-    public R<?> refresh(HttpServletRequest request)
-    {
+    public R<?> refresh(HttpServletRequest request) {
         LoginUser loginUser = tokenService.getLoginUser(request);
-        if (StringUtils.isNotNull(loginUser))
-        {
+        if (StringUtils.isNotNull(loginUser)) {
             // 鍒锋柊浠ょ墝鏈夋晥鏈�
             tokenService.refreshToken(loginUser);
             return R.ok();
@@ -69,11 +85,11 @@
         return R.ok();
     }
 
-    @PostMapping("register")
+    /*@PostMapping("register")
     public R<?> register(@RequestBody RegisterBody registerBody)
     {
         // 鐢ㄦ埛娉ㄥ唽
         sysLoginService.register(registerBody.getUsername(), registerBody.getPassword());
         return R.ok();
-    }
+    }*/
 }

--
Gitblit v1.9.3