Spaces:

Reflectus
/

iper

Sleeping

App Files Files Community

GeminiBot commited on Jan 29

Commit

1ec87c7

1 Parent(s): df592cd

Hardcore logging: Request IDs, Memory tracking, Concurrency monitoring

Browse files

Files changed (1) hide show

src/duckai.ts +104 -135

src/duckai.ts CHANGED Viewed

@@ -3,174 +3,143 @@ import { createHash } from "node:crypto";
 import { Buffer } from "node:buffer";
 import UserAgent from "user-agents";
 export class DuckAI {
-  // Логика из старого рабочего кода
-  private async solveChallenge(vqdHash: string): Promise<string> {
-    const jsScript = Buffer.from(vqdHash, 'base64').toString('utf-8');
-    // 1. Создаем JSDOM с правильной структурой (iframe #jsa)
-    const dom = new JSDOM(
-      `<iframe id="jsa" sandbox="allow-scripts allow-same-origin" srcdoc="<!DOCTYPE html>
-<html>
-<head>
-<meta http-equiv="Content-Security-Policy" content="default-src 'none'; script-src 'unsafe-inline'">
-</head>
-<body></body>
-</html>" style="position: absolute; left: -9999px; top: -9999px;"></iframe>`,
-      {
-          runScripts: 'dangerously',
-          resources: "usable",
-          url: "https://duckduckgo.com/"
-      }
-    );
-    const window = dom.window as any;
-    // Моки, которые нужны JSDOM, чтобы чувствовать себя браузером
-    window.screen = { width: 1920, height: 1080, availWidth: 1920, availHeight: 1080 };
-    window.chrome = { runtime: {} };
-    // Глобальные переменные для eval
-    window.top.__DDG_BE_VERSION__ = 1;
-    window.top.__DDG_FE_CHAT_HASH__ = 1;
-    // 2. Получаем доступ к iframe и вставляем CSP (как в старом коде)
-    const jsa = window.document.querySelector('#jsa');
-    if (!jsa) throw new Error("Iframe #jsa not found in JSDOM");
-    const contentDoc = jsa.contentDocument || jsa.contentWindow.document;
-    const meta = contentDoc.createElement('meta');
-    meta.setAttribute('http-equiv', 'Content-Security-Policy');
-    meta.setAttribute('content', "default-src 'none'; script-src 'unsafe-inline';");
-    contentDoc.head.appendChild(meta);
-    // 3. Выполняем скрипт
-    const result = await window.eval(jsScript) as any;
-    if (!result) throw new Error("Challenge script returned nothing");
-    // 4. Жестко задаем User-Agent, который проходит проверку (из старого кода)
-    // Chrome/138.0.0.0 - это фейковый будущий UA, который использует скрипт DDG
-    result.client_hashes[0] = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Safari/537.36';
-    result.client_hashes = result.client_hashes.map((t: string) => {
-      const hash = createHash('sha256');
-      hash.update(t);
-      return hash.digest('base64');
-    });
-    return btoa(JSON.stringify(result));
-  }
-  getAvailableModels(): string[] {
-    return [
-      "gpt-4o-mini",
-      "gpt-5-mini",
-      "openai/gpt-oss-120b"
-    ];
   }
   async chat(request: any): Promise<string> {
-    // Используем тот же UA для запросов, что и в хеше, для консистентности
-    const userAgent = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Safari/537.36';
-    const headers: any = {
         "User-Agent": userAgent,
         "Accept": "text/event-stream",
         "x-vqd-accept": "1",
-        "accept-language": "en-US,en;q=0.9",
-        "cache-control": "no-cache",
-        "pragma": "no-cache",
-        "sec-fetch-dest": "empty",
-        "sec-fetch-mode": "cors",
-        "sec-fetch-site": "same-origin",
-        // Важный хедер из старого кода!
         "x-fe-version": "serp_20250401_100419_ET-19d438eb199b2bf7c300"
     };
     try {
-        console.log("[DuckAI] Fetching status...");
         const statusRes = await fetch("https://duckduckgo.com/duckchat/v1/status?q=1", { headers });
-        console.log(`[DuckAI] Status response: ${statusRes.status}`);
         const hashHeader = statusRes.headers.get("x-vqd-hash-1");
-        if (!hashHeader) {
-             // Иногда DDG не выдает challenge, а сразу дает x-vqd-4 (редко, но бывает)
-             const vqd4 = statusRes.headers.get("x-vqd-4");
-             if (vqd4) {
-                 console.log("[DuckAI] No challenge, got x-vqd-4 directly.");
-                 return await this.sendChatRequest(request, vqd4, headers);
-             }
-             throw new Error("Missing x-vqd-hash-1 and x-vqd-4");
-        }
-        console.log("[DuckAI] Solving challenge (Old Method)...");
-        const solvedVqd = await this.solveChallenge(hashHeader);
-        console.log("[DuckAI] Challenge solved.");
-        // Отправляем запрос с решенной капчей
         const response = await fetch("https://duckduckgo.com/duckchat/v1/chat", {
             method: "POST",
-            headers: {
-                ...headers,
-                "Content-Type": "application/json",
-                "x-vqd-hash-1": solvedVqd
-            },
             body: JSON.stringify(request)
         });
-        return await this.processResponse(response);
-    } catch (error: any) {
-        console.error(`[DuckAI] Error: ${error.message}`);
-        throw error; // Пробрасываем ошибку, чтобы сервер вернул 500 JSON, а не упал
-    }
-  }
-  // Вспомогательный метод для обработки ответа (чтобы не дублировать код)
-  private async processResponse(response: Response): Promise<string> {
-      console.log(`[DuckAI] Chat response status: ${response.status}`);
-      if (!response.ok) {
-          const errorText = await response.text();
-          console.log(`[DuckAI] Error body: ${errorText}`);
-          if (response.status === 429) {
-               throw new Error("Rate limit exceeded (429)");
           }
-          throw new Error(`DuckDuckGo API Error (${response.status}): ${errorText.substring(0, 100)}`);
-      }
-      const text = await response.text();
-      let llmResponse = "";
-      const lines = text.split("\n");
-      for (const line of lines) {
-        if (line.startsWith("data: ")) {
-          try {
-            const chunk = line.slice(6);
-            if (chunk === "[DONE]") break;
-            const json = JSON.parse(chunk);
-            if (json.message) llmResponse += json.message;
-          } catch (e) {}
         }
-      }
-      if (!llmResponse) throw new Error("Empty response from DuckDuckGo");
-      return llmResponse.trim();
-  }
-  // Метод для отправки запроса, если вдруг капча не нужна (на будущее)
-  private async sendChatRequest(request: any, vqd4: string, headers: any): Promise<string> {
-      const response = await fetch("https://duckduckgo.com/duckchat/v1/chat", {
-            method: "POST",
-            headers: {
-                ...headers,
-                "Content-Type": "application/json",
-                "x-vqd-4": vqd4
-            },
-            body: JSON.stringify(request)
-        });
-      return await this.processResponse(response);
   }
-}

 import { Buffer } from "node:buffer";
 import UserAgent from "user-agents";
