|
|
@@ -0,0 +1,329 @@
|
|
|
+package com.uas.sso.controller;
|
|
|
+
|
|
|
+import com.uas.sso.core.Status;
|
|
|
+import com.uas.sso.entity.*;
|
|
|
+import com.uas.sso.service.UserService;
|
|
|
+import com.uas.sso.support.SystemSession;
|
|
|
+import com.uas.sso.util.CollectionUtils;
|
|
|
+import org.springframework.beans.factory.annotation.Autowired;
|
|
|
+import org.springframework.ui.ModelMap;
|
|
|
+import org.springframework.util.StringUtils;
|
|
|
+import org.springframework.web.bind.annotation.RequestMapping;
|
|
|
+import org.springframework.web.bind.annotation.RequestMethod;
|
|
|
+import org.springframework.web.bind.annotation.RequestParam;
|
|
|
+import org.springframework.web.bind.annotation.RestController;
|
|
|
+
|
|
|
+import java.util.HashMap;
|
|
|
+import java.util.List;
|
|
|
+import java.util.Map;
|
|
|
+
|
|
|
+/**
|
|
|
+ * @author wangmh
|
|
|
+ * @create 2018-01-11 17:11
|
|
|
+ * @desc 修改用户信息(需要登录)
|
|
|
+ **/
|
|
|
+@RestController
|
|
|
+@RequestMapping("/update/user")
|
|
|
+public class UpdateUserController extends BaseController {
|
|
|
+
|
|
|
+ @Autowired
|
|
|
+ private UserService userService;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 校验信息token有效期
|
|
|
+ */
|
|
|
+ private static final int EXPIRES = 7*24*60*60;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 获取可以校验方式
|
|
|
+ *
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @RequestMapping(value = "/CheckType", method = RequestMethod.GET)
|
|
|
+ public ModelMap getCheckType() {
|
|
|
+ // 获取用户信息
|
|
|
+ UserAccount userAccount = SystemSession.getUserAccount();
|
|
|
+ User user = userService.findOne(userAccount.getUserUU());
|
|
|
+
|
|
|
+ // 设置返回数据
|
|
|
+ ModelMap data = new ModelMap();
|
|
|
+ data.put("mobile", Status.AUTHENTICATED.getCode() == user.getMobileValidCode() ? user.getMobile() : null);
|
|
|
+ data.put("email", Status.AUTHENTICATED.getCode() == user.getEmailValidCode() ? user.getEmail() : null);
|
|
|
+ data.put("question", CollectionUtils.isEmpty(user.getQuestions()) ? user.getQuestions() : null);
|
|
|
+ request.getSession().setAttribute("user", user);
|
|
|
+ return success();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 获取校验手机号验证码
|
|
|
+ *
|
|
|
+ * @author wangmh
|
|
|
+ * @date 2018/1/11 20:16
|
|
|
+ * @param mobile 手机号
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @RequestMapping(value = "/check/mobile", method = RequestMethod.GET)
|
|
|
+ public ModelMap checkByMobile(String mobile) {
|
|
|
+ // 获取验证码
|
|
|
+ String token = getMobileCode(mobile);
|
|
|
+
|
|
|
+ // 设置返回数据
|
|
|
+ ModelMap data = new ModelMap();
|
|
|
+ data.put("token", token);
|
|
|
+ request.getSession().setAttribute("token", token);
|
|
|
+ return success(data);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 校验手机号验证码
|
|
|
+ *
|
|
|
+ * @author wangmh
|
|
|
+ * @date 2018/1/11 20:17
|
|
|
+ * @param mobile 手机号
|
|
|
+ * @param code 验证码
|
|
|
+ * @param token 验证码token
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @RequestMapping(value = "/check/mobile", method = RequestMethod.POST)
|
|
|
+ public ModelMap checkByMobile(String mobile, String code, @RequestParam String token) {
|
|
|
+ // 校验空参数
|
|
|
+ if (StringUtils.isEmpty(mobile)) {
|
|
|
+ return error("手机号不能为空");
|
|
|
+ }
|
|
|
+ if (StringUtils.isEmpty(code)) {
|
|
|
+ return error("验证码不能为空");
|
|
|
+ }
|
|
|
+
|
|
|
+ // 校验token
|
|
|
+ Token existToken = tokenService.findOne(token);
|
|
|
+ if (existToken == null) {
|
|
|
+ return error("请重新获取验证码");
|
|
|
+ }
|
|
|
+
|
|
|
+ // 校验验证码
|
|
|
+ checkMobileCode(token, mobile, code);
|
|
|
+ tokenService.delete(token);
|
|
|
+
|
|
|
+ // 设置返回数据
|
|
|
+ User user = userService.findByMobile(mobile);
|
|
|
+ existToken = new Token(user, EXPIRES);
|
|
|
+ return success(existToken.getId());
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 通过邮箱重置密码,发送邮箱确认
|
|
|
+ *
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @RequestMapping(value = "/check/email", method = RequestMethod.GET)
|
|
|
+ public ModelMap checkByEmail(String operate, @RequestParam String email) {
|
|
|
+ // 校验空参数
|
|
|
+ if (StringUtils.isEmpty(email)) {
|
|
|
+ return error("手机号不能为空");
|
|
|
+ }
|
|
|
+
|
|
|
+ // 根据邮箱找到用户
|
|
|
+ User user = userService.findByUsername(email);
|
|
|
+
|
|
|
+ // 设置发送邮件信息
|
|
|
+ Token token = new Token(user, EXPIRES);
|
|
|
+ ModelMap data = new ModelMap();
|
|
|
+ data.put("vipName", user.getVipName());
|
|
|
+ // TODO 邮件认证地址
|
|
|
+ if ("mobile".equals(operate)) {
|
|
|
+ data.put("url", "http://192.168.253.66:8081/update/user/setMobile?token=" + token);
|
|
|
+ } else if ("email".equals(operate)) {
|
|
|
+ data.put("url", "http://192.168.253.66:8081/update/user/setMail?token=" + token);
|
|
|
+ }
|
|
|
+
|
|
|
+ // 发送邮件
|
|
|
+ if (!StringUtils.isEmpty(email)) {
|
|
|
+ Setting mailTplId = settingService.findOne("templateForSendMailWhenResetPassword");
|
|
|
+ if (!StringUtils.isEmpty(mailTplId)) {
|
|
|
+ mailService.send(mailTplId.getValue(), email, data);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ return success();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 重置密码第二步,认证密保
|
|
|
+ * @param answers 密保答案
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @RequestMapping(value = "/check/question", method = RequestMethod.POST)
|
|
|
+ public ModelMap checkByQuestion(List<Map<String, Object>> answers) {
|
|
|
+
|
|
|
+ // 校验密保答案
|
|
|
+ User user = (User) request.getSession().getAttribute("user");
|
|
|
+ List<UserQuestion> questions = user.getQuestions();
|
|
|
+ Map<Long, String> userAnswer = new HashMap<>(questions.size());
|
|
|
+ for (UserQuestion question : questions) {
|
|
|
+ userAnswer.put(question.getId(), question.getAnswer());
|
|
|
+ }
|
|
|
+ for (Map<String, Object> answer : answers) {
|
|
|
+ if (!answer.get("answer").equals(userAnswer.get(answer.get("id")))){
|
|
|
+ return error("答案错误");
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ // 返回token
|
|
|
+ Token expireToken = new Token(user, EXPIRES);
|
|
|
+ ModelMap data = new ModelMap();
|
|
|
+ data.put("token", expireToken.getId());
|
|
|
+ return success(data);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 设置手机号校验获取验证码
|
|
|
+ *
|
|
|
+ * @param mobile 获取验证码手机号
|
|
|
+ * @param token 第一步传递的tokenId
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @RequestMapping(value = "/setMobile", method = RequestMethod.GET)
|
|
|
+ public ModelMap updateMobile(String mobile, @RequestParam String token) {
|
|
|
+ // 校验token
|
|
|
+ Token existToken = tokenService.findOne(token);
|
|
|
+ if (existToken == null || existToken.isExpired()) {
|
|
|
+ return error("请求超时");
|
|
|
+ }
|
|
|
+
|
|
|
+ // 拿出user放入session中
|
|
|
+ User user = (User) existToken.getBind();
|
|
|
+ if (user == null) {
|
|
|
+ return error("请求错误");
|
|
|
+ }
|
|
|
+ tokenService.delete(token);
|
|
|
+
|
|
|
+ // 获取验证码
|
|
|
+ token = getMobileCode(mobile);
|
|
|
+
|
|
|
+ request.getSession().setAttribute("user", user);
|
|
|
+ return success(new ModelMap("token", token));
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 设置手机号
|
|
|
+ *
|
|
|
+ * @author wangmh
|
|
|
+ * @date 2018/1/11 21:15
|
|
|
+ * @param mobile 手机号
|
|
|
+ * @param code 验证码
|
|
|
+ * @param token 验证码tokenId
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @RequestMapping(value = "/setMobile", method = RequestMethod.POST)
|
|
|
+ public ModelMap updateMobile(String mobile, String code, @RequestParam String token) {
|
|
|
+ // 校验空参数
|
|
|
+ if (StringUtils.isEmpty(mobile)) {
|
|
|
+ return error("手机号不能为空");
|
|
|
+ }
|
|
|
+ if (StringUtils.isEmpty(code)) {
|
|
|
+ return error("验证码不能为空");
|
|
|
+ }
|
|
|
+
|
|
|
+ // 从session中获取用户信息
|
|
|
+ User user = (User) request.getSession().getAttribute("user");
|
|
|
+ if (user == null) {
|
|
|
+ return error("请求超时,请刷新重试");
|
|
|
+ }
|
|
|
+
|
|
|
+ // 校验手机号是否被使用
|
|
|
+ if (userService.mobileHasRegistered(mobile)){
|
|
|
+ return error("手机号已注册");
|
|
|
+ }
|
|
|
+
|
|
|
+ // 校验token
|
|
|
+ Token existToken = tokenService.findOne(token);
|
|
|
+ if (existToken == null) {
|
|
|
+ return error("请重新获取验证码");
|
|
|
+ }
|
|
|
+
|
|
|
+ // 校验验证码
|
|
|
+ checkMobileCode(token, mobile, code);
|
|
|
+ tokenService.delete(token);
|
|
|
+
|
|
|
+ // 修改手机号
|
|
|
+ userService.updateMobile(user.getUserUU(), mobile);
|
|
|
+ return success();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 修改邮箱,获取校验邮箱验证码
|
|
|
+ *
|
|
|
+ * @author wangmh
|
|
|
+ * @date 2018/1/11 20:16
|
|
|
+ * @param email 邮箱
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @RequestMapping(value = "/setEmail", method = RequestMethod.GET)
|
|
|
+ public ModelMap updateEmail(String email, @RequestParam String token) {
|
|
|
+ // 校验token
|
|
|
+ Token existToken = tokenService.findOne(token);
|
|
|
+ if (existToken == null || existToken.isExpired()) {
|
|
|
+ return error("请求超时");
|
|
|
+ }
|
|
|
+
|
|
|
+ // 拿出user放入session中
|
|
|
+ User user = (User) existToken.getBind();
|
|
|
+ if (user == null) {
|
|
|
+ return error("请求错误");
|
|
|
+ }
|
|
|
+ tokenService.delete(token);
|
|
|
+
|
|
|
+ // 获取验证码
|
|
|
+ token = getEmailCode(email);
|
|
|
+
|
|
|
+ request.getSession().setAttribute("user", user);
|
|
|
+ return success(new ModelMap("token", token));
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 设置手机号
|
|
|
+ *
|
|
|
+ * @author wangmh
|
|
|
+ * @date 2018/1/11 21:15
|
|
|
+ * @param email 邮箱
|
|
|
+ * @param code 验证码
|
|
|
+ * @param token 验证码tokenId
|
|
|
+ * @return
|
|
|
+ */
|
|
|
+ @RequestMapping(value = "/setEmail", method = RequestMethod.POST)
|
|
|
+ public ModelMap updateEmail(String email, String code, @RequestParam String token) {
|
|
|
+ // 校验空参数
|
|
|
+ if (StringUtils.isEmpty(email)) {
|
|
|
+ return error("邮箱不能为空");
|
|
|
+ }
|
|
|
+ if (StringUtils.isEmpty(code)) {
|
|
|
+ return error("验证码不能为空");
|
|
|
+ }
|
|
|
+
|
|
|
+ // 从session中获取用户信息
|
|
|
+ User user = (User) request.getSession().getAttribute("user");
|
|
|
+ if (user == null) {
|
|
|
+ return error("请求超时,请刷新重试");
|
|
|
+ }
|
|
|
+
|
|
|
+ // 校验手机号是否被使用
|
|
|
+ if (userService.emailHasRegistered(email)){
|
|
|
+ return error("邮箱已注册");
|
|
|
+ }
|
|
|
+
|
|
|
+ // 校验token
|
|
|
+ Token existToken = tokenService.findOne(token);
|
|
|
+ if (existToken == null) {
|
|
|
+ return error("请重新获取验证码");
|
|
|
+ }
|
|
|
+
|
|
|
+ // 校验验证码
|
|
|
+ checkEmailCode(token, email, code);
|
|
|
+ tokenService.delete(token);
|
|
|
+
|
|
|
+ // 修改邮箱
|
|
|
+ userService.updateEmail(user.getUserUU(), email);
|
|
|
+ return success();
|
|
|
+ }
|
|
|
+}
|