channel-worker 1.0.18 → 1.0.20

package/lib/command-poller.js

@@ -49,6 +49,12 @@ class CommandPoller {
         case 'save_file':
           await this.handleSaveFile(command);
           break;
+        case 'set_thumbnail':
+          await this.handleSetThumbnail(command);
+          break;
+        case 'set_tags':
+          await this.handleSetTags(command);
+          break;
         default:
           // Other commands (scan_facebook_pages, etc.) handled by extension
           console.log(`[commands] Skipping ${command.type} — handled by extension`);
@@ -212,6 +218,231 @@ class CommandPoller {
     }
   }
 
+  async handleSetThumbnail(command) {
+    const { thumbnail_url, profile_id, file_path } = command.payload || {};
+    console.log(`[commands] Setting thumbnail for profile: ${profile_id}`);
+    try {
+      const fs = require('fs');
+      const path = require('path');
+      const WebSocket = require('ws');
+
+      // 1. Download thumbnail to disk
+      const dir = path.dirname(file_path);
+      if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true });
+      const res = await fetch(thumbnail_url);
+      if (!res.ok) throw new Error(`Download failed: ${res.status}`);
+      const buffer = Buffer.from(await res.arrayBuffer());
+      fs.writeFileSync(file_path, buffer);
+      console.log(`[commands] Thumbnail saved: ${file_path} (${buffer.length} bytes)`);
+
+      // 2. Find running browser for this profile
+      if (!this.nst) {
+        const apiKey = await this.api.getSetting('nst_api_key');
+        if (apiKey) this.nst = new NstManager(apiKey);
+      }
+      const browsersRes = await fetch('http://localhost:8848/api/v2/browsers', {
+        headers: { 'x-api-key': this.nst?.apiKey || '' },
+      });
+      const browsersData = await browsersRes.json();
+      const browsers = browsersData?.data || [];
+      const browser = browsers.find(b => b.name === profile_id) || browsers[0];
+      if (!browser) throw new Error('No running browser found');
+
+      const debugPort = browser.remoteDebuggingPort;
+      console.log(`[commands] Browser debug port: ${debugPort}`);
+
+      // 3. Get WebSocket URL for YouTube Studio tab
+      const pagesRes = await fetch(`http://localhost:${debugPort}/json/list`);
+      const pages = await pagesRes.json();
+      const studioPage = pages.find(p => p.url?.includes('studio.youtube.com') && p.type === 'page');
+      if (!studioPage) throw new Error('YouTube Studio tab not found');
+
+      const wsUrl = studioPage.webSocketDebuggerUrl;
+      console.log(`[commands] Connecting CDP: ${wsUrl}`);
+
+      // 4. Use CDP to set thumbnail file
+      const result = await new Promise((resolve, reject) => {
+        const ws = new WebSocket(wsUrl);
+        let msgId = 1;
+
+        function send(method, params = {}) {
+          const id = msgId++;
+          return new Promise((res, rej) => {
+            const handler = (data) => {
+              const msg = JSON.parse(data);
+              if (msg.id === id) {
+                ws.removeListener('message', handler);
+                if (msg.error) rej(new Error(msg.error.message));
+                else res(msg.result);
+              }
+            };
+            ws.on('message', handler);
+            ws.send(JSON.stringify({ id, method, params }));
+          });
+        }
+
+        ws.on('open', async () => {
+          try {
+            await send('DOM.enable');
+            const doc = await send('DOM.getDocument');
+            const node = await send('DOM.querySelector', {
+              nodeId: doc.root.nodeId,
+              selector: '#file-loader',
+            });
+
+            if (!node?.nodeId) {
+              ws.close();
+              resolve({ success: false, error: '#file-loader not found' });
+              return;
+            }
+
+            await send('DOM.setFileInputFiles', {
+              nodeId: node.nodeId,
+              files: [file_path],
+            });
+
+            // Trigger change event
+            await send('Runtime.evaluate', {
+              expression: `
+                const inp = document.querySelector('#file-loader');
+                if (inp) {
+                  inp.dispatchEvent(new Event('change', { bubbles: true, composed: true }));
+                  inp.dispatchEvent(new Event('input', { bubbles: true, composed: true }));
+                }
+                'triggered'
+              `,
+            });
+
+            ws.close();
+            resolve({ success: true, file_path });
+          } catch (e) {
+            ws.close();
+            resolve({ success: false, error: e.message });
+          }
+        });
+
+        ws.on('error', (e) => reject(e));
+        setTimeout(() => { ws.close(); reject(new Error('CDP timeout')); }, 15000);
+      });
+
+      console.log(`[commands] Thumbnail CDP result:`, result);
+      await this.api.updateCommand(command._id, {
+        status: result.success ? 'done' : 'failed',
+        result,
+        error: result.error || null,
+      });
+    } catch (err) {
+      console.error(`[commands] Set thumbnail failed: ${err.message}`);
+      await this.api.updateCommand(command._id, { status: 'failed', error: err.message });
+    }
+  }
+
+  async handleSetTags(command) {
+    const { tags, profile_id } = command.payload || {};
+    console.log(`[commands] Setting tags for profile: ${profile_id}`, tags);
+    try {
+      const WebSocket = require('ws');
+
+      // Find browser
+      if (!this.nst) {
+        const apiKey = await this.api.getSetting('nst_api_key');
+        if (apiKey) this.nst = new NstManager(apiKey);
+      }
+      const browsersRes = await fetch('http://localhost:8848/api/v2/browsers', {
+        headers: { 'x-api-key': this.nst?.apiKey || '' },
+      });
+      const browsersData = await browsersRes.json();
+      const browser = (browsersData?.data || []).find(b => b.name === profile_id) || browsersData?.data?.[0];
+      if (!browser) throw new Error('No running browser');
+
+      const pagesRes = await fetch(`http://localhost:${browser.remoteDebuggingPort}/json/list`);
+      const pages = await pagesRes.json();
+      const studioPage = pages.find(p => p.url?.includes('studio.youtube.com') && p.type === 'page');
+      if (!studioPage) throw new Error('YouTube Studio tab not found');
+
+      const tagStr = tags.join(', ');
+
+      const result = await new Promise((resolve, reject) => {
+        const ws = new WebSocket(studioPage.webSocketDebuggerUrl);
+        let msgId = 1;
+        function send(method, params = {}) {
+          const id = msgId++;
+          return new Promise((res, rej) => {
+            const handler = (data) => {
+              const msg = JSON.parse(data);
+              if (msg.id === id) { ws.removeListener('message', handler); msg.error ? rej(new Error(msg.error.message)) : res(msg.result); }
+            };
+            ws.on('message', handler);
+            ws.send(JSON.stringify({ id, method, params }));
+          });
+        }
+
+        ws.on('open', async () => {
+          try {
+            // Click "Show more" button
+            await send('Runtime.evaluate', {
+              expression: `
+                (function() {
+                  const texts = ['show more', 'hiện thêm', 'hiển thị thêm'];
+                  const allEls = document.querySelectorAll('*');
+                  for (const el of allEls) {
+                    if (el.children.length > 3) continue;
+                    const t = el.textContent?.trim()?.toLowerCase() || '';
+                    if (texts.some(x => t === x)) { el.click(); return 'clicked: ' + el.textContent?.trim(); }
+                  }
+                  return 'not_found';
+                })()
+              `,
+            });
+
+            // Wait for section to expand
+            await new Promise(r => setTimeout(r, 2000));
+
+            // Find and fill tags input — placeholder "Add tag"
+            const fillResult = await send('Runtime.evaluate', {
+              expression: `
+                (function() {
+                  // Find input with placeholder "Add tag" or aria-label containing "tag"
+                  const inputs = document.querySelectorAll('input');
+                  for (const inp of inputs) {
+                    const ph = (inp.placeholder || '').toLowerCase();
+                    const aria = (inp.getAttribute('aria-label') || '').toLowerCase();
+                    if (ph.includes('add tag') || ph.includes('thêm thẻ') || aria.includes('tag') || aria.includes('thẻ')) {
+                      inp.focus();
+                      inp.value = '${tagStr.replace(/'/g, "\\'")}';
+                      inp.dispatchEvent(new Event('input', { bubbles: true }));
+                      inp.dispatchEvent(new Event('change', { bubbles: true }));
+                      // Press Enter to confirm
+                      inp.dispatchEvent(new KeyboardEvent('keydown', { key: 'Enter', code: 'Enter', keyCode: 13, bubbles: true }));
+                      inp.dispatchEvent(new KeyboardEvent('keyup', { key: 'Enter', code: 'Enter', keyCode: 13, bubbles: true }));
+                      return 'tags_filled';
+                    }
+                  }
+                  return 'no_tag_input';
+                })()
+              `,
+            });
+
+            ws.close();
+            resolve({ success: true, result: fillResult?.result?.value });
+          } catch (e) { ws.close(); resolve({ success: false, error: e.message }); }
+        });
+        ws.on('error', (e) => reject(e));
+        setTimeout(() => { ws.close(); reject(new Error('CDP timeout')); }, 15000);
+      });
+
+      console.log(`[commands] Tags CDP result:`, result);
+      await this.api.updateCommand(command._id, {
+        status: result.success ? 'done' : 'failed',
+        result,
+        error: result.error || null,
+      });
+    } catch (err) {
+      console.error(`[commands] Set tags 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/package.json

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