qinglong/back/services/cookie.ts

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 {
  private cookies: string = '';
  private s_token: string = '';
  private guid: string = '';
  private lsid: string = '';
  private lstoken: string = '';
  private okl_token: string = '';
  private token: string = '';
  constructor(@Inject('logger') private logger: winston.Logger) {}

  public async getYiYan(): Promise<any> {
    return { yiYan: 'test' };
  }

  private async step1() {
    try {
      let timeStamp = new Date().getTime();
      let url =
        'https://plogin.m.jd.com/cgi-bin/mm/new_login_entrance?lang=chs&appid=300&returnurl=https://wq.jd.com/passport/LoginRedirect?state=' +
        timeStamp +
        '&returnurl=https://home.m.jd.com/myJd/newhome.action?sceneval=2&ufc=&/myJd/home.action&source=wq_passport';
      const text = await fetch(url, {
        method: 'get',
        headers: {
          Connection: 'Keep-Alive',
          'Content-Type': 'application/x-www-form-urlencoded',
          Accept: 'application/json, text/plain, */*',
          'Accept-Language': 'zh-cn',
          Referer:
            'https://plogin.m.jd.com/login/login?appid=300&returnurl=https://wq.jd.com/passport/LoginRedirect?state=' +
            timeStamp +
            '&returnurl=https://home.m.jd.com/myJd/newhome.action?sceneval=2&ufc=&/myJd/home.action&source=wq_passport',
          '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',
          Host: 'plogin.m.jd.com',
        },
      });
      this.praseSetCookies(text);
    } catch (error) {
      this.logger.error(error.response.body);
    }
  }

  private async step2() {
    try {
      if (this.cookies == '') {
        return 0;
      }
      let timeStamp = new Date().getTime();
      let url =
        'https://plogin.m.jd.com/cgi-bin/m/tmauthreflogurl?s_token=' +
        this.s_token +
        '&v=' +
        timeStamp +
        '&remember=true';
      const response = await fetch(url, {
        method: 'post',
        body: JSON.stringify({
          lang: 'chs',
          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',
        }),
        headers: {
          Connection: 'Keep-Alive',
          'Content-Type': 'application/x-www-form-urlencoded; Charset=UTF-8',
          Accept: 'application/json, text/plain, */*',
          Cookie: this.cookies,
          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',
          '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',
          Host: 'plogin.m.jd.com',
        },
      }).then((res) => res.json);
      // this.token = response.body.token
      // this.okl_token = response.headers['set-cookie'][0]
      // this.okl_token = this.okl_token.substring(this.okl_token.indexOf("=") + 1, this.okl_token.indexOf(";"))
      var qrUrl =
        'https://plogin.m.jd.com/cgi-bin/m/tmauth?appid=300&client_type=m&token=' +
        this.token;
      return qrUrl;
    } catch (error) {
      console.log(error.response.body);
      return 0;
    }
  }

  private praseSetCookies(response: any) {
    this.s_token = response.body.s_token;
    this.guid = response.headers['set-cookie'][0];
    this.guid = this.guid.substring(
      this.guid.indexOf('=') + 1,
      this.guid.indexOf(';'),
    );
    this.lsid = response.headers['set-cookie'][2];
    this.lsid = this.lsid.substring(
      this.lsid.indexOf('=') + 1,
      this.lsid.indexOf(';'),
    );
    this.lstoken = response.headers['set-cookie'][3];
    this.lstoken = this.lstoken.substring(
      this.lstoken.indexOf('=') + 1,
      this.lstoken.indexOf(';'),
    );
    this.cookies =
      'guid=' +
      this.guid +
      '; lang=chs; lsid=' +
      this.lsid +
      '; lstoken=' +
      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: '-' };
      });
  }
}