rsshub/lib/routes/skeb/works.ts

Make RSS Great Again!

import { Data, DataItem, Route } from '@/types';
import { config } from '@/config';
import ConfigNotFoundError from '@/errors/types/config-not-found';
import { baseUrl, processWork } from './utils';
import cache from '@/utils/cache';
import ofetch from '@/utils/ofetch';

export const route: Route = {
    path: '/works/:username',
    categories: ['picture'],
    example: '/skeb/works/@brm2_1925',
    parameters: { username: 'Skeb Username with @' },
    features: {
        requireConfig: [
            {
                name: 'SKEB_BEARER_TOKEN',
                optional: false,
                description: '在瀏覽器開發者工具（F12）的主控台中輸入 `localStorage.getItem("token")` 獲取',
            },
        ],
        requirePuppeteer: false,
        antiCrawler: false,
        supportBT: false,
        supportPodcast: false,
        supportScihub: false,
    },
    name: 'Creator Works',
    maintainers: ['SnowAgar25'],
    handler,
    radar: [
        {
            title: 'Creator Works',
            source: ['skeb.jp/:username'],
            target: '/works/:username',
        },
    ],
    description: 'Get the latest works of a specific creator on Skeb',
};

async function handler(ctx): Promise<Data> {
    const username = ctx.req.param('username');

    if (!config.skeb || !config.skeb.bearerToken) {
        throw new ConfigNotFoundError('Skeb works RSS is disabled due to the lack of <a href="https://docs.rsshub.app/deploy/config#route-specific-configurations">relevant config</a>');
    }

    const url = `${baseUrl}/api/users/${username.replace('@', '')}/works`;

    const items = await cache.tryGet(url, async () => {
        const fetchData = async (retryCount = 0, maxRetries = 3) => {
            const data = await ofetch(url, {
                retry: 0,
                method: 'GET',
                query: { role: 'creator', sort: 'date', offset: '0' },
                headers: {
                    'User-Agent': config.ua,
                    Cookie: `request_key=${cache.get('skeb:request_key')}`,
                    Authorization: `Bearer ${config.skeb.bearerToken}`,
                },
            }).catch((error) => {
                if (retryCount >= maxRetries) {
                    throw new Error('Max retries reached');
                }
                const newRequestKey = error.response?._data?.match(/request_key=(.*?);/)?.[1];
                if (newRequestKey) {
                    cache.set('skeb:request_key', newRequestKey);
                    return fetchData(retryCount + 1, maxRetries);
                }
                throw error;
            });
            return data;
        };

        const data = await fetchData();

        if (!data || !Array.isArray(data)) {
            throw new Error('Invalid data received from API');
        }

        return data.map((item) => processWork(item)).filter(Boolean);
    });

    return {
        title: `Skeb - ${username}'s Works`,
        link: `${baseUrl}/${username}`,
        item: items as DataItem[],
    };
}