channel-worker 1.0.0 → 1.0.1

package/lib/api-client.js

@@ -59,6 +59,15 @@ class ApiClient {
     const data = await this.request('GET', `/settings/${key}`);
     return data?.value;
   }
+
+  // Commands
+  async getNextCommand(workerId) {
+    return this.request('GET', `/workers/commands?worker_id=${workerId}`);
+  }
+
+  async updateCommand(commandId, data) {
+    return this.request('PUT', `/workers/commands/${commandId}`, data);
+  }
 }
 
 module.exports = { ApiClient };

package/lib/command-poller.js

@@ -0,0 +1,89 @@
+class CommandPoller {
+  constructor(api, config) {
+    this.api = api;
+    this.config = config;
+    this.timer = null;
+    this.pollIntervalMs = 3000; // check commands every 3s
+  }
+
+  start() {
+    console.log('[commands] Polling for commands (every 3s)');
+    this.poll();
+    this.timer = setInterval(() => this.poll(), this.pollIntervalMs);
+  }
+
+  stop() {
+    if (this.timer) {
+      clearInterval(this.timer);
+      this.timer = null;
+    }
+  }
+
+  async poll() {
+    try {
+      const command = await this.api.getNextCommand(this.config.worker_id);
+      if (!command) return;
+
+      console.log(`[commands] Received: ${command.type} (${command._id})`);
+
+      switch (command.type) {
+        case 'launch_profile':
+          await this.handleLaunchProfile(command);
+          break;
+        case 'close_profile':
+          await this.handleCloseProfile(command);
+          break;
+        case 'verify_logins':
+          await this.handleVerifyLogins(command);
+          break;
+        default:
+          console.warn(`[commands] Unknown command type: ${command.type}`);
+          await this.api.updateCommand(command._id, { status: 'failed', error: 'Unknown command type' });
+      }
+    } catch (err) {
+      if (this.config.verbose) {
+        console.error(`[commands] Poll error: ${err.message}`);
+      }
+    }
+  }
+
+  async handleLaunchProfile(command) {
+    const { profile_id } = command.payload || {};
+    console.log(`[commands] Launching Nstbrowser profile: ${profile_id}`);
+
+    try {
+      // TODO: Integrate with Nstbrowser SDK
+      // const nstApi = require('./nst-manager');
+      // await nstApi.launchProfile(profile_id);
+
+      console.log(`[commands] Profile ${profile_id} launched — user can now login via RDP`);
+
+      await this.api.updateCommand(command._id, {
+        status: 'done',
+        result: { profile_id, launched_at: new Date().toISOString() },
+      });
+    } catch (err) {
+      console.error(`[commands] Failed to launch profile: ${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}`);
+    // TODO: close Nstbrowser profile
+    await this.api.updateCommand(command._id, { status: 'done' });
+  }
+
+  async handleVerifyLogins(command) {
+    const { profile_id, channel_id } = command.payload || {};
+    console.log(`[commands] Verifying logins for profile: ${profile_id}`);
+    // TODO: launch headless, check cookies for Google/Facebook/TikTok
+    await this.api.updateCommand(command._id, { status: 'done' });
+  }
+}
+
+module.exports = { CommandPoller };

package/lib/daemon.js

@@ -1,6 +1,7 @@
 const { ApiClient } = require('./api-client');
 const { Heartbeat } = require('./heartbeat');
 const { JobPoller } = require('./job-poller');
+const { CommandPoller } = require('./command-poller');
 const os = require('os');
 
 class Daemon {
@@ -9,6 +10,7 @@ class Daemon {
     this.api = new ApiClient(config.api_url);
     this.heartbeat = new Heartbeat(this.api, config.worker_id);
     this.poller = new JobPoller(this.api, config);
+    this.commandPoller = new CommandPoller(this.api, config);
   }
 
   async start() {
@@ -45,13 +47,18 @@ class Daemon {
     // Start job polling
     this.poller.start();
     console.log('[daemon] Job poller started (every 5s)');
-    console.log('[daemon] Waiting for jobs...\n');
+
+    // Start command polling
+    this.commandPoller.start();
+    console.log('[daemon] Command poller started (every 3s)');
+    console.log('[daemon] Waiting for jobs & commands...\n');
 
     // Graceful shutdown
     const shutdown = async () => {
       console.log('\n[daemon] Shutting down...');
       this.heartbeat.stop();
       this.poller.stop();
+      this.commandPoller.stop();
 
       // Mark offline
       try {

package/package.json

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