RSSHub/lib/utils/ofetch.ts

import type { HeaderGeneratorOptions } from 'header-generator';
import { register } from 'node-network-devtools';
import { createFetch } from 'ofetch';

import { config } from '@/config';
import logger from '@/utils/logger';

declare module 'ofetch' {
    interface FetchOptions {
        headerGeneratorOptions?: Partial<HeaderGeneratorOptions>;
    }
}

config.enableRemoteDebugging && process.env.NODE_ENV === 'dev' && register();

const rofetch = createFetch({ fetch: (input: Parameters<typeof fetch>[0], init?: Parameters<typeof fetch>[1]) => fetch(input, init) }).create({
    retryStatusCodes: [400, 408, 409, 425, 429, 500, 502, 503, 504],
    retry: config.requestRetry,
    retryDelay: 1000,
    // timeout: config.requestTimeout,
    onResponseError({ request, response, options }) {
        if (!options.retry) {
            return;
        }

        logger.warn(`Request ${request} with error ${response.status} remaining retry attempts: ${options.retry}`);
        if (!options.headers) {
            (options as any).headers = {};
        }
        if (options.headers instanceof Headers) {
            options.headers.set('x-prefer-proxy', '1');
        } else {
            ((options as any).headers as Record<string, string>)['x-prefer-proxy'] = '1';
        }
    },
    onRequestError({ request, error }) {
        logger.error(`Request ${request} fail: ${error.cause} ${error}`);
    },
    onResponse({ request, response }) {
        if (response.redirected) {
            logger.http(`Redirecting to ${response.url} for ${request}`);
        }
    },
});

export default rofetch;