Update src/index.ts

src/index.ts

@@ -1,21 +1,11 @@
 import { Hono } from 'hono';
 import { serve } from '@hono/node-server';
 import { cors } from 'hono/cors';
-import { chromium, type Browser, type Route } from 'playwright';
+import * as playwright from 'playwright';
 
-// Browser instance
-let browser: Browser | null = null;
-
-// Initialize browser
-async function initBrowser() {
-  if (!browser) {
-    browser = await chromium.launch({
-      headless: true,
-      args: ['--no-sandbox', '--disable-setuid-sandbox', '--disable-dev-shm-usage'],
-    });
-  }
-  return browser;
-}
+// Config
+const MAX_RETRY_COUNT = 3;
+const RETRY_DELAY = 5000;
 
 // Fake headers
 const FAKE_HEADERS = {
@@ -27,9 +17,37 @@ const FAKE_HEADERS = {
   'x-vqd-accept': '1',
 };
 
-// Config
-const MAX_RETRY_COUNT = 3;
-const RETRY_DELAY = 5000;
+// Browser instance
+let browser: playwright.Browser | null = null;
+
+// Ensure browser is installed and launched
+async function initBrowser() {
+  if (!browser) {
+    try {
+      // First try to launch
+      browser = await playwright.chromium.launch({
+        headless: true,
+        args: ['--no-sandbox', '--disable-setuid-sandbox', '--disable-dev-shm-usage'],
+      });
+    } catch (error) {
+      console.log('Installing browsers first...');
+      // If that fails, install the browsers
+      const { execSync } = require('child_process');
+      try {
+        execSync('npx playwright install chromium', { stdio: 'inherit' });
+        // Try launching again after installation
+        browser = await playwright.chromium.launch({
+          headless: true,
+          args: ['--no-sandbox', '--disable-setuid-sandbox', '--disable-dev-shm-usage'],
+        });
+      } catch (installError) {
+        console.error('Failed to install or launch browser:', installError);
+        throw installError;
+      }
+    }
+  }
+  return browser;
+}
 
 // Clean headers
 function cleanHeaders(headers: Record<string, string>) {
@@ -55,13 +73,14 @@ function cleanHeaders(headers: Record<string, string>) {
 // Get token
 async function playwrightRequestToken(clientIp?: string) {
   const browser = await initBrowser();
-  const page = await browser.newPage();
+  const context = await browser.newContext();
+  const page = await context.newPage();
 
   try {
     const url = 'https://duckduckgo.com/duckchat/v1/status';
-    let responseBody: any;
-
-    await page.route('**/*', async (route: Route) => {
+    
+    // Set up route handler
+    await page.route('**/*', async (route) => {
       if (route.request().url() === url) {
         const headers = cleanHeaders({
           ...FAKE_HEADERS,
@@ -69,7 +88,6 @@ async function playwrightRequestToken(clientIp?: string) {
         });
 
         await route.continue({
-          method: 'GET',
           headers,
         });
       } else {
@@ -86,11 +104,11 @@ async function playwrightRequestToken(clientIp?: string) {
       throw new Error('Request Token failed!');
     }
 
-    responseBody = await response.json();
-    await page.close();
+    const responseBody = await response.json();
+    await context.close();
     return responseBody?.vqd;
   } catch (error) {
-    await page.close();
+    await context.close();
     console.error('Request Token Error:', error);
     throw error;
   }
@@ -112,7 +130,8 @@ async function playwrightCreateCompletion(
   retryCount = 0
 ): Promise<ApiResponse> {
   const browser = await initBrowser();
-  const page = await browser.newPage();
+  const context = await browser.newContext();
+  const page = await context.newPage();
 
   try {
     const token = await playwrightRequestToken(clientIp);
@@ -125,7 +144,7 @@ async function playwrightCreateCompletion(
     let responseBody: any;
     let status: number;
 
-    await page.route('**/*', async (route: Route) => {
+    await page.route('**/*', async (route) => {
       if (route.request().url() === url) {
         const headers = cleanHeaders({
           ...FAKE_HEADERS,
@@ -156,7 +175,7 @@ async function playwrightCreateCompletion(
     if (!response.ok()) {
       if (status === 418 && retryCount < MAX_RETRY_COUNT) {
         console.warn('Rate limit hit (418), retrying...');
-        await page.close();
+        await context.close();
         await new Promise((resolve) => setTimeout(resolve, RETRY_DELAY));
         return playwrightCreateCompletion(model, content, returnStream, clientIp, retryCount + 1);
       }
@@ -164,7 +183,7 @@ async function playwrightCreateCompletion(
     }
 
     responseBody = await response.body();
-    await page.close();
+    await context.close();
 
     if (!returnStream) {
       const text = responseBody.toString();
@@ -197,7 +216,7 @@ async function playwrightCreateCompletion(
       body: responseBody,
     };
   } catch (error: any) {
-    await page.close();
+    await context.close();
     console.error('Create Completion Error:', error.message);
     throw error;
   }