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 | 44 ++++++++++++++++++++++++++++++++++---------- 1 files changed, 34 insertions(+), 10 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 f2a4728..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,21 +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; @@ -46,28 +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); Tenant tenant = tenantService.getById(user.getTenantId()); - return R.ok("Login Success").add(new LoginResult(accessToken, user, tenant.getName())); + 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") @@ -77,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)); } @@ -101,7 +125,7 @@ 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"); } -- Gitblit v1.9.1