package com.vincent.rsf.server.system.controller; import com.vincent.rsf.common.enums.SystemModeType; import com.vincent.rsf.common.utils.RedisSupport; import com.vincent.rsf.common.utils.Utils; import com.vincent.rsf.framework.common.BaseRes; import com.vincent.rsf.framework.common.Cools; import com.vincent.rsf.framework.common.R; import com.vincent.rsf.framework.exception.CoolException; 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.RegisterParam; 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.enums.EmailType; 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.*; import javax.annotation.Resource; import javax.servlet.http.HttpServletRequest; import java.util.Date; import java.util.List; import java.util.stream.Collectors; /** * 认证控制器 * * Created by vincent on 1/30/2024 */ @RestController public class AuthController extends BaseController { private final RedisSupport redis = RedisSupport.defaultRedisSupport; @Resource private ConfigProperties configProperties; @Resource private UserService userService; @Resource private UserLoginService userLoginService; @Resource private RoleMenuService roleMenuService; @Resource private TenantService tenantService; @Autowired private EmailService emailService; @PostMapping("/login") public R login(@RequestBody LoginParam param, HttpServletRequest request) { // System.out.println(userService.encodePassword("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(StatusType.ENABLE.val)) { return R.error("Account Frozen"); } if (!userService.comparePassword(user.getPassword(), param.getPassword())) { 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("Sign In Success").add(new LoginResult(accessToken, user, tenant.getName())); } @GetMapping("/email/code") public R emailCode(@RequestParam("email") String email) { if (Cools.isEmpty(email)) { return R.parse(BaseRes.PARAM); } if (null != userService.getByEmail(email, null)) { return R.error("Email Already Exist"); } if (redis.getExpire(EmailType.REGISTER_VERIFY.toString(), email) > (configProperties.getCodeTime() - 60)) { return R.error("Please don't request code too frequently."); } String code = Utils.randomNumbers(configProperties.getCodeLength()); if (emailService.sendEmail(email, EmailType.REGISTER_VERIFY, Cools.add("code", code))) { redis.setValue(EmailType.REGISTER_VERIFY.toString(), email, code, configProperties.getCodeTime()); } return R.ok(); } @OperationLog("Register") @PostMapping("/register") public R register(@RequestBody RegisterParam param, HttpServletRequest request) { if (Cools.isEmpty(param.getUsername(), param.getPassword(), param.getEmail(), param.getCode())) { return R.parse(BaseRes.PARAM); } String cacheCode = redis.getValue(EmailType.REGISTER_VERIFY.toString(), param.getEmail()); if (Cools.isEmpty(cacheCode)) { return R.error("The verification code has expired."); } if (!cacheCode.equals(param.getCode())) { return R.error("The verification code is incorrect."); } User user = new User(); user.setUsername(param.getUsername()); user.setNickname(param.getUsername()); user.setPassword(userService.encodePassword(param.getPassword())); user.setEmail(param.getEmail()); user.setEmailVerified(1); user.setStatus(StatusType.ENABLE.val); user.setCreateTime(new Date()); if (!userService.save(user)) { throw new CoolException(""); } return R.ok(); } @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") public R tenantList() { return R.ok().add(tenantService.list()); } @GetMapping("/auth/user") public R userInfo() { User user = userService.getById(getLoginUserId()); return R.ok(userService.setUserAuthInfo(user)); } @GetMapping("/auth/menu") public R userMenu() { List menus = roleMenuService.listMenuByUserId(getLoginUserId(), Menu.TYPE_MENU); List 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)); } @OperationLog("Update UserInfo") @PostMapping("/auth/user") public R updateInfo(@RequestBody User user) { user.setId(getLoginUserId()); user.setUsername(null); user.setPassword(null); user.setEmailVerified(null); user.setTenantId(null); user.setStatus(null); if (userService.updateById(user)) { return R.ok("Save Success").add(userService.getById(user.getId())); } return R.error("Save Fail"); } @OperationLog("Reset Password") @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(); if (userId == null) { return R.error("Please Login First"); } if (!userService.comparePassword(userService.getById(userId).getPassword(), param.getOldPassword())) { return R.parse("408-The Current Password Was Incorrect"); } User user = new User(); user.setId(userId); user.setPassword(userService.encodePassword(param.getNewPassword())); if (userService.updateById(user)) { return R.ok("Reset Password Success"); } return R.error("Reset Password Fail"); } // ---------------------------------------------------- private MenuVo convertToVo(Menu menu) { if (menu == null) { return null; } MenuVo vo = new MenuVo(); vo.setId(menu.getId()); vo.setName(menu.getName()); vo.setParentId(menu.getParentId()); // vo.setParentName(menu.getParentName()); vo.setPath(menu.getPath()); // vo.setPathName(menu.getPathName()); vo.setRoute(menu.getRoute()); vo.setComponent(menu.getComponent()); vo.setType(menu.getType()); vo.setIcon(menu.getIcon()); vo.setSort(menu.getSort()); if (menu.getChildren() != null && !menu.getChildren().isEmpty()) { List childDTOs = menu.getChildren().stream() .map(this::convertToVo) .collect(Collectors.toList()); vo.setChildren(childDTOs); } return vo; } }