channel-worker 1.0.4 → 1.0.6

package/lib/command-poller.js

@@ -46,6 +46,9 @@ class CommandPoller {
         case 'close_profile':
           await this.handleCloseProfile(command);
           break;
+        case 'scan_facebook_pages':
+          await this.handleScanFacebookPages(command);
+          break;
         case 'verify_logins':
           await this.handleVerifyLogins(command);
           break;
@@ -61,12 +64,11 @@ class CommandPoller {
   }
 
   async handleLaunchProfile(command) {
-    const { profile_id } = command.payload || {};
+    const { profile_id, channel_id } = command.payload || {};
     console.log(`[commands] Launching Nstbrowser profile: ${profile_id}`);
 
     try {
       if (!this.nst) {
-        // No Nstbrowser SDK — try to get API key from dashboard settings
         const apiKey = await this.api.getSetting('nst_api_key');
         if (apiKey) {
           this.nst = new NstManager(apiKey);
@@ -75,12 +77,22 @@ class CommandPoller {
         }
       }
 
-      const result = await this.nst.launchProfile(profile_id);
-      console.log(`[commands] Profile ${profile_id} launched — user can now login via RDP`);
+      // Fetch channel proxy if channel_id provided
+      let proxy = null;
+      if (channel_id) {
+        try {
+          const channel = await this.api.getChannel(channel_id);
+          proxy = channel?.proxy || null;
+        } catch { /* ignore */ }
+      }
+
+      const extensionPath = this.config.extension_path || '';
+      const result = await this.nst.launchProfile(profile_id, { proxy, extensionPath });
+      console.log(`[commands] Profile ${profile_id} launched${proxy ? ' (proxy: ' + proxy + ')' : ''}${extensionPath ? ' (ext: ' + extensionPath + ')' : ''}`);
 
       await this.api.updateCommand(command._id, {
         status: 'done',
-        result: { profile_id: result.profileId, launched_at: new Date().toISOString() },
+        result: { profile_id: result.profileId, launched_at: new Date().toISOString(), proxy: proxy || null },
       });
     } catch (err) {
       console.error(`[commands] Failed to launch profile: ${err.message}`);
@@ -91,6 +103,59 @@ class CommandPoller {
     }
   }
 
+  async handleScanFacebookPages(command) {
+    const { profile_id } = command.payload || {};
+    console.log(`[commands] Scanning Facebook pages via profile: ${profile_id}`);
+
+    try {
+      if (!this.nst) {
+        const apiKey = await this.api.getSetting('nst_api_key');
+        if (apiKey) {
+          this.nst = new NstManager(apiKey);
+        } else {
+          throw new Error('Nstbrowser API key not configured.');
+        }
+      }
+
+      // Launch profile and get CDP endpoint
+      const { profileId, wsEndpoint } = await this.nst.connectProfile(profile_id);
+
+      // TODO: Use Puppeteer/Playwright to navigate facebook.com/pages and scrape page list
+      // For now, placeholder — worker needs puppeteer-core to actually scrape
+      console.log(`[commands] Browser connected at ${wsEndpoint}`);
+      console.log(`[commands] TODO: Navigate to facebook.com/pages and scrape page list`);
+
+      // Placeholder result — in real implementation, scrape pages from Facebook
+      const pages = [];
+      // const puppeteer = require('puppeteer-core');
+      // const browser = await puppeteer.connect({ browserWSEndpoint: wsEndpoint });
+      // const page = await browser.newPage();
+      // await page.goto('https://www.facebook.com/pages/?category=your_pages');
+      // pages = await page.evaluate(() => { /* scrape page names/urls */ });
+
+      // Send results to API
+      if (pages.length > 0) {
+        await this.api.request('POST', '/facebook-pages/scan-result', {
+          pages,
+          scanned_from_profile: profile_id,
+        });
+      }
+
+      await this.api.updateCommand(command._id, {
+        status: 'done',
+        result: { profile_id: profileId, pages_found: pages.length },
+      });
+
+      console.log(`[commands] Facebook scan complete — ${pages.length} pages found`);
+    } catch (err) {
+      console.error(`[commands] Facebook scan failed: ${err.message}`);
+      await this.api.updateCommand(command._id, {
+        status: 'failed',
+        error: err.message,
+      });
+    }
+  }
+
   async handleCloseProfile(command) {
     const { profile_id } = command.payload || {};
     console.log(`[commands] Closing profile: ${profile_id}`);

package/lib/nst-manager.js

@@ -57,7 +57,7 @@ class NstManager {
   }
 
   // Create profile if not exists, return profileId
-  async ensureProfile(name) {
+  async ensureProfile(name, options = {}) {
     const existing = await this.findProfile(name);
     if (existing) {
       console.log(`[nst] Profile "${name}" exists: ${existing}`);
@@ -65,7 +65,7 @@ class NstManager {
     }
 
     console.log(`[nst] Creating profile "${name}"...`);
-    const res = await this.client.profiles().createProfile({
+    const profileData = {
       name,
       platform: 'Windows',
       kernelMilestone: '132',
@@ -80,7 +80,17 @@ class NstManager {
         hardwareConcurrency: 8,
         deviceMemory: 8,
       },
-    });
+    };
+
+    // Add extension load arg if provided
+    if (options.extensionPath) {
+      profileData.args = {
+        '--load-extension': options.extensionPath,
+        '--disable-extensions-except': options.extensionPath,
+      };
+    }
+
+    const res = await this.client.profiles().createProfile(profileData);
 
     const profileId = res?.data?.profileId || res?.data?._id;
     if (!profileId) throw new Error('Failed to create profile — no ID returned');
@@ -89,11 +99,19 @@ class NstManager {
     return profileId;
   }
 
-  // Launch browser — skip if already running
-  async launchProfile(profileIdOrName) {
+  // Set proxy on profile
+  async setProxy(profileId, proxyUrl) {
+    if (!proxyUrl) return;
+    console.log(`[nst] Setting proxy for ${profileId}: ${proxyUrl}`);
+    await this.client.profiles().updateProfileProxy(profileId, { url: proxyUrl });
+    console.log(`[nst] Proxy set`);
+  }
+
+  // Launch browser — skip if already running, set proxy if provided
+  async launchProfile(profileIdOrName, options = {}) {
     let profileId = profileIdOrName;
     if (!this.isUUID(profileIdOrName)) {
-      profileId = await this.ensureProfile(profileIdOrName);
+      profileId = await this.ensureProfile(profileIdOrName, { extensionPath: options.extensionPath });
     }
 
     // Check if already running
@@ -102,6 +120,11 @@ class NstManager {
       return { profileId, alreadyRunning: true };
     }
 
+    // Set proxy before launch
+    if (options.proxy) {
+      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`);

package/package.json

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