RSSHub/lib/v2/nua/utils.js

/* eslint-disable no-undef */
/* eslint-disable no-case-declarations */
const got = require('@/utils/got');
const cheerio = require('cheerio');
const { parseDate } = require('@/utils/parse-date');
const timezone = require('@/utils/timezone');
const { fetchArticle } = require('@/utils/wechat-mp');

const pageType = (href) => {
    if (!href.startsWith('http')) {
        return 'in-nua';
    }
    const url = new URL(href);
    if (url.hostname === 'mp.weixin.qq.com') {
        return 'wechat-mp';
    } else if (url.hostname === 'www.nua.edu.cn') {
        return 'nua';
    } else {
        return 'unknown';
    }
};

function arti_link(text, href) {
    a = '<a href="' + href + '">' + text + '</a>';
    a.textContent = text;
    a.href = href;
    return a;
}

async function ProcessList(newsUrl, baseUrl, listName, listDate, webPageName) {
    const result = await got(newsUrl, {
        https: {
            rejectUnauthorized: false,
        },
    });
    const $ = cheerio.load(result.data);

    const pageName = $(webPageName).text();

    const items = $(listName)
        .toArray()
        .map((item) => {
            item = $(item);
            const href = $(item).find('a').attr('href');
            const type = pageType(href);

            return {
                link: type === 'in-nua' ? baseUrl + href : href,
                title: item.find('a').attr('title'),
                pubDate: timezone(parseDate(item.find(listDate).first().text(), 'YYYY-MM-DD'), +8),
                type,
            };
        });

    return [items, pageName];
}

const ProcessFeed = async (items, artiContent, ctx) =>
    await Promise.all(
        items.map((item) =>
            ctx.cache.tryGet(item.link, async () => {
                switch (item.type) {
                    case 'in-nua' || 'nua':
                        const result = await got({ url: item.link, https: { rejectUnauthorized: false } });
                        const $ = cheerio.load(result.data);
                        item.author = $('.arti_publisher').text() + '  ' + $('.arti_views').text();
                        item.description = $(artiContent).html();
                        return item;
                    case 'wechat-mp':
                        return fetchArticle(ctx, item.link);
                    case 'unknown':
                        item.description = `暂不支持解析该内容，请点击 ${arti_link('原文', item.link)}`;
                        return item;
                    default:
                        item.description = `default`;
                        return item;
                }
            })
        )
    );

module.exports = {
    ProcessList,
    ProcessFeed,
};