added: proxy support

Proxy: (http|https|sock\d{1})\n\nProxy usado pelo WebSocket\nWS_PROXY_URL=\nProxy usado para upload/download de mídia\nFETCH_PROXY_URL=\n\n⚠ **As duas variáveis podem conter as mesmas informações**

Author: jrCleber

.env.dev

@@ -89,6 +89,13 @@ PROVIDER_HOST=127.0.0.1
 PROVIDER_PORT=5656
 PROVIDER_PREFIX=codechat
 
+# Proxy: (http|https|sock\d{1})
+# 
+# Proxy usado pelo WebSocket
+WS_PROXY_URL=
+# Proxy usado para upload/download de mídia
+FETCH_PROXY_URL=
+
 # URL em que a documetação será exibida
 # EX.: v1.dodmain.com
 API_BACKEND=

package.json

@@ -57,6 +57,7 @@
     "fluent-ffmpeg": "^2.1.3",
     "form-data": "^4.0.4",
     "hbs": "^4.2.0",
+    "https-proxy-agent": "^7.0.6",
     "js-yaml": "^4.1.0",
     "jsonschema": "^1.4.1",
     "jsonwebtoken": "^9.0.2",
@@ -71,8 +72,10 @@
     "qrcode": "^1.5.1",
     "qrcode-terminal": "^0.12.0",
     "sharp": "^0.34.2",
+    "socks-proxy-agent": "^8.0.5",
     "swagger-ui-express": "^5.0.1",
     "ulid": "^3.0.1",
+    "undici": "^7.16.0",
     "ws": "^8.18.3",
     "yamljs": "^0.3.0"
   },

src/config/env.config.ts

@@ -111,6 +111,8 @@ export type GlobalWebhook = { URL: string; ENABLED: boolean };
 export type ConfigSessionPhone = { CLIENT: string; NAME: string };
 export type QrCodLimit = number;
 
+export type EnvProxy = { WS?: string; FETCH?: string };
+
 export interface Env {
   SERVER: HttpServer;
   STORE: StoreConf;
@@ -127,6 +129,7 @@ export interface Env {
   SESSION_SECRET: string;
   S3?: Bucket;
   WA_VERSION: string;
+  PROXY: EnvProxy;
 }
 
 export type Key = keyof Env;
@@ -245,6 +248,10 @@ export class ConfigService {
         USE_SSL: process.env?.S3_USE_SSL === 'true',
       },
       WA_VERSION: process.env?.WA_VERSION,
+      PROXY: {
+        WS: process.env?.WS_PROXY_URL || null,
+        FETCH: process.env?.FETCH_PROXY_URL || null,
+      },
     };
   }
 }

src/utils/proxy.ts

@@ -0,0 +1,55 @@
+import { HttpsProxyAgent } from 'https-proxy-agent';
+import { SocksProxyAgent } from 'socks-proxy-agent';
+import { ProxyAgent as UndiciProxyAgent } from 'undici';
+
+export interface ProxyAgents {
+  wsAgent?: any;
+  fetchAgent?: any;
+}
+
+export function createProxyAgents(
+  wsProxyUrl?: string,
+  fetchProxyUrl?: string,
+): ProxyAgents {
+  const agents: ProxyAgents = {};
+
+  const detectProtocol = (url: string) => url.split(':')[0].toLowerCase();
+
+  // WS
+  if (wsProxyUrl) {
+    const proto = detectProtocol(wsProxyUrl);
+    switch (proto) {
+      case 'http':
+      case 'https':
+        agents.wsAgent = new HttpsProxyAgent(wsProxyUrl);
+        break;
+      case 'socks':
+      case 'socks4':
+      case 'socks5':
+        agents.wsAgent = new SocksProxyAgent(wsProxyUrl);
+        break;
+      default:
+        console.warn(`[Proxy] Protocolo desconhecido para WS: ${proto}`);
+        break;
+    }
+  }
+
+  // FETCH
+  if (fetchProxyUrl) {
+    const proto = detectProtocol(fetchProxyUrl);
+    switch (proto) {
+      case 'http':
+      case 'https':
+      case 'socks':
+      case 'socks4':
+      case 'socks5':
+        agents.fetchAgent = new UndiciProxyAgent(fetchProxyUrl);
+        break;
+      default:
+        console.warn(`[Proxy] Protocolo desconhecido para Fetch: ${proto}`);
+        break;
+    }
+  }
+
+  return agents;
+}

src/whatsapp/services/whatsapp.service.ts

@@ -522,6 +522,9 @@ export class WAStartupService {
     const { EXPIRATION_TIME } = this.configService.get<QrCode>('QRCODE');
     const CONNECTION_TIMEOUT = this.configService.get<number>('CONNECTION_TIMEOUT');
 
+    const proxy = this.configService.get<EnvProxy>('PROXY');
+    const agents = createProxyAgents(proxy?.WS, proxy?.FETCH);
+
     const socketConfig: UserFacingSocketConfig = {
       auth: {
         creds: this.authState.state.creds,
@@ -530,6 +533,8 @@ export class WAStartupService {
           P({ level: 'silent' }) as any,
         ),
       },
+      agent: agents?.wsAgent,
+      fetchAgent: agents?.fetchAgent,
       logger: P({ level: 'silent' }) as any,
       printQRInTerminal: false,
       browser,