From 87b391b8a81ee2abdaa4131d245784ecc7a54e9a Mon Sep 17 00:00:00 2001
From: admin <weikou2014>
Date: 星期三, 17 六月 2020 19:19:40 +0800
Subject: [PATCH] 用户系统与卡密系统完善

---
 src/main/java/com/ks/tool/bkz/controller/ParseController.java |  101 ++++++++++++++++++++++++++++++++++++++++++--------
 1 files changed, 85 insertions(+), 16 deletions(-)

diff --git a/src/main/java/com/ks/tool/bkz/controller/ParseController.java b/src/main/java/com/ks/tool/bkz/controller/ParseController.java
index 163ce26..5713c29 100644
--- a/src/main/java/com/ks/tool/bkz/controller/ParseController.java
+++ b/src/main/java/com/ks/tool/bkz/controller/ParseController.java
@@ -1,44 +1,74 @@
 package com.ks.tool.bkz.controller;
 
 import com.alibaba.fastjson.JSONObject;
-import com.ks.tool.bkz.util.FirstOrderSubDataUtil;
-import com.ks.tool.bkz.util.UrlUtil;
+import com.ks.tool.bkz.dto.FirstOrderSubParseResult;
+import com.ks.tool.bkz.entity.FirstOrderSubInfo;
+import com.ks.tool.bkz.exception.TBCookieException;
+import com.ks.tool.bkz.service.FirstOrderSubInfoService;
+import com.ks.tool.bkz.service.manager.RedisManager;
+import com.ks.tool.bkz.util.*;
+import org.omg.CORBA.PUBLIC_MEMBER;
 import org.springframework.stereotype.Controller;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.ResponseBody;
 
+import javax.annotation.Resource;
+import java.io.PrintWriter;
+import java.io.UnsupportedEncodingException;
+import java.net.URLDecoder;
+import java.net.URLEncoder;
+import java.util.List;
 import java.util.Map;
 
 @Controller
 @RequestMapping("parse")
 public class ParseController {
+    @Resource
+    private RedisManager redisManager;
 
-    String cookie = null;
+    @Resource
+    private FirstOrderSubInfoService firstOrderSubInfoService;
+
+    private String getCookieKey(String tbUid) {
+        return "tbcookie-" + tbUid;
+    }
 
     @RequestMapping("getRequestUrl")
     @ResponseBody
-    public String getRequestUrl(String baseUrl) {
+    public String getRequestUrl(String baseUrl, String tbUid) {
         String host = UrlUtil.getHost(baseUrl);
         Map<String, String> params = UrlUtil.getParamsMap(baseUrl);
         params.put("t", System.currentTimeMillis() + "");
+
+        String cookie = redisManager.get(getCookieKey(tbUid));
         String token = FirstOrderSubDataUtil.getTokenFromCookie(cookie);
         String data = params.get("data");
+        try {
+            data = URLDecoder.decode(data, "UTF-8");
+        } catch (UnsupportedEncodingException e) {
+            e.printStackTrace();
+        }
         JSONObject dataJSON = JSONObject.parseObject(data);
         String paramsStr = dataJSON.getString("params");
         dataJSON = JSONObject.parseObject(paramsStr);
         int pageNo = dataJSON.getInteger("pageNo");
-        dataJSON.put("pageNo", pageNo + 1);
 
-        JSONObject paramsJSON = new JSONObject();
-        paramsJSON.put("params", dataJSON.toJSONString());
-        data = paramsJSON.toJSONString();
-        String sign = FirstOrderSubDataUtil.getSign(token, params.get("t"), params.get("appKey"), data);
-        params.put("data", data);
-        params.put("sign", sign);
-        String requestUrl = UrlUtil.getRequestUrl(host, params);
-
-
-        return requestUrl;
+        if (pageNo < 30) {
+            dataJSON.put("pageNo", pageNo + 1);
+            JSONObject paramsJSON = new JSONObject();
+            paramsJSON.put("params", dataJSON.toJSONString());
+            data = paramsJSON.toJSONString();
+            String sign = FirstOrderSubDataUtil.getSign(token, params.get("t"), params.get("appKey"), data);
+            params.put("data", data);
+            params.put("sign", sign);
+            String requestUrl = UrlUtil.getRequestUrl(host, params);
+            JSONObject root = new JSONObject();
+            root.put("url", requestUrl);
+            System.out.println(requestUrl);
+            return JsonUtil.loadTrueResult(root);
+        } else {
+            return JsonUtil.loadFalseResult(1, "");
+        }
     }
 
 
@@ -49,10 +79,49 @@
      * @return
      */
     @RequestMapping("uploadCookies")
+    @ResponseBody
     public String uploadCookie(String cookies) {
-        this.cookie = cookies;
+        String tbUid = TBCookieUtil.getTBUid(cookies);
+        if (!StringUtil.isNullOrEmpty(tbUid)) {
+            redisManager.save(getCookieKey(tbUid), cookies, 60 * 60);
+        }
         return "";
     }
 
+    @RequestMapping("getLoginUrl")
+    public void getLoginUrl(PrintWriter out) {
+        JSONObject data = new JSONObject();
+        data.put("loginUrl", String.format("http://login.taobao.com/member/login.jhtml?style=mini&newMini2=true&redirectURL=%s&full_redirect=true&disableQuickLogin=true", URLEncoder.encode("https://m.tb.cn/h.VN61jro?sm=9ed9c4")));
+        out.print(JsonUtil.loadTrueResult(data));
+    }
+
+
+    @RequestMapping("getCookie")
+    @ResponseBody
+    public String getCookie() {
+        return redisManager.get("cookie");
+    }
+
+    @RequestMapping("uploadContent")
+    @ResponseBody
+    public String uploadContent(String content) {
+        try {
+            FirstOrderSubParseResult result = FirstOrderSubDataUtil.parseOrderSubData(content);
+            if (result != null && result.isHasNextPage()) {
+                if (result.getGoodsList() != null)
+                    for (FirstOrderSubInfo info : result.getGoodsList()) {
+                        firstOrderSubInfoService.add(info);
+                        System.out.println(info.getId() + "-" + info.getTitle());
+                    }
+                System.out.println("----------------------------");
+                return JsonUtil.loadTrueResult(null);
+            } else {
+                return JsonUtil.loadFalseResult(1, "鏃犳洿澶氭暟鎹�");
+            }
+        } catch (TBCookieException e) {
+            return JsonUtil.loadFalseResult(1001, "娣樺疂鎺堟潈鍑洪敊");
+        }
+    }
+
 
 }

--
Gitblit v1.8.0