AdminUserManagerController.java 8.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188
  1. package com.hys.app.controller.system;
  2. import cn.hutool.core.util.IdUtil;
  3. import cn.hutool.core.util.StrUtil;
  4. import com.alibaba.fastjson.JSON;
  5. import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
  6. import com.hys.app.controller.erp.use.msg.AliSmsUtil;
  7. import com.hys.app.controller.erp.use.msg.MsgContant;
  8. import com.hys.app.controller.erp.use.msg.MsgUtilClient;
  9. import com.hys.app.framework.context.user.AdminUserContext;
  10. import com.hys.app.framework.exception.ResourceNotFoundException;
  11. import com.hys.app.framework.exception.ServiceException;
  12. import com.hys.app.framework.util.BeanUtil;
  13. import com.hys.app.framework.util.StringUtil;
  14. import com.hys.app.framework.validation.annotation.DemoSiteDisable;
  15. import com.hys.app.model.erp.entity.CompanyErp;
  16. import com.hys.app.model.errorcode.SystemErrorCode;
  17. import com.hys.app.model.system.dos.AdminUser;
  18. import com.hys.app.model.system.vo.AdminLoginVO;
  19. import com.hys.app.model.system.vo.AdminUserVO;
  20. import com.hys.app.service.base.service.ValidatorManager;
  21. import com.hys.app.service.erp.CompanyErpManager;
  22. import com.hys.app.service.system.AdminUserManager;
  23. import io.jsonwebtoken.ExpiredJwtException;
  24. import io.swagger.annotations.Api;
  25. import io.swagger.annotations.ApiImplicitParam;
  26. import io.swagger.annotations.ApiImplicitParams;
  27. import io.swagger.annotations.ApiOperation;
  28. import org.springframework.beans.factory.annotation.Autowired;
  29. import org.springframework.data.redis.core.RedisTemplate;
  30. import org.springframework.validation.annotation.Validated;
  31. import org.springframework.web.bind.annotation.*;
  32. import springfox.documentation.annotations.ApiIgnore;
  33. import javax.validation.constraints.NotEmpty;
  34. import javax.validation.constraints.NotNull;
  35. import java.util.HashMap;
  36. import java.util.List;
  37. import java.util.Map;
  38. import java.util.Random;
  39. import java.util.concurrent.TimeUnit;
  40. /**
  41. * 平台管理员控制器
  42. *
  43. * @author zh
  44. * @version v7.0
  45. * @since v7.0.0
  46. * 2018-06-20 20:38:26
  47. */
  48. @RestController
  49. @RequestMapping("/admin/systems/admin-users")
  50. @Api(description = "平台管理员相关API")
  51. @Validated
  52. public class AdminUserManagerController {
  53. @Autowired
  54. private AdminUserManager adminUserManager;
  55. @Autowired
  56. private ValidatorManager validatorManager;
  57. @Autowired
  58. private CompanyErpManager companyManager;
  59. @Autowired
  60. private RedisTemplate redisTemplate;
  61. @Autowired
  62. MsgUtilClient msgUtilClient;
  63. @GetMapping("/login")
  64. @ApiOperation(value = "用户名(手机号)/密码登录API")
  65. @ApiImplicitParams({
  66. @ApiImplicitParam(name = "username", value = "用户名", required = true, dataType = "String", paramType = "query"),
  67. @ApiImplicitParam(name = "password", value = "密码", required = true, dataType = "String", paramType = "query")
  68. })
  69. public AdminLoginVO login(String username,
  70. String password,
  71. @NotEmpty(message = "平台登录sys,公司登录com") String type,
  72. String phone,
  73. String checkCodeKey,
  74. String checkCode,
  75. @NotEmpty(message = "账号登录 password 手机登录 phone") String loginType,
  76. String companyName) {
  77. if ("password".equals(loginType)) {
  78. return adminUserManager.login(username, password, companyName, type);
  79. }else if ("phone".equals(loginType)){
  80. String code = (String) redisTemplate.opsForValue().get(checkCodeKey);
  81. if (!code.equals(checkCode)) {
  82. throw new ServiceException("验证码错误");
  83. }
  84. return adminUserManager.loginByPhone(phone, companyName, type);
  85. }
  86. throw new ServiceException("不支持的登录类型");
  87. }
  88. @ApiOperation(value = "刷新token")
  89. @PostMapping("/token")
  90. @ApiImplicitParam(name = "refresh_token", value = "刷新token", required = true, dataType = "String", paramType = "query")
  91. public String refreshToken(@ApiIgnore @NotEmpty(message = "刷新token不能为空") String refreshToken) {
  92. try {
  93. return adminUserManager.exchangeToken(refreshToken);
  94. } catch (ExpiredJwtException e) {
  95. throw new ServiceException("当前token已经失效");
  96. }
  97. }
  98. @PutMapping
  99. @ApiOperation(value = "修改管理员密码及头像", response = AdminUser.class)
  100. @ApiImplicitParams({
  101. @ApiImplicitParam(name = "face", value = "头像", required = true, dataType = "String", paramType = "query"),
  102. @ApiImplicitParam(name = "old_passwprd", value = "原密码", required = false, dataType = "String", paramType = "query"),
  103. @ApiImplicitParam(name = "password", value = "新密码", required = false, dataType = "String", paramType = "query")
  104. })
  105. @DemoSiteDisable
  106. public AdminUser edit(@NotEmpty(message = "管理员头像不能为空") String face, @ApiIgnore String oldPasswprd, String password) {
  107. Long uid = AdminUserContext.getAdmin().getUid();
  108. AdminUser adminUser = this.adminUserManager.getModel(uid);
  109. if (adminUser == null) {
  110. throw new ResourceNotFoundException("当前管理员不存在");
  111. }
  112. AdminUserVO adminUserVO = new AdminUserVO();
  113. BeanUtil.copyProperties(adminUser, adminUserVO);
  114. //校验密码
  115. if (!StringUtil.isEmpty(oldPasswprd) && StringUtil.isEmpty(password)) {
  116. throw new ServiceException(SystemErrorCode.E923.code(), "新密码不能为空");
  117. }
  118. if (StringUtil.isEmpty(oldPasswprd) && !StringUtil.isEmpty(password)) {
  119. throw new ServiceException(SystemErrorCode.E922.code(), "原始密码不能为空");
  120. }
  121. if (!StringUtil.isEmpty(oldPasswprd) && !StringUtil.isEmpty(password)) {
  122. String dbOldPassword = StringUtil.md5(oldPasswprd + adminUser.getUsername().toLowerCase());
  123. if (!dbOldPassword.equals(adminUser.getPassword())) {
  124. throw new ServiceException(SystemErrorCode.E921.code(), "原密码错误");
  125. }
  126. adminUserVO.setPassword(password);
  127. } else {
  128. adminUserVO.setPassword("");
  129. }
  130. adminUserVO.setFace(face);
  131. AdminUser upAdminUser = adminUserManager.edit(adminUserVO, adminUser.getId());
  132. this.adminUserManager.logout(uid);
  133. return upAdminUser;
  134. }
  135. @ApiOperation(value = "注销管理员登录")
  136. @PostMapping(value = "/logout")
  137. @ApiImplicitParams({
  138. @ApiImplicitParam(name = "uid", value = "管理员id", dataType = "int", paramType = "query", required = true)
  139. })
  140. public String loginOut(@NotNull(message = "管理员id不能为空") Long uid) {
  141. this.adminUserManager.logout(uid);
  142. return null;
  143. }
  144. @ApiOperation(value = "查询当前管理员的信息")
  145. @GetMapping(value = "/info")
  146. public AdminUser info() {
  147. return this.adminUserManager.getModel(AdminUserContext.getAdminUserId());
  148. }
  149. @GetMapping("/login/getCompany")
  150. public List<CompanyErp> loginGetCompany(String key) {
  151. return companyManager.list(new LambdaQueryWrapper<CompanyErp>()
  152. .select(CompanyErp::getId, CompanyErp::getName)
  153. .and(StrUtil.isNotBlank(key), item -> item.like(CompanyErp::getName, key)
  154. .or()
  155. .like(CompanyErp::getFirstPy, key)
  156. .or()
  157. .like(CompanyErp::getPinying, key)));
  158. }
  159. @GetMapping("/loginGetCheckCode")
  160. @ApiOperation(value = "获取验证码")
  161. public String getCheckCode(String phone) {
  162. Map<String, Object> map = new HashMap<>();
  163. String fastUUID = IdUtil.fastUUID();
  164. Random random = new Random();
  165. int randomNumber = 100000 + random.nextInt(900000);
  166. map.put("code", randomNumber);
  167. AliSmsUtil.sendSms(phone, MsgContant.CHECK_CODE_SMS, JSON.toJSONString(map));
  168. redisTemplate.opsForValue().set(fastUUID, String.valueOf(randomNumber));
  169. redisTemplate.expire(fastUUID, 5, TimeUnit.HOURS);
  170. return fastUUID;
  171. }
  172. }