94 lines
2.1 KiB
TypeScript
94 lines
2.1 KiB
TypeScript
import * as http from 'http';
|
|
import * as https from 'https';
|
|
import * as cache from 'lookup-dns-cache';
|
|
import fetch, { HeadersInit } from 'node-fetch';
|
|
import { HttpProxyAgent } from 'http-proxy-agent';
|
|
import { HttpsProxyAgent } from 'https-proxy-agent';
|
|
import config from '@/config';
|
|
import { URL } from 'url';
|
|
|
|
export async function getJson(url: string, accept = 'application/json, */*', timeout = 10000, headers?: HeadersInit) {
|
|
const res = await fetch(url, {
|
|
headers: Object.assign({
|
|
'User-Agent': config.userAgent,
|
|
Accept: accept
|
|
}, headers || {}),
|
|
timeout,
|
|
agent: getAgentByUrl,
|
|
});
|
|
|
|
if (!res.ok) {
|
|
throw {
|
|
name: `StatusError`,
|
|
statusCode: res.status,
|
|
message: `${res.status} ${res.statusText}`,
|
|
};
|
|
}
|
|
|
|
return await res.json();
|
|
}
|
|
|
|
export async function getHtml(url: string, accept = 'text/html, */*', timeout = 10000, headers?: HeadersInit) {
|
|
const res = await fetch(url, {
|
|
headers: Object.assign({
|
|
'User-Agent': config.userAgent,
|
|
Accept: accept
|
|
}, headers || {}),
|
|
timeout,
|
|
agent: getAgentByUrl,
|
|
});
|
|
|
|
if (!res.ok) {
|
|
throw {
|
|
name: `StatusError`,
|
|
statusCode: res.status,
|
|
message: `${res.status} ${res.statusText}`,
|
|
};
|
|
}
|
|
|
|
return await res.text();
|
|
}
|
|
|
|
/**
|
|
* Get http non-proxy agent
|
|
*/
|
|
const _http = new http.Agent({
|
|
keepAlive: true,
|
|
keepAliveMsecs: 30 * 1000,
|
|
});
|
|
|
|
/**
|
|
* Get https non-proxy agent
|
|
*/
|
|
const _https = new https.Agent({
|
|
keepAlive: true,
|
|
keepAliveMsecs: 30 * 1000,
|
|
lookup: cache.lookup,
|
|
});
|
|
|
|
/**
|
|
* Get http proxy or non-proxy agent
|
|
*/
|
|
export const httpAgent = config.proxy
|
|
? new HttpProxyAgent(config.proxy) as unknown as http.Agent
|
|
: _http;
|
|
|
|
/**
|
|
* Get https proxy or non-proxy agent
|
|
*/
|
|
export const httpsAgent = config.proxy
|
|
? new HttpsProxyAgent(config.proxy) as unknown as https.Agent
|
|
: _https;
|
|
|
|
/**
|
|
* Get agent by URL
|
|
* @param url URL
|
|
* @param bypassProxy Allways bypass proxy
|
|
*/
|
|
export function getAgentByUrl(url: URL, bypassProxy = false) {
|
|
if (bypassProxy || (config.proxyBypassHosts || []).includes(url.hostname)) {
|
|
return url.protocol == 'http:' ? _http : _https;
|
|
} else {
|
|
return url.protocol == 'http:' ? httpAgent : httpsAgent;
|
|
}
|
|
}
|