添加cookie管理页

back/services/cookie.ts

@@ -1,6 +1,15 @@
 import { Service, Inject } from 'typedi';
 import winston from 'winston';
 import fetch from 'node-fetch';
+import { getFileContentByName } from '../config/util';
+import config from '../config';
+import FormData from 'form-data';
+
+enum Status {
+  '正常',
+  '失效',
+  '状态异常',
+}
 
 @Service()
 export default class CookieService {
@@ -122,4 +131,165 @@ export default class CookieService {
       this.lstoken +
       '; ';
   }
+
+  private getCookie(response: any) {
+    var TrackerID = response.headers['set-cookie'][0];
+    TrackerID = TrackerID.substring(
+      TrackerID.indexOf('=') + 1,
+      TrackerID.indexOf(';'),
+    );
+    var pt_key = response.headers['set-cookie'][1];
+    pt_key = pt_key.substring(pt_key.indexOf('=') + 1, pt_key.indexOf(';'));
+    var pt_pin = response.headers['set-cookie'][2];
+    pt_pin = pt_pin.substring(pt_pin.indexOf('=') + 1, pt_pin.indexOf(';'));
+    var pt_token = response.headers['set-cookie'][3];
+    pt_token = pt_token.substring(
+      pt_token.indexOf('=') + 1,
+      pt_token.indexOf(';'),
+    );
+    var pwdt_id = response.headers['set-cookie'][4];
+    pwdt_id = pwdt_id.substring(pwdt_id.indexOf('=') + 1, pwdt_id.indexOf(';'));
+    var s_key = response.headers['set-cookie'][5];
+    s_key = s_key.substring(s_key.indexOf('=') + 1, s_key.indexOf(';'));
+    var s_pin = response.headers['set-cookie'][6];
+    s_pin = s_pin.substring(s_pin.indexOf('=') + 1, s_pin.indexOf(';'));
+    this.cookies =
+      'TrackerID=' +
+      TrackerID +
+      '; pt_key=' +
+      pt_key +
+      '; pt_pin=' +
+      pt_pin +
+      '; pt_token=' +
+      pt_token +
+      '; pwdt_id=' +
+      pwdt_id +
+      '; s_key=' +
+      s_key +
+      '; s_pin=' +
+      s_pin +
+      '; wq_skey=';
+    var userCookie = 'pt_key=' + pt_key + ';pt_pin=' + pt_pin + ';';
+    console.log('\n############  登录成功，获取到 Cookie  #############\n\n');
+    console.log('Cookie1="' + userCookie + '"\n');
+    console.log('\n####################################################\n\n');
+    return userCookie;
+  }
+
+  public async addCookie() {
+    const cookie: any = await this.checkLogin();
+    if (cookie.body.errcode == 0) {
+      let ucookie = this.getCookie(cookie);
+      return ucookie;
+    } else {
+      return '';
+    }
+  }
+
+  private async checkLogin() {
+    try {
+      if (this.cookies == '') {
+        return 0;
+      }
+      let timeStamp = new Date().getTime();
+      let url =
+        'https://plogin.m.jd.com/cgi-bin/m/tmauthchecktoken?&token=' +
+        this.token +
+        '&ou_state=0&okl_token=' +
+        this.okl_token;
+      let payload = {
+        lang: 'chs',
+        appid: 300,
+        returnurl:
+          'https://wqlogin2.jd.com/passport/LoginRedirect?state=1100399130787&returnurl=//home.m.jd.com/myJd/newhome.action?sceneval=2&ufc=&/myJd/home.action',
+        source: 'wq_passport',
+      };
+      let form = new FormData();
+      for (const key in payload) {
+        form.append(key, (payload as any)[key] as any);
+      }
+      const response = await fetch(url, {
+        method: 'post',
+        body: form,
+        headers: {
+          Referer:
+            'https://plogin.m.jd.com/login/login?appid=300&returnurl=https://wqlogin2.jd.com/passport/LoginRedirect?state=' +
+            timeStamp +
+            '&returnurl=//home.m.jd.com/myJd/newhome.action?sceneval=2&ufc=&/myJd/home.action&source=wq_passport',
+          Cookie: this.cookies,
+          Connection: 'Keep-Alive',
+          'Content-Type': 'application/x-www-form-urlencoded; Charset=UTF-8',
+          Accept: 'application/json, text/plain, */*',
+          'User-Agent':
+            'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/86.0.4240.111 Safari/537.36',
+        },
+      });
+
+      return response;
+    } catch (error) {
+      console.log(error.response.body);
+      let res: any = {};
+      res.body = { check_ip: 0, errcode: 222, message: '出错' };
+      res.headers = {};
+      return res;
+    }
+  }
+
+  public async getCookies() {
+    const content = getFileContentByName(config.confFile);
+    const regx = /Cookie[0-9]{1}\=\"(.+?)\"/g;
+    let m,
+      data = [];
+    while ((m = regx.exec(content)) !== null) {
+      data.push(m[1]);
+    }
+    return this.formatCookie(data);
+  }
+
+  private async formatCookie(data: any[]) {
+    const result = [];
+    for (const x of data) {
+      const { nickname, status } = await this.getJdInfo(x);
+      result.push({
+        pin: x.match(/pt_pin=(.+?);/)[1],
+        cookie: x,
+        status,
+        nickname: nickname,
+      });
+    }
+    return result;
+  }
+
+  private getJdInfo(cookie: string) {
+    return fetch(
+      `https://me-api.jd.com/user_new/info/GetJDUserInfoUnion?orgFlag=JD_PinGou_New&callSource=mainorder&channel=4&isHomewhite=0&sceneval=2&_=${Date.now()}&sceneval=2&g_login_type=1&g_ty=ls`,
+      {
+        method: 'get',
+        headers: {
+          Accept: '*/*',
+          'Accept-Encoding': 'gzip, deflate, br',
+          'Accept-Language': 'zh-cn',
+          Connection: 'keep-alive',
+          Cookie: cookie,
+          Referer: 'https://home.m.jd.com/myJd/newhome.action',
+          'User-Agent':
+            'Mozilla/5.0 (iPhone; CPU iPhone OS 14_4_1 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/14.0.3 Mobile/15E148 Safari/604.1',
+          Host: 'me-api.jd.com',
+        },
+      },
+    )
+      .then((x) => {
+        console.log(x);
+        return x.json();
+      })
+      .then((x) => {
+        console.log(x.data.userInfo);
+        if (x.retcode === '0' && x.data && x.data.userInfo) {
+          return { nickname: x.data.userInfo.baseInfo.nickname, status: 0 };
+        } else if (x.retcode === 13) {
+          return { status: 1, nickname: '-' };
+        }
+        return { status: 2, nickname: '-' };
+      });
+  }
 }