channel-worker 1.0.19 → 1.0.21

package/lib/command-poller.js

@@ -52,6 +52,12 @@ class CommandPoller {
         case 'set_thumbnail':
           await this.handleSetThumbnail(command);
           break;
+        case 'set_tags':
+          await this.handleSetTags(command);
+          break;
+        case 'set_file_input':
+          await this.handleSetFileInput(command);
+          break;
         default:
           // Other commands (scan_facebook_pages, etc.) handled by extension
           console.log(`[commands] Skipping ${command.type} — handled by extension`);
@@ -334,6 +340,177 @@ class CommandPoller {
     }
   }
 
+  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 handleSetFileInput(command) {
+    const { profile_id, file_path, selector, fallback_selector, url_match } = command.payload || {};
+    console.log(`[commands] Setting file input: ${selector} → ${file_path}`);
+    try {
+      const WebSocket = require('ws');
+      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 browser = ((await browsersRes.json())?.data || []).find(b => b.name === profile_id) || (await browsersRes.json())?.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 targetPage = pages.find(p => p.type === 'page' && (!url_match || p.url?.includes(url_match)));
+      if (!targetPage) throw new Error(`No tab matching ${url_match}`);
+
+      const result = await new Promise((resolve, reject) => {
+        const ws = new WebSocket(targetPage.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 {
+            await send('DOM.enable');
+            const doc = await send('DOM.getDocument');
+            let node = await send('DOM.querySelector', { nodeId: doc.root.nodeId, selector });
+            if (!node?.nodeId && fallback_selector) {
+              node = await send('DOM.querySelector', { nodeId: doc.root.nodeId, selector: fallback_selector });
+            }
+            if (!node?.nodeId) { ws.close(); resolve({ success: false, error: 'Input not found' }); return; }
+
+            await send('DOM.setFileInputFiles', { nodeId: node.nodeId, files: [file_path] });
+            await send('Runtime.evaluate', {
+              expression: `document.querySelector('${selector.replace(/'/g, "\\'")}')?.dispatchEvent(new Event('change', { bubbles: true })); 'ok'`,
+            });
+
+            ws.close();
+            resolve({ success: true });
+          } catch (e) { ws.close(); resolve({ success: false, error: e.message }); }
+        });
+        ws.on('error', (e) => reject(e));
+        setTimeout(() => { ws.close(); reject(new Error('Timeout')); }, 15000);
+      });
+
+      await this.api.updateCommand(command._id, { status: result.success ? 'done' : 'failed', result, error: result.error || null });
+    } catch (err) {
+      console.error(`[commands] Set file input 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.19",
+  "version": "1.0.21",
   "description": "Channel Manager worker daemon — runs on remote machines to execute video pipeline jobs",
   "main": "lib/daemon.js",
   "bin": {