rsshub/lib/routes/subhd/index.ts

Make RSS Great Again!

import { Route } from '@/types';
import cache from '@/utils/cache';
import got from '@/utils/got';
import { load } from 'cheerio';
import timezone from '@/utils/timezone';
import { parseDate } from '@/utils/parse-date';

const config = {
    sub: {
        title: '字幕',
        category: 'new',
    },
    zu: {
        title: '字幕组',
        category: '14',
    },
    newest: {
        category: 'for backwards compatibility',
    },
};

export const route: Route = {
    path: '/:type?/:category?',
    name: 'Unknown',
    maintainers: [],
    handler,
};

async function handler(ctx) {
    const type = ctx.req.param('type') ?? 'sub';
    const category = ctx.req.param('category') ?? config[type].category;

    const rootUrl = 'https://subhd.tv';
    const currentUrl = `${rootUrl}/${type === 'newest' ? 'sub/new' : `${type}/${category}${type === 'zu' ? '/l' : ''}`}`;

    const response = await got({
        method: 'get',
        url: currentUrl,
    });

    const $ = load(response.data);

    $('.align-middle').each(function () {
        $(this).removeClass('link-dark');
    });

    let items = $('.link-dark')
        .toArray()
        .map((item) => {
            item = $(item);

            const pubDate = item.parent().parent().find('.align-text-top').last().text();
            const today = `${new Date().getFullYear()}-${new Date().getMonth()}-${new Date().getDate()}`;

            return {
                link: `${rootUrl}${item.attr('href')}`,
                author: item.parent().parent().find('.text-dark').last().text(),
                pubDate: timezone(parseDate(pubDate.includes('-') ? pubDate : `${today} ${pubDate}`), +8),
                title: `${item.parent().parent().find('.align-middle').text()} ${item.text().replace(/ - SubHD/, '')}`,
            };
        });

    items = await Promise.all(
        items.map((item) =>
            cache.tryGet(item.link, async () => {
                const detailResponse = await got({
                    method: 'get',
                    url: item.link,
                });

                const content = load(detailResponse.data);

                content('.rounded-circle').remove();
                content('.view-text').last().remove();

                item.description = content('.view-text').html() + content('.bg-white').first().html();

                return item;
            })
        )
    );

    return {
        title: $('title').text(),
        link: currentUrl,
        item: items,
    };
}