channel-worker 1.0.8 → 1.0.10

package/lib/nst-manager.js

@@ -114,10 +114,11 @@ class NstManager {
       profileId = await this.ensureProfile(profileIdOrName, { extensionPath: options.extensionPath });
     }
 
-    // Check if already running
+    // If already running, stop first to reload with fresh extension
     if (await this.isProfileRunning(profileId)) {
-      console.log(`[nst] Profile ${profileId} already running — skipping`);
-      return { profileId, alreadyRunning: true };
+      console.log(`[nst] Profile ${profileId} already running — restarting to load extension`);
+      await this.stopProfile(profileId);
+      await new Promise(r => setTimeout(r, 3000)); // wait for browser to close
     }
 
     // Set proxy before launch
@@ -125,43 +126,26 @@ class NstManager {
       await this.setProxy(profileId, options.proxy);
     }
 
-    console.log(`[nst] Starting browser for profile: ${profileId}`);
-    const res = await this.client.browsers().startBrowser(profileId);
-    console.log(`[nst] Browser started`);
+    // Use connectBrowser instead of startBrowser — allows passing --load-extension args
+    const connectConfig = {
+      headless: false,
+      autoClose: false,
+    };
 
-    // Set nstProfileId in chrome.storage via CDP
-    try {
-      // Wait for browser to fully start
-      await new Promise(r => setTimeout(r, 3000));
-
-      const debugInfo = await this.client.browsers().getBrowserDebugger(profileId);
-      const wsUrl = debugInfo?.data?.webSocketDebuggerUrl;
-      if (wsUrl) {
-        const WebSocket = require('ws');
-        const ws = new WebSocket(wsUrl);
-        await new Promise((resolve, reject) => {
-          const timeout = setTimeout(() => { ws.close(); reject(new Error('CDP timeout')); }, 10000);
-          ws.on('open', () => {
-            // Navigate a new tab to set storage
-            ws.send(JSON.stringify({
-              id: 1,
-              method: 'Runtime.evaluate',
-              params: {
-                expression: `chrome.storage && chrome.storage.local ? chrome.storage.local.set({nstProfileId: "${profileIdOrName}"}) : null`,
-                awaitPromise: true,
-              },
-            }));
-            setTimeout(() => { clearTimeout(timeout); ws.close(); resolve(); }, 2000);
-          });
-          ws.on('error', (e) => { clearTimeout(timeout); reject(e); });
-        });
-        console.log(`[nst] Profile ID "${profileIdOrName}" set in extension storage`);
-      }
-    } catch (e) {
-      console.log(`[nst] Could not set profile ID in storage: ${e.message}`);
+    // Add extension path if provided
+    if (options.extensionPath) {
+      connectConfig.args = {
+        '--load-extension': options.extensionPath,
+        '--disable-extensions-except': options.extensionPath,
+      };
+      console.log(`[nst] Loading extension: ${options.extensionPath}`);
     }
 
-    return { profileId, response: res };
+    console.log(`[nst] Connecting browser for profile: ${profileId}`);
+    const res = await this.client.cdpEndpoints().connectBrowser(profileId, connectConfig);
+    console.log(`[nst] Browser started`);
+
+    return { profileId, wsEndpoint: res?.data?.webSocketDebuggerUrl, response: res };
   }
 
   // Stop browser

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "channel-worker",
-  "version": "1.0.8",
+  "version": "1.0.10",
   "description": "Channel Manager worker daemon — runs on remote machines to execute video pipeline jobs",
   "main": "lib/daemon.js",
   "bin": {