yiyan-browser-agent 1.0.23 → 1.0.25

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "yiyan-browser-agent",
-  "version": "1.0.23",
+  "version": "1.0.25",
   "description": "AI coding agent powered by Yiyan (文心一言) via browser automation — no API key needed",
   "main": "src/index.js",
   "bin": {

package/src/browser-manager.js

@@ -23,11 +23,15 @@ class BrowserManager {
     // Try connect existing browser
     try {
       if (fs.existsSync(CDP_PORT_FILE)) {
-        const portInfo = JSON.parse(fs.readFileSync(CDP_PORT_FILE, 'utf8'));
-        const browserURL = `http://localhost:${portInfo.port || CDP_PORT}`;
+        const endpointInfo = JSON.parse(fs.readFileSync(CDP_PORT_FILE, 'utf8'));
+        const wsEndpoint = endpointInfo.wsEndpoint;
+
+        if (!wsEndpoint) {
+          throw new Error('No wsEndpoint in file');
+        }
 
         logger.info('Connecting to existing browser...');
-        const browser = await chromium.connectOverCDP(browserURL, { timeout: 5000 });
+        const browser = await chromium.connect({ wsEndpoint, timeout: 5000 });
 
         // Create NEW context and page (tab) for this task
         const context = await browser.newContext({
@@ -54,25 +58,33 @@ class BrowserManager {
    * Launch new browser with CDP port
    */
   static async _launchNew() {
-    logger.info('Launching new browser with CDP port ' + CDP_PORT + '...');
+    logger.info('Launching new browser server...');
 
-    // Launch browser (not persistentContext) to expose CDP properly
-    const browser = await chromium.launch({
+    // Use launchServer to create independent browser process
+    const browserServer = await chromium.launchServer({
       headless: false,
-      port: CDP_PORT,
       args: [
         '--disable-blink-features=AutomationControlled',
         '--no-first-run',
         '--no-sandbox',
-        `--user-data-dir=${SESSION_DIR}`,
       ],
     });
 
-    // Save CDP port info
+    // Get the WebSocket endpoint
+    const wsEndpoint = browserServer.wsEndpoint();
+    logger.dim('WebSocket endpoint: ' + wsEndpoint);
+
+    // Save endpoint info IMMEDIATELY before any other operations
     fs.writeFileSync(CDP_PORT_FILE, JSON.stringify({
-      port: CDP_PORT,
+      wsEndpoint: wsEndpoint,
       launchedAt: Date.now()
     }));
+    logger.dim('Endpoint saved to: ' + CDP_PORT_FILE);
+
+    logger.success('Browser server started!');
+
+    // Connect to our own server
+    const browser = await chromium.connect({ wsEndpoint });
 
     // Create context and page
     const context = await browser.newContext({
@@ -84,7 +96,10 @@ class BrowserManager {
     await page.goto(config.YIYAN_URL, { waitUntil: 'domcontentloaded', timeout: 15000 });
     await page.waitForTimeout(800);
 
-    logger.success('Browser launched! Tab ready.');
+    logger.success('Tab ready.');
+
+    // Return without browserServer - let it run independently
+    // Do NOT close browserServer on process exit
     return { browser, context, page, isNew: true };
   }
 

package/src/index.js

@@ -151,24 +151,25 @@ async function main() {
   const agent = new YiyanAgent({ saveLog: opts.saveLog });
 
   // ── Graceful shutdown handler ──────────────────────────────────────────────
-  // Only close browser on explicit exit (Ctrl+C, error)
-  const shutdown = async (code = 0) => {
-    logger.info('\nShutting down...');
-    try { await BrowserManager.forceClose(); } catch {}
+  // Only delete endpoint file on Ctrl+C (explicit exit)
+  const shutdown = async (code = 0, keepBrowser = true) => {
+    if (!keepBrowser) {
+      try { BrowserManager.forceClose(); } catch {}
+    }
     process.exit(code);
   };
 
-  process.on('SIGINT',  () => shutdown(0));
-  process.on('SIGTERM', () => shutdown(0));
+  process.on('SIGINT',  () => shutdown(0, false));  // Ctrl+C: close browser
+  process.on('SIGTERM', () => shutdown(0, false));  // Kill: close browser
   process.on('uncaughtException', async err => {
     logger.error(`Uncaught error: ${err.message}`);
     if (config.DEBUG) console.error(err.stack);
-    await shutdown(1);
+    await shutdown(1, false);  // Error: close browser
   });
   process.on('unhandledRejection', async reason => {
     logger.error(`Unhandled rejection: ${reason}`);
     if (config.DEBUG) console.error(reason);
-    await shutdown(1);
+    await shutdown(1, false);  // Error: close browser
   });
 
   // ── Calibrate mode ─────────────────────────────────────────────────────────