wangmh 7 роки тому
батько
коміт
cb50ef7a4b

+ 9 - 1
sso-server/src/main/java/com/uas/sso/controller/UserspaceRegisterController.java

@@ -8,6 +8,7 @@ import com.uas.sso.core.PasswordStrength;
 import com.uas.sso.core.Status;
 import com.uas.sso.core.Step;
 import com.uas.sso.core.Type;
+import com.uas.sso.entity.App;
 import com.uas.sso.entity.User;
 import com.uas.sso.entity.UserAccount;
 import com.uas.sso.entity.Userspace;
@@ -165,17 +166,24 @@ public class UserspaceRegisterController extends BaseController {
         // 添加日志
         registerLogger.info(Type.REGISTER_USERSPACE.getValue(), Step.SECOND.getValue(), "企业注册", userspace, appId);
 
+        App app = appService.findOne(appId);
         if ("mall".equals(appId)) {
             UserAccount userAccount = userAccountService.findOneByUserUU(appId, admin.getUserUU(), userspace.getSpaceUU());
             List<String> loginUrls = appService.findAllLoginUrl();
             ModelMap data = new ModelMap();
             data.addAttribute("data", userAccount)
                     .addAttribute("loginUrls", loginUrls)
-                    .addAttribute("currentUrl", baseUrl);
+                    .addAttribute("currentUrl", baseUrl)
+                    .addAttribute("type", "mall");
             SSOToken st = new SSOToken(request, userAccount.getMobile());
             st.setData(JSON.toJSONString(userAccount));
             SSOHelper.setSSOCookie(request, response, st, true);
             return success(data);
+        } else if ("city".equals(appId) || "city".equals(app.getUserControl())) {
+            ModelMap data = new ModelMap();
+            data.addAttribute("bgImage", "https://dfs.ubtob.com/group1/M00/88/AE/CgpkyFsPxC-ADHD8ACBM14PiAS8632.png")
+                    .addAttribute("type", "city");
+            return success(data);
         }
         return success();
     }

+ 14 - 0
sso-server/src/main/java/com/uas/sso/entity/PageStyle.java

@@ -58,6 +58,12 @@ public class PageStyle implements Serializable {
     @Column(name = "bg_color")
     private String bgColor;
 
+    /**
+     * 注册完成背景颜色
+     */
+    @Column(name = "rebg_url")
+    private String rebgUrl;
+
     /**
      * 背景图片链接
      */
@@ -162,6 +168,14 @@ public class PageStyle implements Serializable {
         this.bgUrl = bgUrl;
     }
 
+    public String getRebgUrl() {
+        return rebgUrl;
+    }
+
+    public void setRebgUrl(String rebgUrl) {
+        this.rebgUrl = rebgUrl;
+    }
+
     public String getBgColor() {
         return bgColor;
     }