diff --git a/package-lock.json b/package-lock.json index c9c50513..c1c6d180 100644 --- a/package-lock.json +++ b/package-lock.json @@ -65,6 +65,7 @@ "socks-proxy-agent": "^8.0.5", "swagger-ui-express": "^5.0.1", "tsup": "^8.3.5", + "undici": "^7.16.0", "uuid": "^13.0.0" }, "devDependencies": { @@ -15480,6 +15481,15 @@ "url": "https://github.com/sponsors/ljharb" } }, + "node_modules/undici": { + "version": "7.16.0", + "resolved": "https://registry.npmjs.org/undici/-/undici-7.16.0.tgz", + "integrity": "sha512-QEg3HPMll0o3t2ourKwOeUAZ159Kn9mx5pnzHRQO8+Wixmh88YdZRiIwat0iNzNNXn0yoEtXJqFpyW7eM8BV7g==", + "license": "MIT", + "engines": { + "node": ">=20.18.1" + } + }, "node_modules/undici-types": { "version": "7.16.0", "resolved": "https://registry.npmjs.org/undici-types/-/undici-types-7.16.0.tgz", diff --git a/package.json b/package.json index 009782b7..aa7e508f 100644 --- a/package.json +++ b/package.json @@ -121,6 +121,7 @@ "socks-proxy-agent": "^8.0.5", "swagger-ui-express": "^5.0.1", "tsup": "^8.3.5", + "undici": "^7.16.0", "uuid": "^13.0.0" }, "devDependencies": { diff --git a/src/utils/makeProxyAgent.ts b/src/utils/makeProxyAgent.ts index 3b1379bd..e882876e 100644 --- a/src/utils/makeProxyAgent.ts +++ b/src/utils/makeProxyAgent.ts @@ -1,6 +1,8 @@ import { HttpsProxyAgent } from 'https-proxy-agent'; import { SocksProxyAgent } from 'socks-proxy-agent'; +import { ProxyAgent } from 'undici' + type Proxy = { host: string; password?: string; @@ -42,3 +44,40 @@ export function makeProxyAgent(proxy: Proxy | string): HttpsProxyAgent | return selectProxyAgent(proxyUrl); } + +export function makeProxyAgentUndici(proxy: Proxy | string): ProxyAgent { + let proxyUrl: string + let protocol: string + + if (typeof proxy === 'string') { + const url = new URL(proxy) + protocol = url.protocol.replace(':', '') + proxyUrl = proxy + } else { + const { host, password, port, protocol: proto, username } = proxy + protocol = (proto || 'http').replace(':', '') + + if (protocol === 'socks') { + protocol = 'socks5' + } + + const auth = username && password ? `${username}:${password}@` : '' + proxyUrl = `${protocol}://${auth}${host}:${port}` + } + + const PROXY_HTTP_PROTOCOL = 'http' + const PROXY_HTTPS_PROTOCOL = 'https' + const PROXY_SOCKS4_PROTOCOL = 'socks4' + const PROXY_SOCKS5_PROTOCOL = 'socks5' + + switch (protocol) { + case PROXY_HTTP_PROTOCOL: + case PROXY_HTTPS_PROTOCOL: + case PROXY_SOCKS4_PROTOCOL: + case PROXY_SOCKS5_PROTOCOL: + return new ProxyAgent(proxyUrl) + + default: + throw new Error(`Unsupported proxy protocol: ${protocol}`) + } +}