|
|
@@ -1,11 +1,17 @@
|
|
|
package com.uas.sso.controller;
|
|
|
|
|
|
+import com.alibaba.fastjson.JSON;
|
|
|
+import com.uas.sso.SSOHelper;
|
|
|
+import com.uas.sso.SSOToken;
|
|
|
import com.uas.sso.core.Step;
|
|
|
import com.uas.sso.core.Type;
|
|
|
import com.uas.sso.core.PasswordStrength;
|
|
|
import com.uas.sso.entity.User;
|
|
|
+import com.uas.sso.entity.UserAccount;
|
|
|
import com.uas.sso.logging.LoggerManager;
|
|
|
import com.uas.sso.logging.RegisterBufferedLogger;
|
|
|
+import com.uas.sso.service.AppService;
|
|
|
+import com.uas.sso.service.PersonalAccountService;
|
|
|
import com.uas.sso.service.UserService;
|
|
|
import com.uas.sso.util.PasswordLevelUtils;
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
|
@@ -16,6 +22,8 @@ import org.springframework.web.bind.annotation.RequestMapping;
|
|
|
import org.springframework.web.bind.annotation.RequestMethod;
|
|
|
import org.springframework.web.bind.annotation.ResponseBody;
|
|
|
|
|
|
+import java.util.List;
|
|
|
+
|
|
|
/**
|
|
|
* 个人注册controller
|
|
|
*
|
|
|
@@ -29,6 +37,12 @@ public class PersonalRegisterController extends BaseController {
|
|
|
@Autowired
|
|
|
private UserService userService;
|
|
|
|
|
|
+ @Autowired
|
|
|
+ private PersonalAccountService personalAccountService;
|
|
|
+
|
|
|
+ @Autowired
|
|
|
+ private AppService appService;
|
|
|
+
|
|
|
private RegisterBufferedLogger registerLogger = LoggerManager.getLogger(RegisterBufferedLogger.class);
|
|
|
|
|
|
/**
|
|
|
@@ -42,7 +56,7 @@ public class PersonalRegisterController extends BaseController {
|
|
|
*/
|
|
|
@RequestMapping(method = RequestMethod.POST)
|
|
|
@ResponseBody
|
|
|
- public ModelMap register(User user, String appId, String code, String token) {
|
|
|
+ public ModelMap register(User user, String appId, String code, String token, String baseUrl) {
|
|
|
// 获取参数
|
|
|
String vipName = user.getVipName();
|
|
|
String mobile = user.getMobile();
|
|
|
@@ -79,6 +93,19 @@ public class PersonalRegisterController extends BaseController {
|
|
|
// 注册成功后删除验证码token
|
|
|
tokenService.delete(token);
|
|
|
|
|
|
+ if ("mall".equals(appId)) {
|
|
|
+ UserAccount userAccount = personalAccountService.findOneByUserUU(appId, user.getUserUU());
|
|
|
+ List<String> loginUrls = appService.findAllLoginUrl();
|
|
|
+ ModelMap data = new ModelMap();
|
|
|
+ data.addAttribute("data", userAccount)
|
|
|
+ .addAttribute("loginUrls", loginUrls)
|
|
|
+ .addAttribute("currentUrl", baseUrl);
|
|
|
+ SSOToken st = new SSOToken(request, userAccount.getMobile());
|
|
|
+ st.setData(JSON.toJSONString(userAccount));
|
|
|
+ SSOHelper.setSSOCookie(request, response, st, true);
|
|
|
+ return success(data);
|
|
|
+ }
|
|
|
+
|
|
|
// 设置返回数据
|
|
|
return success(new ModelMap("userUU", user.getUserUU()));
|
|
|
}
|