From 634c7515c04d9073d90483dae82b32029e4d6cbc Mon Sep 17 00:00:00 2001
From: java110 <928255095@qq.com>
Date: 星期二, 16 六月 2020 14:05:53 +0800
Subject: [PATCH] 优化本地代码

---
 service-front/src/main/java/com/java110/front/smo/ownerLogin/impl/OwnerAppLoginSMOImpl.java |   77 ++++++++++++++++++++++++++++++++++----
 1 files changed, 68 insertions(+), 9 deletions(-)

diff --git a/service-front/src/main/java/com/java110/front/smo/ownerLogin/impl/OwnerAppLoginSMOImpl.java b/service-front/src/main/java/com/java110/front/smo/ownerLogin/impl/OwnerAppLoginSMOImpl.java
index 756a30c..a756701 100644
--- a/service-front/src/main/java/com/java110/front/smo/ownerLogin/impl/OwnerAppLoginSMOImpl.java
+++ b/service-front/src/main/java/com/java110/front/smo/ownerLogin/impl/OwnerAppLoginSMOImpl.java
@@ -7,6 +7,7 @@
 import com.java110.core.context.PageData;
 import com.java110.core.factory.AuthenticationFactory;
 import com.java110.dto.owner.OwnerAppUserDto;
+import com.java110.dto.user.UserDto;
 import com.java110.front.properties.WechatAuthProperties;
 import com.java110.front.smo.ownerLogin.IOwnerAppLoginSMO;
 import com.java110.utils.cache.CommonCache;
@@ -110,15 +111,63 @@
     }
 
     @Override
