47 lines
1.6 KiB
TypeScript
47 lines
1.6 KiB
TypeScript
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;
|