From 9180185b46daf03b63d658d3f1168903e5114b61 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@gmail.com>
Date: 星期三, 12 二月 2025 20:11:21 +0800
Subject: [PATCH] #

---
 rsf-server/src/main/java/com/vincent/rsf/server/system/controller/AuthController.java |   61 +++++++++++++++++++++---------
 1 files changed, 43 insertions(+), 18 deletions(-)

diff --git a/rsf-server/src/main/java/com/vincent/rsf/server/system/controller/AuthController.java b/rsf-server/src/main/java/com/vincent/rsf/server/system/controller/AuthController.java
index 383a5b0..01afd24 100644
--- a/rsf-server/src/main/java/com/vincent/rsf/server/system/controller/AuthController.java
+++ b/rsf-server/src/main/java/com/vincent/rsf/server/system/controller/AuthController.java
@@ -1,20 +1,28 @@
 package com.vincent.rsf.server.system.controller;
 
+import com.vincent.rsf.common.enums.SystemModeType;
 import com.vincent.rsf.common.utils.Utils;
 import com.vincent.rsf.framework.common.Cools;
 import com.vincent.rsf.framework.common.R;
 import com.vincent.rsf.server.common.annotation.OperationLog;
 import com.vincent.rsf.server.common.config.ConfigProperties;
 import com.vincent.rsf.server.common.security.JwtSubject;
+import com.vincent.rsf.server.common.service.EmailService;
 import com.vincent.rsf.server.common.utils.JwtUtil;
 import com.vincent.rsf.server.system.controller.param.LoginParam;
 import com.vincent.rsf.server.system.controller.param.UpdatePasswordParam;
 import com.vincent.rsf.server.system.controller.result.LoginResult;
 import com.vincent.rsf.server.system.controller.result.MenuVo;
+import com.vincent.rsf.server.system.controller.result.SystemInfoVo;
 import com.vincent.rsf.server.system.entity.Menu;
+import com.vincent.rsf.server.system.entity.Tenant;
 import com.vincent.rsf.server.system.entity.User;
 import com.vincent.rsf.server.system.entity.UserLogin;
-import com.vincent.rsf.server.system.service.*;
+import com.vincent.rsf.server.system.enums.StatusType;
+import com.vincent.rsf.server.system.service.RoleMenuService;
+import com.vincent.rsf.server.system.service.TenantService;
+import com.vincent.rsf.server.system.service.UserLoginService;
+import com.vincent.rsf.server.system.service.UserService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;
@@ -32,7 +40,6 @@
  * Created by vincent on 1/30/2024
  */
 @RestController
-//@RequestMapping("")
 public class AuthController extends BaseController {
 
     @Resource
@@ -46,27 +53,40 @@
     @Resource
     private TenantService tenantService;
     @Autowired
-    private UserRoleService userRoleService;
+    private EmailService emailService;
 
     @PostMapping("/login")
     public R login(@RequestBody LoginParam param, HttpServletRequest request) {
 //        System.out.println(userService.encodePassword("123456"));
+//        emailService.sendVerificationEmail("t1341870251@gmail.com", "123456");
+
         String username = param.getUsername();
         Long tenantId = param.getTenantId();
         User user = userService.getByUsername(username, tenantId);
         if (user == null) {
             return R.error("Username Does Not Exist");
         }
-        if (!user.getStatus().equals(1)) {
+        if (!user.getStatus().equals(StatusType.ENABLE.val)) {
             return R.error("Account Frozen");
         }
         if (!userService.comparePassword(user.getPassword(), param.getPassword())) {
-            return R.error("Wrong Password");
+            return R.error("Invalid Password");
         }
         String accessToken = JwtUtil.buildToken(new JwtSubject(username, user.getTenantId()),
                 configProperties.getTokenExpireTime(), configProperties.getTokenKey());
         userLoginService.saveAsync(user.getId(), accessToken, UserLogin.TYPE_LOGIN, tenantId, null, request);
-        return R.ok("Login Success").add(new LoginResult(accessToken, user));
+        Tenant tenant = tenantService.getById(user.getTenantId());
+        return R.ok("Sign In Success").add(new LoginResult(accessToken, user, tenant.getName()));
+    }
+
+    @GetMapping("/system/info")
+    public R systemInfo() {
+        SystemModeType systemMode = configProperties.getSystemMode();
+        SystemInfoVo infoVo = new SystemInfoVo();
+        infoVo.setName(configProperties.getSystemName());
+        infoVo.setVersion(configProperties.getSystemVersion());
+        infoVo.setMode(systemMode.toString());
+        return R.ok(infoVo);
     }
 
     @GetMapping("/tenant/list")
@@ -76,16 +96,21 @@
 
     @GetMapping("/auth/user")
     public R userInfo() {
-        User user = userService.getByIdRel(getLoginUserId());
-        user.setRoles(userRoleService.listByUserId(user.getId()));
-        return R.ok(user);
+        User user = userService.getById(getLoginUserId());
+        return R.ok(userService.setUserAuthInfo(user));
     }
 
     @GetMapping("/auth/menu")
     public R userMenu() {
         List<Menu> menus = roleMenuService.listMenuByUserId(getLoginUserId(), Menu.TYPE_MENU);
-//        return R.ok().add(Utils.toTreeData(menus, 0L, Menu::getParentId, Menu::getId, Menu::setChildren));
         List<MenuVo> voList = menus.stream().map(this::convertToVo).collect(Collectors.toList());
+        // exclude tenant
+        if (!configProperties.getSuperUserList().contains(getLoginUser().getUsername())) {
+            voList =  voList.stream().filter(vo ->
+                    !vo.getName().equals("menu.tenant")
+                            && !vo.getName().equals("menu.menu")
+            ).collect(Collectors.toList());
+        }
         return R.ok().add(Utils.toTreeData(voList, 0L, MenuVo::getParentId, MenuVo::getId, MenuVo::setChildren));
     }
 
@@ -100,15 +125,15 @@
         user.setTenantId(null);
         user.setStatus(null);
         if (userService.updateById(user)) {
-            return R.ok("Save Success").add(userService.getByIdRel(user.getId()));
+            return R.ok("Save Success").add(userService.getById(user.getId()));
         }
         return R.error("Save Fail");
     }
 
     @OperationLog("Reset Password")
-    @PostMapping("/auth/password")
-    public R updatePassword(@RequestBody UpdatePasswordParam param) {
-        if (Cools.isEmpty(param.getOldPassword(), param.getPassword())) {
+    @PostMapping("/auth/reset/password")
+    public R resetPassword(@RequestBody UpdatePasswordParam param) {
+        if (Cools.isEmpty(param.getOldPassword(), param.getNewPassword())) {
             return R.error("Parameters Cannot Be Empty");
         }
         Long userId = getLoginUserId();
@@ -116,15 +141,15 @@
             return R.error("Please Login First");
         }
         if (!userService.comparePassword(userService.getById(userId).getPassword(), param.getOldPassword())) {
-            return R.error("The Origin Password Was Incorrect");
+            return R.parse("408-The Current Password Was Incorrect");
         }
         User user = new User();
         user.setId(userId);
-        user.setPassword(userService.encodePassword(param.getPassword()));
+        user.setPassword(userService.encodePassword(param.getNewPassword()));
         if (userService.updateById(user)) {
-            return R.ok("Update Success");
+            return R.ok("Reset Password Success");
         }
-        return R.error("Update Fail");
+        return R.error("Reset Password Fail");
     }
 
 

--
Gitblit v1.9.1