const url = require('url');
const got = require('@/utils/got');
const cheerio = require('cheerio');
const iconv = require('iconv-lite');

const rootUrl = 'http://www.cs.com.cn/';

const config = {
    zzkx: {
        link: '/sylm/jsbd/',
        title: '中证快讯',
        query: 'ul.some-list li',
    },
    hyzx: {
        link: '/cj/hyzx/',
        title: '行业资讯',
        query: 'ul.list-lm li',
    },
};

module.exports = async (ctx) => {
    const cfg = config[ctx.params.caty];
    if (!cfg) {
        throw Error('Bad category. See <a href="https://docs.rsshub.app/finance.html#zhong-zheng-wang-zi-xun">docs</a>');
    }

    const currentUrl = url.resolve(rootUrl, cfg.link);
    const response = await got({
        method: 'get',
        url: currentUrl,
        responseType: 'buffer',
    });
    const $ = cheerio.load(iconv.decode(response.data, 'gbk'));
    const list = $(cfg.query)
        .slice(0, 20)
        .map((_, item) => {
            item = $(item);
            const a = item.find('a');
            return {
                title: item.find('h3').text() || a.text(),
                link: url.resolve(currentUrl, a.attr('href')),
                pubDate: new Date(item.find('em').text() || item.find('span').text()).toUTCString(),
            };
        })
        .get();

    const items = await Promise.all(
        list.map((item) =>
            ctx.cache.tryGet(item.link, async () => {
                const res = await got({
                    method: 'get',
                    url: item.link,
                    responseType: 'buffer',
                });
                const content = cheerio.load(iconv.decode(res.data, 'gbk'));

                item.description = content('.cont_article section').html() || content('.article-t').html();

                return item;
            })
        )
    );

    ctx.state.data = {
        title: '中证网 - ' + cfg.title,
        link: rootUrl,
        item: items,
    };
};