|
|
@@ -1,5 +1,6 @@
|
|
|
package com.uas.sso.controller;
|
|
|
|
|
|
+import com.alibaba.fastjson.JSON;
|
|
|
import com.uas.sso.core.Status;
|
|
|
import com.uas.sso.entity.*;
|
|
|
import com.uas.sso.service.UserService;
|
|
|
@@ -13,6 +14,9 @@ import org.springframework.web.bind.annotation.*;
|
|
|
import java.util.HashMap;
|
|
|
import java.util.List;
|
|
|
import java.util.Map;
|
|
|
+import java.util.Set;
|
|
|
+
|
|
|
+import static com.alibaba.fastjson.JSON.parseArray;
|
|
|
|
|
|
/**
|
|
|
* @author wangmh
|
|
|
@@ -62,11 +66,12 @@ public class UpdateUserController extends BaseController {
|
|
|
@RequestMapping(value = "/check/mobile", method = RequestMethod.GET)
|
|
|
public ModelMap checkByMobile(String mobile) {
|
|
|
// 获取验证码
|
|
|
- String token = getMobileCode(mobile);
|
|
|
+ String token = getMobileToken(mobile);
|
|
|
|
|
|
// 设置返回数据
|
|
|
ModelMap data = new ModelMap();
|
|
|
data.put("token", token);
|
|
|
+ data.put("code", request.getSession().getAttribute("code"));
|
|
|
request.getSession().setAttribute("token", token);
|
|
|
return success(data);
|
|
|
}
|
|
|
@@ -104,6 +109,7 @@ public class UpdateUserController extends BaseController {
|
|
|
// 设置返回数据
|
|
|
User user = userService.findByMobile(mobile);
|
|
|
existToken = new Token(user, EXPIRES);
|
|
|
+ tokenService.save(existToken);
|
|
|
return success(existToken.getId());
|
|
|
}
|
|
|
|
|
|
@@ -125,6 +131,7 @@ public class UpdateUserController extends BaseController {
|
|
|
|
|
|
// 设置发送邮件信息
|
|
|
Token token = new Token(user, EXPIRES);
|
|
|
+ tokenService.save(token);
|
|
|
ModelMap data = new ModelMap();
|
|
|
data.put("vipName", user.getVipName());
|
|
|
// TODO 邮件认证地址
|
|
|
@@ -133,7 +140,7 @@ public class UpdateUserController extends BaseController {
|
|
|
} else if ("email".equals(operate)) {
|
|
|
data.put("url", "http://192.168.253.66:8081/update/user/setMail?token=" + token);
|
|
|
} else if ("question".equals(operate)) {
|
|
|
- data.put("url", "http://192.168.253.66:8081/update/user//setQuestion?token=" + token);
|
|
|
+ data.put("url", "http://192.168.253.66:8081/update/user/setQuestion?token=" + token);
|
|
|
}
|
|
|
|
|
|
// 发送邮件
|
|
|
@@ -153,23 +160,25 @@ public class UpdateUserController extends BaseController {
|
|
|
* @return
|
|
|
*/
|
|
|
@RequestMapping(value = "/check/question", method = RequestMethod.POST)
|
|
|
- public ModelMap checkByQuestion(List<Map<String, Object>> answers) {
|
|
|
+ public ModelMap checkByQuestion(String answers) {
|
|
|
+ List<Map> answersArray = JSON.parseArray(answers, Map.class);
|
|
|
|
|
|
// 校验密保答案
|
|
|
User user = (User) request.getSession().getAttribute("user");
|
|
|
List<UserQuestion> questions = user.getQuestions();
|
|
|
- Map<Long, String> userAnswer = new HashMap<>(questions.size());
|
|
|
+ Map<String, String> userAnswer = new HashMap<>(questions.size());
|
|
|
for (UserQuestion question : questions) {
|
|
|
- userAnswer.put(question.getId(), question.getAnswer());
|
|
|
+ userAnswer.put(question.getSort(), question.getAnswer());
|
|
|
}
|
|
|
- for (Map<String, Object> answer : answers) {
|
|
|
- if (!answer.get("answer").equals(userAnswer.get(answer.get("id")))){
|
|
|
+ for (Map<String, Object> answer : answersArray) {
|
|
|
+ if (!answer.get("answer").equals(userAnswer.get(answer.get("sort")))){
|
|
|
return error("答案错误");
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// 返回token
|
|
|
Token expireToken = new Token(user, EXPIRES);
|
|
|
+ tokenService.save(expireToken);
|
|
|
ModelMap data = new ModelMap();
|
|
|
data.put("token", expireToken.getId());
|
|
|
return success(data);
|
|
|
@@ -195,13 +204,14 @@ public class UpdateUserController extends BaseController {
|
|
|
if (user == null) {
|
|
|
return error("请求错误");
|
|
|
}
|
|
|
- tokenService.delete(token);
|
|
|
|
|
|
// 获取验证码
|
|
|
- token = getMobileCode(mobile);
|
|
|
+ token = getMobileToken(mobile);
|
|
|
|
|
|
request.getSession().setAttribute("user", user);
|
|
|
- return success(new ModelMap("token", token));
|
|
|
+ ModelMap data = new ModelMap("token", token);
|
|
|
+ data.put("code", request.getSession().getAttribute("code"));
|
|
|
+ return success(data);
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
@@ -274,7 +284,7 @@ public class UpdateUserController extends BaseController {
|
|
|
tokenService.delete(token);
|
|
|
|
|
|
// 获取验证码
|
|
|
- token = getEmailCode(email);
|
|
|
+ token = getEmailToken(email);
|
|
|
|
|
|
request.getSession().setAttribute("user", user);
|
|
|
return success(new ModelMap("token", token));
|
|
|
@@ -350,16 +360,23 @@ public class UpdateUserController extends BaseController {
|
|
|
* @param type 校验接收验证码类型(mobile or email)
|
|
|
* @param token 验证码token
|
|
|
* @param code 验证码
|
|
|
- * @param account 接收验证码账号
|
|
|
+ * @param mobile 接收验证码手机号
|
|
|
+ * @param email 接收验证码邮箱
|
|
|
* @return
|
|
|
*/
|
|
|
@RequestMapping(value = "/checkCode/{type}", method = RequestMethod.POST)
|
|
|
- public ModelMap checkCode(@PathVariable String type, @RequestParam String token, String code, String account) {
|
|
|
+ public ModelMap checkCode(@PathVariable String type, @RequestParam String token, String code, String mobile, String email) {
|
|
|
// 校验验证码
|
|
|
if ("mobile".equals(type)) {
|
|
|
- checkMobileCode(token, account, code);
|
|
|
+ if (StringUtils.isEmpty(mobile)) {
|
|
|
+ return error("手机号不能为空");
|
|
|
+ }
|
|
|
+ checkMobileCode(token, mobile, code);
|
|
|
} else if ("email".equals(type)) {
|
|
|
- checkEmailCode(token, account, code);
|
|
|
+ if (StringUtils.isEmpty(email)) {
|
|
|
+ return error("邮箱不能为空");
|
|
|
+ }
|
|
|
+ checkEmailCode(token, email, code);
|
|
|
}
|
|
|
|
|
|
return success();
|