Spaces:
Running
Running
const { HttpProxyAgent } = require('http-proxy-agent') | |
const { HttpsProxyAgent } = require('https-proxy-agent') | |
const { SocksProxyAgent } = require('socks-proxy-agent') | |
const { LRUCache } = require('lru-cache') | |
const { InvalidProxyProtocolError } = require('./errors.js') | |
const PROXY_CACHE = new LRUCache({ max: 20 }) | |
const SOCKS_PROTOCOLS = new Set(SocksProxyAgent.protocols) | |
const PROXY_ENV_KEYS = new Set(['https_proxy', 'http_proxy', 'proxy', 'no_proxy']) | |
const PROXY_ENV = Object.entries(process.env).reduce((acc, [key, value]) => { | |
key = key.toLowerCase() | |
if (PROXY_ENV_KEYS.has(key)) { | |
acc[key] = value | |
} | |
return acc | |
}, {}) | |
const getProxyAgent = (url) => { | |
url = new URL(url) | |
const protocol = url.protocol.slice(0, -1) | |
if (SOCKS_PROTOCOLS.has(protocol)) { | |
return SocksProxyAgent | |
} | |
if (protocol === 'https' || protocol === 'http') { | |
return [HttpProxyAgent, HttpsProxyAgent] | |
} | |
throw new InvalidProxyProtocolError(url) | |
} | |
const isNoProxy = (url, noProxy) => { | |
if (typeof noProxy === 'string') { | |
noProxy = noProxy.split(',').map((p) => p.trim()).filter(Boolean) | |
} | |
if (!noProxy || !noProxy.length) { | |
return false | |
} | |
const hostSegments = url.hostname.split('.').reverse() | |
return noProxy.some((no) => { | |
const noSegments = no.split('.').filter(Boolean).reverse() | |
if (!noSegments.length) { | |
return false | |
} | |
for (let i = 0; i < noSegments.length; i++) { | |
if (hostSegments[i] !== noSegments[i]) { | |
return false | |
} | |
} | |
return true | |
}) | |
} | |
const getProxy = (url, { proxy, noProxy }) => { | |
url = new URL(url) | |
if (!proxy) { | |
proxy = url.protocol === 'https:' | |
? PROXY_ENV.https_proxy | |
: PROXY_ENV.https_proxy || PROXY_ENV.http_proxy || PROXY_ENV.proxy | |
} | |
if (!noProxy) { | |
noProxy = PROXY_ENV.no_proxy | |
} | |
if (!proxy || isNoProxy(url, noProxy)) { | |
return null | |
} | |
return new URL(proxy) | |
} | |
module.exports = { | |
getProxyAgent, | |
getProxy, | |
proxyCache: PROXY_CACHE, | |
} | |