+    public ResponseEntity<String> doLoginByKey(IPageData pd) throws SMOException {
+        JSONObject paramIn = JSONObject.parseObject(pd.getReqData());
+        Assert.hasKeyAndValue(paramIn, "key", "璇锋眰鎶ユ枃涓湭鍖呭惈涓存椂绉橀挜");
+        logger.debug("doLogin鍏ュ弬锛�" + paramIn.toJSONString());
+        ResponseEntity<String> responseEntity;
+
+        JSONObject loginInfo = JSONObject.parseObject(pd.getReqData());
+
+        UserDto userDto = new UserDto();
+        userDto.setKey(paramIn.getString("key"));
+        userDto = super.postForApi(pd, userDto, ServiceCodeConstant.SERVICE_CODE_USER_LOGIN, UserDto.class);
+
+
+        //鏍规嵁鐢ㄦ埛鏌ヨ鍟嗘埛淇℃伅
+        String userId = userDto.getUserId();
+
+        pd = PageData.newInstance().builder(userId, "", "", pd.getReqData(),
+                "", "", "", "",
+                pd.getAppId());
+        OwnerAppUserDto ownerAppUserDto = new OwnerAppUserDto();
+        ownerAppUserDto.setUserId(userId);
+        List<OwnerAppUserDto> ownerAppUserDtos = super.getForApis(pd, ownerAppUserDto, ServiceCodeConstant.LIST_APPUSERBINDINGOWNERS, OwnerAppUserDto.class);
+
+
+        if (ownerAppUserDtos == null || ownerAppUserDtos.size() < 1) {
+            responseEntity = new ResponseEntity<>("鐢ㄦ埛鏈粦瀹氫笟涓�", HttpStatus.BAD_REQUEST);
+            return responseEntity;
+        }
+
+        JSONObject appUser = JSONObject.parseObject(JSONObject.toJSONString(ownerAppUserDtos.get(0)));
+        appUser.put("userId", userId);
+        appUser.put("userName", paramIn.getString("username"));
+        JSONObject paramOut = new JSONObject();
+        paramOut.put("code", 0);
+        paramOut.put("msg", "鎴愬姛");
+        paramOut.put("owner", appUser);
+        paramOut.put("token", userDto.getToken());
+        paramOut.put("key", userDto.getKey());
+        return new ResponseEntity<>(paramOut.toJSONString(), HttpStatus.OK);
+
+    }
+
+    @Override
     public ResponseEntity<String> getPageAccessToken(IPageData pd) throws SMOException {
         JSONObject paramIn = JSONObject.parseObject(pd.getReqData());
         String authCode = paramIn.getString("code");
         String state = paramIn.getString("state");
-        String urlCode = CommonCache.getAndRemoveValue(paramIn.getString("urlCode"));
+        String paramStr = CommonCache.getAndRemoveValue(paramIn.getString("urlCode"));
 
-        if (StringUtil.isEmpty(urlCode)) {
-            return ResultVo.redirectPage("/#/pages/login/login");
+        if (StringUtil.isEmpty(paramStr)) {
+            return ResultVo.redirectPage("/");
         }
+
+        JSONObject param = JSONObject.parseObject(paramStr);
+        String redirectUrl = param.getString("redirectUrl");
+        String errorUrl = param.getString("errorUrl");
+
 
         String url = WechatConstant.APP_GET_ACCESS_TOKEN_URL.replace("APPID", wechatAuthProperties.getWechatAppId())
                 .replace("SECRET", wechatAuthProperties.getWechatAppSecret())
@@ -126,9 +175,9 @@
 
         ResponseEntity<String> paramOut = outRestTemplate.getForEntity(url, String.class);
 
-        logger.debug("璋冪敤寰俊鎹㈠幓token ", paramOut);
+        logger.debug("璋冪敤寰俊鎹㈠幓openId ", paramOut);
         if (paramOut.getStatusCode() != HttpStatus.OK) {
-            return ResultVo.redirectPage("/#/pages/login/login");
+            return ResultVo.redirectPage(errorUrl);
 
         }
 
@@ -149,7 +198,7 @@
             //灏唎penId鏀惧埌redis 缂撳瓨锛岀粰鍓嶆涓嬪彂涓存椂绁ㄦ嵁
             String code = UUID.randomUUID().toString();
             CommonCache.setValue(code, openId, expireTime);
-            return ResultVo.redirectPage("/#/pages/login/login?code=" + code);
+            return ResultVo.redirectPage(errorUrl + "?code=" + code);
         }
 
         // String accessToken = paramObj.getString("access_token");//鏆傛椂涓嶇敤
@@ -163,7 +212,12 @@
         } catch (Exception e) {
             logger.error("鍒涘缓token澶辫触");
         }
-        return ResultVo.redirectPage("/");
+        //鏌ヨ鐢ㄦ埛key
+        UserDto userDto = new UserDto();
+        userDto.setUserId(ownerAppUserDtos.get(0).getUserId());
+        UserDto tmpUserDto = super.getForApi(pd, userDto, ServiceCodeConstant.QUERY_USER_SECRET, UserDto.class);
+        redirectUrl = redirectUrl + (redirectUrl.indexOf("?") > 0 ? "&key=" + tmpUserDto.getKey() : "?key=" + tmpUserDto.getKey());
+        return ResultVo.redirectPage(redirectUrl);
 
     }
 
@@ -178,10 +232,15 @@
      * @throws SMOException
      */
     @Override
-    public ResponseEntity<String> refreshToken(IPageData pd, String redirectUrl, HttpServletRequest request, HttpServletResponse response) throws SMOException {
+    public ResponseEntity<String> refreshToken(IPageData pd, String redirectUrl,
+                                               String errorUrl,
+                                               HttpServletRequest request, HttpServletResponse response) throws SMOException {
         //鍒嗛厤urlCode
         String urlCode = UUID.randomUUID().toString();
-        CommonCache.setValue(urlCode, redirectUrl, expireTime);
+        JSONObject param = new JSONObject();
+        param.put("redirectUrl", redirectUrl);
+        param.put("errorUrl", errorUrl);
+        CommonCache.setValue(urlCode, param.toJSONString(), expireTime);
 
         URL url = null;
         String openUrl = "";

--
Gitblit v1.8.0