Merge branch 'main' into bugfix/waiting-for-message

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<string> |
 
   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}`)
+  }
+}