crawler-lib/dist/lib/ids.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.Ids = exports.randomtail = exports.IdsSource = exports.IdsChannel = void 0;
const request_1 = require("./request");
const crypto_1 = require("crypto");
var IdsChannel;
(function (IdsChannel) {
    IdsChannel[IdsChannel["Csdn"] = 1] = "Csdn";
    IdsChannel[IdsChannel["Wechat"] = 2] = "Wechat";
    IdsChannel[IdsChannel["Zhihu"] = 3] = "Zhihu";
    IdsChannel[IdsChannel["Juejin"] = 4] = "Juejin";
})(IdsChannel = exports.IdsChannel || (exports.IdsChannel = {}));
var IdsSource;
(function (IdsSource) {
    IdsSource[IdsSource["Manual"] = 0] = "Manual";
    IdsSource[IdsSource["User"] = 1] = "User";
    IdsSource[IdsSource["AliyunMarket"] = 2] = "AliyunMarket";
    IdsSource[IdsSource["TencentCloudMarket"] = 3] = "TencentCloudMarket";
    IdsSource[IdsSource["HuaweiCloudMarket"] = 4] = "HuaweiCloudMarket";
    IdsSource[IdsSource["Xiniu"] = 6] = "Xiniu";
    IdsSource[IdsSource["Website"] = 7] = "Website";
})(IdsSource = exports.IdsSource || (exports.IdsSource = {}));
const randomtail = () => {
    return `_=${crypto_1.randomBytes(8).toString('hex')}`;
};
exports.randomtail = randomtail;
class Ids {
    constructor(config) {
        this.config = config;
    }
    async request(config) {
        const { endpoint, accessKeyId, accessKeySecret } = this.config;
        config.baseURL = endpoint;
        config.url = config.url || '/';
        const timestamp = Math.floor(Date.now() / 1000);
        const data = config.url.toLowerCase() + timestamp;
        const signature = crypto_1.createHmac('sha256', accessKeySecret).update(data).digest('base64');
        config.headers = config.headers || {};
        config.headers['Authorization'] = `IDS-HMAC-SHA256 Credential=${accessKeyId}/${timestamp},Signature=${signature}`;
        const res = await request_1.request(config);
        if (res.data.status != 1) {
            console.error(res.data);
            throw new Error(`[${res.data.status}] ${res.data.info}`);
        }
        return res.data;
    }
    getCrawlAuthors(channel) {
        return this.request({
            method: 'POST',
            url: '/api/ids/getCrawlAuthors',
            data: {
                code: channel
            }
        });
    }
    getCrawlArticleRules(params) {
        const { ids, channel } = params;
        return this.request({
            method: 'POST',
            url: '/api/ids/getCrawlArticleRules',
            data: {
                code: channel,
                sn_codes: ids
            }
        });
    }
    putArticle(data, rule) {
        data.external_read_num = data.external_read_num || 0;
        data.external_see_num = data.external_see_num || 0;
        data.external_like_num = data.external_like_num || 0;
        data.external_comment_num = data.external_comment_num || 0;
        return this.request({
            method: 'POST',
            url: `/api/ids/putArticleData?crawl=${rule}&${exports.randomtail()}`,
            data
        });
    }
    putProvider(data) {
        return this.request({
            method: 'POST',
            url: `/api/ids/putSourceData?${exports.randomtail()}`,
            data
        });
    }
    putSolution(data) {
        data.ext_sold = data.ext_sold || 0;
        data.ext_score = data.ext_score || 0;
        if (data.extra && typeof data.extra === 'object') {
            data.extra = JSON.stringify(data.extra);
        }
        return this.request({
            method: 'POST',
            url: `/api/ids/putSolutionData?${exports.randomtail()}`,
            data
        });
    }
}
exports.Ids = Ids;