RSSHub/lib/routes/kuaidi100/utils.js

const got = require('@/utils/got');
const wwwid_key = 'kuaidi100-wwwid';
const csrf_key = 'kuaidi100-csrf';
const query_count = 'kuaidi100-cookie-count';
const max_query_count = 30;

async function getCookie(ctx) {
    // Check if this key should be replace? every 30 times should be fine.
    await shouldUpdateCookie(ctx);
    let wwwid = await ctx.cache.get(wwwid_key);
    let csrf = await ctx.cache.get(csrf_key);
    if (!wwwid || !csrf) {
        const indexResponse = await got({
            method: 'get',
            url: 'https://www.kuaidi100.com/?from=appstore',
            headers: {
                // App store
                Referer: 'https://apps.apple.com/cn/app/%E5%BF%AB%E9%80%92100-%E5%8F%8C11%E5%AF%84%E4%BB%B6%E9%80%80%E8%B4%A7-%E4%B8%8A%E5%BF%AB%E9%80%92100/id458270120',
            },
        });
        const set_cookie = indexResponse.headers['set-cookie'];
        if (set_cookie) {
            for (const e of set_cookie) {
                if (e.indexOf('WWWID') === 0) {
                    wwwid = e.split(';')[0];
                } else if (e.indexOf('csrftoken') === 0) {
                    csrf = e.split(';')[0];
                }
            }
        }

        ctx.cache.set(wwwid_key, wwwid, 600);
        ctx.cache.set(csrf_key, csrf, 600);
        // We have acquired new cookie. It may due to cache timeout.
        // Force update counter and don't wait it finished.
        shouldUpdateCookie(ctx, true);
    }

    return {
        wwwid,
        csrf,
    };
}

/*
    Example Company
    {
        "available": false,  <- 这个不知道是什么的可用性，反正不是可查询的
        "canOrder": false,
        "comTypeName": "国内运输商",
        "contactTel": "626-818-2750",
        "createTime": 1486202888000,
        "hasNetwork": false,
        "id": 7172420,
        "name": "金岸物流",
        "number": "jinan",
        "shortName": "金岸物流",
        "shortNumber": "jinan",
        "siteUrl": "http://www.gpl-express.com/",
        "type": 0,
        "version": 776
    }
*/
async function getCompanyList(ctx) {
    // Using date as cache key and it will automatically expired by 1d
    const key = `kuaidi100-company-name-${new Date().toISOString().split('T')[0]}`;
    let list = await ctx.cache.get(key);
    if (!list) {
        const cookie = await getCookie(ctx);
        const wwwid = cookie.wwwid;
        const companyResponse = await got({
            method: 'post',
            url: 'https://www.kuaidi100.com/company.do?method=js&t=201701051440',
            headers: {
                Referer: 'https://www.kuaidi100.com/',
                Cookie: wwwid,
            },
        });
        list = companyResponse.body;

        // Parsing the js file
        try {
            list = list
                .substr(12)
                .replace(/};/g, '}')
                .replace(/'/g, '"');
            list = JSON.parse(list);
            list = list.company;
        } catch (e) {
            list = [];
        }

        ctx.cache.set(key, list);
    } else {
        list = JSON.parse(list);
    }
    return list;
}

async function shouldUpdateCookie(ctx, forcedUpdate = false) {
    if (forcedUpdate) {
        ctx.cache.set(query_count, 0);
    } else {
        const count = ctx.cache.get(query_count);
        if (!count) {
            ctx.cache.set(query_count, 1);
        } else {
            if (count > max_query_count) {
                ctx.cache.set(query_count, 0);
                await clearCookie(ctx);
            } else {
                ctx.cache.set(query_count, count + 1);
            }
        }
    }
}

async function clearCookie(ctx) {
    ctx.cache.set(wwwid_key, null);
    ctx.cache.set(csrf_key, null);
}

module.exports = {
    company: async (ctx) => await getCompanyList(ctx),
    checkCode: async (ctx, number, id, phone) => {
        const list = await getCompanyList(ctx);
        const company = list.find((c) => c.number === number);
        if (!company) {
            return {
                status: false,
                message: '快递公司编号不受支持！',
                company: {
                    name: '未知',
                },
            };
        } else {
            if (number.indexOf('shunfeng') !== -1 && !isNaN(phone) && String(phone).length !== 4) {
                return {
                    status: false,
                    message: '顺丰查询需要手机号后四位！',
                    company,
                };
            } else if (company.checkReg) {
                return {
                    status: true,
                    regex: new RegExp(company.checkReg).test(id),
                    company,
                };
            } else {
                return {
                    status: true,
                    regex: undefined,
                    company,
                };
            }
        }
    },

    /*
      Example Query
      {
          "message": "ok",
          "nu": "73123917441103",
          "ischeck": "1",
          "condition": "F00",
          "com": "zhongtong",
          "status": "200",
          "state": "3",
          "data": [{
                      "time": "2019-12-12 12:41:15",
                      "ftime": "2019-12-12 12:41:15",
                      "context": "【温州市】 快件已由【菜鸟的温州燎原华庭7栋103号店】代签收, 如有问题请电联（18581563547 / 4001787878）, 感谢您使用中通快递, 期待再次为您服务!",
                      "location": ""
                  }
                ...]
        Example failed Query
        {
            "message": "快递公司参数异常：验证码错误",
            "nu": "",
            "ischeck": "0",
            "condition": "",
            "com": "",
            "status": "408",
            "state": "0",
            "data": []
        }
    */

    getQuery: async (ctx, number, id, phone) => {
        const query_key = `kuaidi100-query-${number}-${id}`;
        let query = await ctx.cache.get(query_key);
        if (!query) {
            const cookie = await getCookie(ctx);
            const queryResponse = await got({
                method: 'get',
                url: `https://www.kuaidi100.com/query?type=${number}&postid=${id}&temp=${Math.random()}&phone=${phone ? phone : ''}`,
                headers: {
                    Referer: 'https://www.kuaidi100.com/',
                    Cookie: `${cookie.csrf}; ${cookie.wwwid}`,
                },
            });

            query = queryResponse.data;
            if (query.ischeck === '0' && query.status === '200') {
                // Not yet complete, don't cache for now.
                // Per owner suggestion
            } else {
                ctx.cache.set(query_key, query); // Finished, or error id
            }
        } else {
            query = JSON.parse(query);
        }

        return query;
    },
};