+let activeRequests = 0;
 export class DuckAI {
+  private async solveChallenge(vqdHash: string, reqId: string): Promise<string> {
+    const start = Date.now();
+    const memBefore = process.memoryUsage().heapUsed / 1024 / 1024;
+    console.log(`[${reqId}] [Challenge] START. Mem: ${memBefore.toFixed(2)}MB`);
+    try {
+        const jsScript = Buffer.from(vqdHash, 'base64').toString('utf-8');
+        console.log(`[${reqId}] [Challenge] Creating JSDOM...`);
+        const dom = new JSDOM(
+          `<iframe id="jsa" sandbox="allow-scripts allow-same-origin" srcdoc="<!DOCTYPE html>
+    <html>
+    <head>
+    <meta http-equiv="Content-Security-Policy" content="default-src 'none'; script-src 'unsafe-inline'">
+    </head>
+    <body></body>
+    </html>" style="position: absolute; left: -9999px; top: -9999px;"></iframe>`,
+          {
+              runScripts: 'dangerously',
+              resources: "usable",
+              url: "https://duckduckgo.com/"
+          }
+        );
+        const window = dom.window as any;
+        window.screen = { width: 1920, height: 1080, availWidth: 1920, availHeight: 1080 };
+        window.chrome = { runtime: {} };
+        window.top.__DDG_BE_VERSION__ = 1;
+        window.top.__DDG_FE_CHAT_HASH__ = 1;
+        const jsa = window.document.querySelector('#jsa');
+        const contentDoc = jsa.contentDocument || jsa.contentWindow.document;
+        const meta = contentDoc.createElement('meta');
+        meta.setAttribute('http-equiv', 'Content-Security-Policy');
+        meta.setAttribute('content', "default-src 'none'; script-src 'unsafe-inline';");
+        contentDoc.head.appendChild(meta);
+        console.log(`[${reqId}] [Challenge] Running Eval...`);
+        const evalStart = Date.now();
+        const result = await window.eval(jsScript) as any;
+        console.log(`[${reqId}] [Challenge] Eval finished in ${Date.now() - evalStart}ms`);
+        if (!result) throw new Error("Challenge script returned nothing (possible JSDOM crash)");
+        result.client_hashes[0] = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Safari/537.36';
+        const solved = btoa(JSON.stringify({
+            ...result,
+            client_hashes: result.client_hashes.map((t: string) => {
+                const hash = createHash('sha256');
+                hash.update(t);
+                return hash.digest('base64');
+            })
+        }));
+        const duration = Date.now() - start;
+        const memAfter = process.memoryUsage().heapUsed / 1024 / 1024;
+        console.log(`[${reqId}] [Challenge] SUCCESS. Time: ${duration}ms. Mem Delta: +( ${(memAfter - memBefore).toFixed(2)} )MB`);
+        // Очистка JSDOM для экономии памяти
+        dom.window.close();
+        return solved;
+    } catch (e: any) {
+        console.error(`[${reqId}] [Challenge] CRITICAL ERROR: ${e.message}\n${e.stack}`);
+        throw e;
+    }
   }
   async chat(request: any): Promise<string> {
+    const reqId = Math.random().toString(36).substring(7).toUpperCase();
+    activeRequests++;
+    const startTime = Date.now();
+    console.log(`[${reqId}] [Chat] NEW REQUEST. Parallel active: ${activeRequests}`);
+    const userAgent = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/138.0.0.0 Safari/537.36';
+    const headers = {
         "User-Agent": userAgent,
         "Accept": "text/event-stream",
         "x-vqd-accept": "1",
         "x-fe-version": "serp_20250401_100419_ET-19d438eb199b2bf7c300"
     };
     try {
+        console.log(`[${reqId}] [Chat] Fetching status...`);
         const statusRes = await fetch("https://duckduckgo.com/duckchat/v1/status?q=1", { headers });
         const hashHeader = statusRes.headers.get("x-vqd-hash-1");
+        if (!hashHeader) throw new Error(`Status ${statusRes.status}: Missing VQD hash`);
+        const solvedVqd = await this.solveChallenge(hashHeader, reqId);
+        console.log(`[${reqId}] [Chat] Sending request to DDG...`);
         const response = await fetch("https://duckduckgo.com/duckchat/v1/chat", {
             method: "POST",
+            headers: { ...headers, "Content-Type": "application/json", "x-vqd-hash-1": solvedVqd },
             body: JSON.stringify(request)
         });
+        console.log(`[${reqId}] [Chat] DDG response status: ${response.status}`);
+        if (!response.ok) {
+            const body = await response.text();
+            console.error(`[${reqId}] [Chat] DDG ERROR BODY: ${body}`);
+            throw new Error(`DDG Error ${response.status}: ${body.substring(0, 100)}`);
+        }
+        const text = await response.text();
+        let llmResponse = "";
+        const lines = text.split("\n");
+        for (const line of lines) {
+          if (line.startsWith("data: ")) {
+            try {
+              const chunk = line.slice(6);
+              if (chunk === "[DONE]") break;
+              const json = JSON.parse(chunk);
+              if (json.message) llmResponse += json.message;
+            } catch (e) {}
           }
         }
+        console.log(`[${reqId}] [Chat] SUCCESS. Total time: ${Date.now() - startTime}ms`);
+        return llmResponse.trim() || "Empty response";
+    } catch (error: any) {
+        console.error(`[${reqId}] [Chat] FAILED: ${error.message}`);
+        throw error;
+    } finally {
+        activeRequests--;
+    }
   }
+  getAvailableModels() { return ["gpt-4o-mini", "gpt-5-mini", "openai/gpt-oss-120b"]; }
+}