cdp-tunnel 2.3.0 → 2.4.1

package/cli/index.js

@@ -456,8 +456,9 @@ program
       
       if (wasRunning) {
         log('cyan', '🔄 重启服务器...');
-        startServer(savedPort, savedWatchdog, config.autoRestart);
-        log('green', '✅ 服务器已重启');
+        const child = startServer(savedPort, false, config.autoRestart);
+        child.unref();
+        log('green', '✅ 服务器已重启 (PID: ' + child.pid + ')');
       } else {
         log('cyan', '  运行 cdp-tunnel start 启动服务器');
       }

package/extension-new/background.js

@@ -124,11 +124,11 @@ importScripts('features/automation-badge.js');
     var sessionId = State.findSessionByTabId(tabId);
     if (sessionId) {
       var targetId = State.getTargetIdBySession(sessionId);
-      EventBuilder.send('Target.targetDestroyed', { targetId: targetId });
       EventBuilder.send('Target.detachedFromTarget', {
         sessionId: sessionId,
         targetId: targetId
       });
+      EventBuilder.send('Target.targetDestroyed', { targetId: targetId });
       State.unmapSession(sessionId);
       if (removedClientId) {
         SpecialHandler.updateTabGroupName(removedClientId);
@@ -138,6 +138,7 @@ importScripts('features/automation-badge.js');
     if (State.getCurrentTabId() === tabId) {
       State.persist(null, false);
     }
+    State.removeTabIdToClientId(tabId);
   });
 
   chrome.tabs.onUpdated.addListener(function(tabId, changeInfo, tab) {

package/extension-new/cdp/handler/forward.js

@@ -9,12 +9,12 @@ var ForwardHandler = (function() {
 
     if (!tabId) {
       Logger.warn('[Forward] No tabId for command:', method);
-      return Promise.resolve({});
+      return Promise.reject({ code: -32000, message: 'No target found for command: ' + method });
     }
 
     if (!State.isTabAttached(tabId)) {
       Logger.warn('[Forward] Tab not attached, skipping command:', method, 'tabId:', tabId);
-      return Promise.resolve({});
+      return Promise.reject({ code: -32000, message: 'Target is not attached' });
     }
 
     Logger.debug('[Forward]', method, '-> tabId:', tabId);

package/extension-new/cdp/handler/special.js

@@ -228,16 +228,17 @@ var SpecialHandler = (function() {
     if (targetId) {
       var tabId = State.getTabIdByTargetId(targetId);
       if (tabId) {
-        State.removeAttachedTab(tabId);
         var closeClientId = State.getClientIdByTabId(tabId);
         return new Promise(function(resolve) {
           chrome.tabs.remove(tabId, function() {
+            State.removeAttachedTab(tabId);
             if (closeClientId) {
               updateTabGroupName(closeClientId);
             }
             resolve({ success: true });
           });
         }).catch(function() {
+          State.removeAttachedTab(tabId);
           return { success: true };
         });
       }

package/extension-new/cdp/index.js

@@ -82,8 +82,8 @@ function routeCDPCommand(message) {
           resolve({ result: result });
         })
         .catch(function(error) {
-          Logger.error('[CDP] ERROR id=' + id + ' method=' + method + ' msg=' + error.message);
-          resolve({ error: { message: error.message } });
+          Logger.error('[CDP] ERROR id=' + id + ' method=' + method + ' msg=' + (error.message || error));
+          resolve({ error: { code: error.code || -32000, message: error.message || String(error) } });
         });
     } else {
       ForwardHandler.execute({ id: id, method: method, params: params, sessionId: sessionId, clientId: clientId })
@@ -92,8 +92,8 @@ function routeCDPCommand(message) {
           resolve({ result: result });
         })
         .catch(function(error) {
-          Logger.error('[CDP] ERROR id=' + id + ' method=' + method + ' msg=' + error.message + ' (forwarded)');
-          resolve({ error: { message: error.message } });
+          Logger.error('[CDP] ERROR id=' + id + ' method=' + method + ' msg=' + (error.message || error) + ' (forwarded)');
+          resolve({ error: { code: error.code || -32000, message: error.message || String(error) } });
         });
     }
   }).then(function(response) {

package/extension-new/core/debugger.js

@@ -169,6 +169,7 @@ var DebuggerManager = (function() {
         Logger.error('[Debugger] Failed to detach from tab', tabId, ':', error.message);
         State.removeAttachedTab(tabId);
         AutomationBadge.remove(tabId);
+        Screencast.stopPolling(tabId);
       });
   }
 

package/extension-new/core/state.js

@@ -342,6 +342,10 @@ var State = (function() {
     _state.tabIdToClientId.set(tabId, clientId);
   }
 
+  function removeTabIdToClientId(tabId) {
+    _state.tabIdToClientId.delete(tabId);
+  }
+
   function getClientIdByTabId(tabId) {
     return _state.tabIdToClientId.get(tabId);
   }
@@ -481,6 +485,7 @@ var State = (function() {
     clearAllState: clearAllState,
     cleanupAllTabs: cleanupAllTabs,
     setTabIdToClientId: setTabIdToClientId,
+    removeTabIdToClientId: removeTabIdToClientId,
     getClientIdByTabId: getClientIdByTabId,
     setGroupIdForClient: setGroupIdForClient,
     getGroupIdForClient: getGroupIdForClient,

package/extension-new/core/websocket.js

@@ -339,16 +339,20 @@ var WebSocketManager = (function() {
       return;
     }
     Logger.info('[WS] Closing ' + tabIds.length + ' attached tabs for clientId:', clientId);
+    var pending = tabIds.length;
     tabIds.forEach(function(tabId) {
       chrome.tabs.remove(tabId, function() {
         if (chrome.runtime.lastError) {
           Logger.info('[WS] Tab already closed:', tabId);
         }
+        chrome.debugger.detach({ tabId: tabId }).catch(function() {});
+        State.removeAttachedTab(tabId);
+        pending--;
+        if (pending === 0) {
+          resolve();
+        }
       });
-      chrome.debugger.detach({ tabId: tabId }).catch(function() {});
-      State.removeAttachedTab(tabId);
     });
-    resolve();
   }
 
   function handleServerRestart() {
@@ -372,17 +376,27 @@ var WebSocketManager = (function() {
     Logger.info('[WS] Browser.close received, cleaning up... clientId:', clientId);
     
     closeTabGroupByClientId(clientId).then(function() {
-      var attachedTabIds = State.getAttachedTabIds();
-      var promises = attachedTabIds.map(function(tabId) {
-          return chrome.debugger.detach({ tabId: tabId }).catch(function(e) {
-            Logger.info('[WS] Detach failed for tab', tabId, ':', e.message);
-          });
+      return new Promise(function(resolve) {
+        closeTabsByClientId(clientId, resolve);
+      });
+    }).then(function() {
+      var preExistingTabs = State.getPreExistingTabs();
+      var clientPreExisting = preExistingTabs.filter(function(tabId) {
+        return State.getClientIdByTabId(tabId) === clientId;
       });
-      Promise.all(promises).then(function() {
+      clientPreExisting.forEach(function(tabId) {
+        chrome.debugger.detach({ tabId: tabId }).catch(function() {});
+        State.removeAttachedTab(tabId);
+      });
+      State.clearPreExistingTabsForClient(clientId);
+
+      State.removeCDPClient(clientId);
+      if (State.getCDPClients().length === 0) {
         State.clearAllState();
         State.persist(null, false);
-        Logger.info('[WS] Browser.close cleanup complete');
-      });
+      }
+      broadcastStateUpdate();
+      Logger.info('[WS] Browser.close cleanup complete for client:', clientId);
     });
   }
 
@@ -398,7 +412,18 @@ var WebSocketManager = (function() {
     var cdpClients = State.getCDPClients() || [];
     var attachedTabIds = State.getAttachedTabIds();
     
+    if (attachedTabIds.length === 0) {
+      chrome.runtime.sendMessage({
+        type: 'stateUpdate',
+        connected: isConnected,
+        cdpClients: cdpClients,
+        attachedPages: []
+      }).catch(function() {});
+      return;
+    }
+
     var attachedPages = [];
+    var pending = attachedTabIds.length;
     attachedTabIds.forEach(function(tabId) {
       chrome.tabs.get(tabId, function(tab) {
         if (tab && !chrome.runtime.lastError) {
@@ -408,15 +433,17 @@ var WebSocketManager = (function() {
             url: tab.url || ''
           });
         }
+        pending--;
+        if (pending === 0) {
+          chrome.runtime.sendMessage({
+            type: 'stateUpdate',
+            connected: isConnected,
+            cdpClients: cdpClients,
+            attachedPages: attachedPages
+          }).catch(function() {});
+        }
       });
     });
-    
-    chrome.runtime.sendMessage({
-      type: 'stateUpdate',
-      connected: isConnected,
-      cdpClients: cdpClients,
-      attachedPages: attachedPages
-    }).catch(function() {});
   }
 
   function getQueueStats() {

package/extension-new/manifest.json

@@ -1,7 +1,7 @@
 {
   "manifest_version": 3,
   "name": "CDP Bridge",
-  "version": "2.3.0",
+  "version": "2.4.1",
   "description": "Chrome DevTools Protocol Bridge for Playwright/Puppeteer automation",
   "permissions": [
     "debugger",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cdp-tunnel",
-  "version": "2.3.0",
+  "version": "2.4.1",
   "description": "Bridge Chrome's debugger API to WebSocket — control your existing browser with Playwright/Puppeteer via CDP",
   "main": "server/proxy-server.js",
   "bin": "./cli/index.js",

package/server/proxy-server.js

@@ -1017,6 +1017,20 @@ function handleClientConnection(ws, clientInfo, customClientId = null) {
         // 广播更新后的客户端列表
         broadcastClientList();
 
+        // 清理该 client 的所有映射
+        for (const [tId, cId] of targetIdToClientId.entries()) {
+            if (cId === id) targetIdToClientId.delete(tId);
+        }
+        for (const [bcId, cId] of browserContextToClientId.entries()) {
+            if (cId === id) browserContextToClientId.delete(bcId);
+        }
+        if (clientIdToBrowserContext.has(id)) {
+            clientIdToBrowserContext.delete(id);
+        }
+        for (const [gId, mapping] of globalRequestIdMap.entries()) {
+            if (mapping.clientId === id) globalRequestIdMap.delete(gId);
+        }
+
         // 清理配对关系
         if (ws.pairedPlugin) {
             ws.pairedPlugin.pairedClientId = null;