channel-worker 1.1.9 → 1.3.0

package/bin/cli.js

@@ -128,6 +128,24 @@ if (cmd === 'pair') {
   const daemon = new Daemon(config);
   daemon.start();
 
+} else if (cmd === 'update') {
+  const { checkAndUpdate, getLocalVersion } = require('../lib/updater');
+  console.log(`[channel-worker] Current version: ${getLocalVersion()}`);
+  (async () => {
+    try {
+      const result = await checkAndUpdate({ autoRestart: false });
+      if (result.updated) {
+        console.log(`[channel-worker] Updated: ${result.local} → ${result.latest}`);
+        console.log('[channel-worker] Restart the daemon to use the new version.');
+      } else {
+        console.log(`[channel-worker] Already up to date (${result.local})`);
+      }
+    } catch (err) {
+      console.error(`[channel-worker] Update failed: ${err.message}`);
+      process.exit(1);
+    }
+  })();
+
 } else if (cmd === 'config') {
   const config = loadConfig();
   // Hide token in display
@@ -142,7 +160,8 @@ channel-worker — Channel Manager worker daemon
 Commands:
   pair     Pair with dashboard using a one-time code (recommended)
   init     Configure worker manually
-  start    Start the daemon
+  start    Start the daemon (auto-checks for updates every 5min)
+  update   Check and install updates manually
   config   Show current config
 
 Pairing (recommended):

package/lib/api-client.js

@@ -62,12 +62,32 @@ class ApiClient {
 
   // Commands
   async getNextCommand(workerId) {
-    return this.request('GET', `/workers/commands?worker_id=${workerId}`);
+    const workerTypes = 'launch_profile,close_profile,save_file,set_thumbnail,set_tags,set_file_input,click_and_upload,type_text,verify_logins,update_extension';
+    return this.request('GET', `/workers/commands?worker_id=${workerId}&types=${encodeURIComponent(workerTypes)}`);
   }
 
   async updateCommand(commandId, data) {
     return this.request('PUT', `/workers/commands/${commandId}`, data);
   }
+
+  // Extension download
+  async getExtensionVersion() {
+    const data = await this.request('GET', '/extension-download/version');
+    return data.version;
+  }
+
+  async downloadExtension(destPath) {
+    const url = `${this.baseUrl}/extension-download/zip`;
+    const res = await fetch(url, {
+      headers: { 'x-worker-token': this.workerToken },
+    });
+    if (!res.ok) throw new Error(`Download failed: ${res.status}`);
+
+    const fs = require('fs');
+    const buffer = Buffer.from(await res.arrayBuffer());
+    fs.writeFileSync(destPath, buffer);
+    return destPath;
+  }
 }
 
 module.exports = { ApiClient };

package/lib/command-poller.js

@@ -1,4 +1,5 @@
 const { NstManager } = require('./nst-manager');
+const { checkAndUpdateExtension } = require('./extension-updater');
 
 class CommandPoller {
   constructor(api, config) {
@@ -64,6 +65,9 @@ class CommandPoller {
         case 'type_text':
           await this.handleTypeText(command);
           break;
+        case 'update_extension':
+          await this.handleUpdateExtension(command);
+          break;
         default:
           // Other commands (scan_facebook_pages, etc.) handled by extension
           console.log(`[commands] Skipping ${command.type} — handled by extension`);
@@ -827,6 +831,25 @@ class CommandPoller {
     await this.api.updateCommand(command._id, { status: 'done' });
   }
 
+  async handleUpdateExtension(command) {
+    console.log('[commands] Updating extension...');
+    try {
+      if (!this.config.extension_path) {
+        throw new Error('No extension_path configured');
+      }
+      const result = await checkAndUpdateExtension(this.api, this.config.extension_path);
+      if (result.updated) {
+        console.log(`[commands] Extension updated: ${result.from || 'none'} → ${result.to}`);
+      } else {
+        console.log(`[commands] Extension already up to date (v${result.local})`);
+      }
+      await this.api.updateCommand(command._id, { status: 'done', result });
+    } catch (err) {
+      console.error(`[commands] Extension update failed: ${err.message}`);
+      await this.api.updateCommand(command._id, { status: 'failed', error: err.message });
+    }
+  }
+
   async handleVerifyLogins(command) {
     const { profile_id, channel_id } = command.payload || {};
     console.log(`[commands] Verifying logins for profile: ${profile_id}`);

package/lib/daemon.js

@@ -2,6 +2,8 @@ const { ApiClient } = require('./api-client');
 const { Heartbeat } = require('./heartbeat');
 const { JobPoller } = require('./job-poller');
 const { CommandPoller } = require('./command-poller');
+const { UpdateChecker, getLocalVersion } = require('./updater');
+const { checkAndUpdateExtension } = require('./extension-updater');
 
 class Daemon {
   constructor(config) {
@@ -10,12 +12,14 @@ class Daemon {
     this.heartbeat = new Heartbeat(this.api, config.worker_id);
     this.poller = new JobPoller(this.api, config);
     this.commandPoller = new CommandPoller(this.api, config);
+    this.updateChecker = new UpdateChecker(5 * 60 * 1000); // check every 5min
   }
 
   async start() {
+    const version = getLocalVersion();
     console.log(`
 ╔══════════════════════════════════════╗
-║       channel-worker v1.0.0          ║
+║       channel-worker v${version.padEnd(14)}║
 ╠══════════════════════════════════════╣
 ║  Worker ID:  ${this.config.worker_id.padEnd(22)}║
 ║  API URL:    ${this.config.api_url.padEnd(22)}║
@@ -31,6 +35,20 @@ class Daemon {
         max_concurrent: this.config.max_concurrent,
       });
       console.log('[daemon] Registered with dashboard ✓');
+
+      // Auto-update extension on startup
+      if (this.config.extension_path) {
+        try {
+          const result = await checkAndUpdateExtension(this.api, this.config.extension_path);
+          if (result.updated) {
+            console.log(`[daemon] Extension updated: ${result.from || 'none'} → ${result.to}`);
+          } else {
+            console.log(`[daemon] Extension up to date (v${result.local})`);
+          }
+        } catch (err) {
+          console.warn(`[daemon] Extension update check failed: ${err.message}`);
+        }
+      }
     } catch (err) {
       console.error(`[daemon] Failed to register: ${err.message}`);
       console.error('[daemon] Is the dashboard API running? Retrying in 10s...');
@@ -49,6 +67,10 @@ class Daemon {
     // Start command polling
     this.commandPoller.start();
     console.log('[daemon] Command poller started (every 3s)');
+
+    // Start auto-update checker
+    this.updateChecker.start();
+    console.log('[daemon] Auto-update checker started (every 5min)');
     console.log('[daemon] Waiting for jobs & commands...\n');
 
     // Graceful shutdown
@@ -57,6 +79,7 @@ class Daemon {
       this.heartbeat.stop();
       this.poller.stop();
       this.commandPoller.stop();
+      this.updateChecker.stop();
 
       // Mark offline
       try {

package/lib/extension-updater.js

@@ -0,0 +1,78 @@
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const { execSync } = require('child_process');
+
+function getLocalExtensionVersion(extensionPath) {
+  try {
+    const manifest = JSON.parse(fs.readFileSync(path.join(extensionPath, 'manifest.json'), 'utf8'));
+    return manifest.version || null;
+  } catch {
+    return null;
+  }
+}
+
+function isNewer(remote, local) {
+  const r = remote.split('.').map(Number);
+  const l = local.split('.').map(Number);
+  for (let i = 0; i < 3; i++) {
+    if ((r[i] || 0) > (l[i] || 0)) return true;
+    if ((r[i] || 0) < (l[i] || 0)) return false;
+  }
+  return false;
+}
+
+async function checkAndUpdateExtension(api, extensionPath) {
+  const local = getLocalExtensionVersion(extensionPath);
+  const remote = await api.getExtensionVersion();
+
+  if (local && !isNewer(remote, local)) {
+    return { updated: false, local, remote };
+  }
+
+  console.log(`[ext-updater] ${local ? `Update available: ${local} → ${remote}` : `Downloading extension v${remote}...`}`);
+
+  // Download to temp
+  const tmpArchive = path.join(os.tmpdir(), `channel-manager-ext-${Date.now()}.tar.gz`);
+  const tmpExtract = path.join(os.tmpdir(), `channel-manager-ext-new-${Date.now()}`);
+
+  try {
+    await api.downloadExtension(tmpArchive);
+
+    // Extract
+    fs.mkdirSync(tmpExtract, { recursive: true });
+    execSync(`tar -xzf "${tmpArchive}" -C "${tmpExtract}"`, { timeout: 30000 });
+
+    // tar extracts into a subfolder named "channel-manager-ext"
+    const extracted = path.join(tmpExtract, 'channel-manager-ext');
+    const sourceDir = fs.existsSync(extracted) ? extracted : tmpExtract;
+
+    // Replace existing extension dir
+    if (fs.existsSync(extensionPath)) {
+      fs.rmSync(extensionPath, { recursive: true, force: true });
+    }
+    fs.mkdirSync(extensionPath, { recursive: true });
+
+    // Copy files
+    const files = fs.readdirSync(sourceDir);
+    for (const f of files) {
+      const src = path.join(sourceDir, f);
+      const dest = path.join(extensionPath, f);
+      const stat = fs.statSync(src);
+      if (stat.isFile()) {
+        fs.copyFileSync(src, dest);
+      } else if (stat.isDirectory()) {
+        fs.cpSync(src, dest, { recursive: true });
+      }
+    }
+
+    console.log(`[ext-updater] Extension updated to v${remote}`);
+    return { updated: true, from: local, to: remote };
+  } finally {
+    // Cleanup temp files
+    try { fs.unlinkSync(tmpArchive); } catch {}
+    try { fs.rmSync(tmpExtract, { recursive: true, force: true }); } catch {}
+  }
+}
+
+module.exports = { checkAndUpdateExtension, getLocalExtensionVersion };

package/lib/updater.js

@@ -0,0 +1,93 @@
+const { execSync } = require('child_process');
+const path = require('path');
+
+const PKG_NAME = 'channel-worker';
+
+function getLocalVersion() {
+  const pkg = require(path.join(__dirname, '..', 'package.json'));
+  return pkg.version;
+}
+
+async function getLatestVersion() {
+  const res = await fetch(`https://registry.npmjs.org/${PKG_NAME}/latest`);
+  if (!res.ok) throw new Error(`npm registry returned ${res.status}`);
+  const data = await res.json();
+  return data.version;
+}
+
+function isNewer(remote, local) {
+  const r = remote.split('.').map(Number);
+  const l = local.split('.').map(Number);
+  for (let i = 0; i < 3; i++) {
+    if (r[i] > l[i]) return true;
+    if (r[i] < l[i]) return false;
+  }
+  return false;
+}
+
+function installUpdate(version) {
+  console.log(`[updater] Installing ${PKG_NAME}@${version}...`);
+  execSync(`npm install -g ${PKG_NAME}@${version}`, { stdio: 'inherit' });
+  console.log(`[updater] Installed ${PKG_NAME}@${version}`);
+}
+
+async function checkAndUpdate({ autoRestart = false } = {}) {
+  const local = getLocalVersion();
+  const latest = await getLatestVersion();
+
+  if (!isNewer(latest, local)) {
+    return { updated: false, local, latest };
+  }
+
+  console.log(`[updater] Update available: ${local} → ${latest}`);
+  installUpdate(latest);
+
+  if (autoRestart) {
+    console.log('[updater] Restarting daemon...');
+    // Spawn detached process with same args, then exit current
+    const { spawn } = require('child_process');
+    const child = spawn(process.argv[0], process.argv.slice(1), {
+      detached: true,
+      stdio: 'ignore',
+      cwd: process.cwd(),
+    });
+    child.unref();
+    process.exit(0);
+  }
+
+  return { updated: true, local, latest };
+}
+
+class UpdateChecker {
+  constructor(intervalMs = 5 * 60 * 1000) {
+    this.intervalMs = intervalMs;
+    this.timer = null;
+  }
+
+  start() {
+    // Check on startup after 10s delay
+    setTimeout(() => this.check(), 10000);
+    // Then periodically
+    this.timer = setInterval(() => this.check(), this.intervalMs);
+  }
+
+  stop() {
+    if (this.timer) {
+      clearInterval(this.timer);
+      this.timer = null;
+    }
+  }
+
+  async check() {
+    try {
+      const result = await checkAndUpdate({ autoRestart: true });
+      if (!result.updated) {
+        // Silent — only log in verbose or when there's an update
+      }
+    } catch (err) {
+      console.warn(`[updater] Check failed: ${err.message}`);
+    }
+  }
+}
+
+module.exports = { checkAndUpdate, getLocalVersion, getLatestVersion, isNewer, UpdateChecker };

package/package.json

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