@lightcone-ai/daemon 0.9.16 → 0.9.17

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lightcone-ai/daemon",
-  "version": "0.9.16",
+  "version": "0.9.17",
   "type": "module",
   "main": "src/index.js",
   "bin": {

package/src/agent-manager.js

@@ -5,7 +5,7 @@ import path from 'path';
 import { buildSystemPrompt } from './drivers/claude.js';
 import { buildCodexSpawn, buildCodexSystemPrompt, parseCodexLine } from './drivers/codex.js';
 import { buildKimiSpawn, buildKimiInitMessages, parseKimiLine, encodeKimiStdin } from './drivers/kimi.js';
-import { startSession, stopSession } from './xhs-login.js';
+import { startSession, stopSession, profileDir } from './browser-login.js';
 
 export class AgentManager {
   constructor({ serverUrl, machineApiKey }) {
@@ -26,8 +26,8 @@ export class AgentManager {
       case 'agent:start':       return this._startAgent(msg, connection);
       case 'agent:stop':        return this._stopAgent(msg.agentId, msg.teamId, connection);
       case 'agent:deliver':     return this._deliverMessage(msg, connection);
-      case 'xhs:start_login':   return this._startXhsLogin(msg, connection);
-      case 'xhs:stop_login':    return this._stopXhsLogin();
+      case 'browser:start_login': return this._startBrowserLogin(msg, connection);
+      case 'browser:stop_login':  return this._stopBrowserLogin(msg);
       case 'ping':              return connection.send({ type: 'pong' });
       default:
         console.log(`[AgentManager] Unhandled: ${msg.type}`);
@@ -237,7 +237,11 @@ export class AgentManager {
           if (a === '{platform_mcp_path}')
             return new URL('../../mcp-servers/platform/index.js', import.meta.url).pathname;
           if (a === '{xhs_profile_dir}')
-            return path.join(homedir(), '.lightcone', 'chrome-profiles', 'xhs');
+            return profileDir('xhs', config.userId ?? 'default');
+          if (a === '{douyin_profile_dir}')
+            return profileDir('douyin', config.userId ?? 'default');
+          if (a === '{kuaishou_profile_dir}')
+            return profileDir('kuaishou', config.userId ?? 'default');
           return a;
         });
         const resolvedEnv = {};
@@ -369,19 +373,21 @@ export class AgentManager {
     this._flushPending(key, connection);
   }
 
-  async _startXhsLogin(msg, connection) {
-    console.log('[AgentManager] Starting XHS login session');
+  async _startBrowserLogin(msg, connection) {
+    const platform = msg.platform ?? 'xhs';
+    console.log(`[AgentManager] Starting browser login for platform=${platform}`);
     try {
-      await startSession(connection, msg.userId ?? 'default');
+      await startSession(platform, connection, msg.userId ?? 'default');
     } catch (err) {
-      console.error('[AgentManager] XHS login start failed:', err.message);
-      connection.send({ type: 'xhs:login_error', error: err.message });
+      console.error(`[AgentManager] Browser login start failed (${platform}):`, err.message);
+      connection.send({ type: 'browser:login_error', platform, error: err.message });
     }
   }
 
-  _stopXhsLogin() {
-    console.log('[AgentManager] Stopping XHS login session');
-    stopSession();
+  _stopBrowserLogin(msg) {
+    const platform = msg.platform ?? 'xhs';
+    console.log(`[AgentManager] Stopping browser login for platform=${platform}`);
+    stopSession(platform);
   }
 
   _stopAgent(agentId, teamId, connection) {

package/src/{xhs-login.js → browser-login.js}

@@ -1,7 +1,8 @@
 /**
- * XHS (Xiaohongshu) login session via Chrome CDP.
- * Runs on the daemon machine (VPS) where Chrome is installed.
- * Screenshots are sent back to the server via the daemon WebSocket connection.
+ * Generic browser login via Chrome CDP.
+ * Supports multiple platforms (xhs, douyin, etc.)
+ * Runs on the daemon machine where Chrome is installed.
+ * Screenshots are sent back to server via daemon WebSocket.
  */
 import { spawn } from 'child_process';
 import { homedir } from 'os';
@@ -10,12 +11,33 @@ import path from 'path';
 import http from 'http';
 import { WebSocket } from 'ws';
 
-export function xhsProfileDir(userId) {
-  // User-scoped so multiple users on the same hosted machine don't conflict
-  return path.join(homedir(), '.lightcone', 'chrome-profiles', `xhs-${userId}`);
+// ── Platform configs ──────────────────────────────────────────────────────────
+
+export const PLATFORM_CONFIGS = {
+  xhs: {
+    loginUrl: 'https://www.xiaohongshu.com',
+    isLoggedIn: (cookies) => cookies.some(c => c.name === 'web_session' && c.value?.length > 0),
+  },
+  douyin: {
+    loginUrl: 'https://www.douyin.com',
+    isLoggedIn: (cookies) => cookies.some(c =>
+      (c.name === 'sessionid' || c.name === 'sid_guard') && c.value?.length > 0
+    ),
+  },
+  kuaishou: {
+    loginUrl: 'https://www.kuaishou.com',
+    isLoggedIn: (cookies) => cookies.some(c =>
+      (c.name === 'userId' || c.name === 'passToken') && c.value?.length > 0
+    ),
+  },
+};
+
+export function profileDir(platform, userId) {
+  return path.join(homedir(), '.lightcone', 'chrome-profiles', `${platform}-${userId}`);
 }
 
-const CDP_PORT = 9225;
+// ── Chrome detection ──────────────────────────────────────────────────────────
+
 function detectChrome() {
   if (process.env.CHROME_BIN) return process.env.CHROME_BIN;
   const candidates = [
@@ -31,10 +53,14 @@ function detectChrome() {
   }
   return candidates[0];
 }
+
 const CHROME_BIN = detectChrome();
+const CDP_PORT = 9225;
 const SCREENSHOT_INTERVAL_MS = 2000;
 const LOGIN_CHECK_INTERVAL_MS = 3000;
 
+// ── Helpers ───────────────────────────────────────────────────────────────────
+
 function httpGet(url) {
   return new Promise((resolve, reject) => {
     http.get(url, (res) => {
@@ -49,19 +75,25 @@ function sleep(ms) {
   return new Promise(r => setTimeout(r, ms));
 }
 
-export class XhsLoginSession {
-  constructor() {
+// ── BrowserLoginSession ───────────────────────────────────────────────────────
+
+export class BrowserLoginSession {
+  constructor(platform) {
+    this._platform = platform;
+    this._config = PLATFORM_CONFIGS[platform];
+    if (!this._config) throw new Error(`Unknown platform: ${platform}`);
     this._proc = null;
     this._ws = null;
     this._nextId = 1;
     this._pending = new Map();
     this._screenshotTimer = null;
     this._loginCheckTimer = null;
+    this._profileDir = null;
   }
 
   async start(connection, userId) {
-    this._profileDir = xhsProfileDir(userId);
-    // Spawn Chrome
+    this._profileDir = profileDir(this._platform, userId);
+
     this._proc = spawn(CHROME_BIN, [
       `--remote-debugging-port=${CDP_PORT}`,
       '--no-sandbox',
@@ -71,22 +103,18 @@ export class XhsLoginSession {
       '--window-size=800,900',
       '--disable-blink-features=AutomationControlled',
       '--disable-infobars',
-      '--excludeSwitches=enable-automation',
       '--user-agent=Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
       'about:blank',
-    ], {
-      stdio: 'ignore',
-      detached: false,
-    });
+    ], { stdio: 'ignore', detached: false });
 
     this._proc.on('error', (err) => {
-      console.error(`[XhsLogin] Chrome spawn error: ${err.message}`);
+      console.error(`[BrowserLogin][${this._platform}] Chrome spawn error: ${err.message}`);
       this._stopTimers();
-      connection.send({ type: 'xhs:login_error', error: `Chrome 启动失败: ${err.message}。请确认 Chrome 已安装（路径: ${CHROME_BIN}），或通过 CHROME_BIN 环境变量指定路径` });
+      connection.send({ type: 'browser:login_error', platform: this._platform, error: `Chrome 启动失败: ${err.message}` });
     });
 
     this._proc.on('exit', (code) => {
-      console.log(`[XhsLogin] Chrome exited (code=${code})`);
+      console.log(`[BrowserLogin][${this._platform}] Chrome exited (code=${code})`);
       this._stopTimers();
     });
 
@@ -102,7 +130,6 @@ export class XhsLoginSession {
     }
     if (!ready) throw new Error('Chrome CDP did not become ready in time');
 
-    // Get WebSocket debugger URL
     const pagesJson = await httpGet(`http://localhost:${CDP_PORT}/json`);
     const pages = JSON.parse(pagesJson);
     const page = pages.find(p => p.type === 'page') ?? pages[0];
@@ -112,13 +139,11 @@ export class XhsLoginSession {
 
     await this.send('Page.enable', {});
     await this.send('Network.enable', {});
-    // Hide automation fingerprint
     await this.send('Page.addScriptToEvaluateOnNewDocument', {
       source: `Object.defineProperty(navigator, 'webdriver', { get: () => undefined });`,
     });
-    await this.send('Page.navigate', { url: 'https://www.xiaohongshu.com' });
+    await this.send('Page.navigate', { url: this._config.loginUrl });
 
-    // Start sending screenshots and checking login
     this._startPolling(connection);
   }
 
@@ -141,10 +166,7 @@ export class XhsLoginSession {
       });
       ws.on('error', reject);
       ws.on('close', () => {
-        for (const [, entry] of this._pending) {
-          clearTimeout(entry.timer);
-          entry.reject(new Error('WebSocket closed'));
-        }
+        for (const [, entry] of this._pending) { clearTimeout(entry.timer); entry.reject(new Error('WebSocket closed')); }
         this._pending.clear();
         this._ws = null;
       });
@@ -155,10 +177,7 @@ export class XhsLoginSession {
     return new Promise((resolve, reject) => {
       if (!this._ws) return reject(new Error('WebSocket not connected'));
       const id = this._nextId++;
-      const timer = setTimeout(() => {
-        this._pending.delete(id);
-        reject(new Error(`CDP timeout for ${method}`));
-      }, 10_000);
+      const timer = setTimeout(() => { this._pending.delete(id); reject(new Error(`CDP timeout for ${method}`)); }, 10_000);
       this._pending.set(id, { resolve, reject, timer });
       this._ws.send(JSON.stringify({ id, method, params }));
     });
@@ -166,37 +185,34 @@ export class XhsLoginSession {
 
   async screenshot() {
     const result = await this.send('Page.captureScreenshot', { format: 'jpeg', quality: 70 });
-    return result.data; // base64 jpeg
+    return result.data;
   }
 
   async isLoggedIn() {
     const result = await this.send('Network.getAllCookies', {});
-    const cookies = result.cookies ?? [];
-    return cookies.some(c => c.name === 'web_session' && c.value?.length > 0);
+    return this._config.isLoggedIn(result.cookies ?? []);
   }
 
   _startPolling(connection) {
-    // Send screenshots periodically
     this._screenshotTimer = setInterval(async () => {
       try {
         const screenshot = await this.screenshot();
-        connection.send({ type: 'xhs:screenshot', screenshot });
+        connection.send({ type: 'browser:screenshot', platform: this._platform, screenshot });
       } catch (err) {
-        console.error('[XhsLogin] Screenshot error:', err.message);
+        console.error(`[BrowserLogin][${this._platform}] Screenshot error:`, err.message);
       }
     }, SCREENSHOT_INTERVAL_MS);
 
-    // Check login status periodically
     this._loginCheckTimer = setInterval(async () => {
       try {
         const loggedIn = await this.isLoggedIn();
         if (loggedIn) {
           this._stopTimers();
-          connection.send({ type: 'xhs:login_complete', profileDir: this._profileDir });
+          connection.send({ type: 'browser:login_complete', platform: this._platform, profileDir: this._profileDir });
           this.close();
         }
       } catch (err) {
-        console.error('[XhsLogin] Login check error:', err.message);
+        console.error(`[BrowserLogin][${this._platform}] Login check error:`, err.message);
       }
     }, LOGIN_CHECK_INTERVAL_MS);
   }
@@ -208,29 +224,27 @@ export class XhsLoginSession {
 
   close() {
     this._stopTimers();
-    if (this._ws) {
-      try { this._ws.close(); } catch { /* ignore */ }
-      this._ws = null;
-    }
-    if (this._proc) {
-      try { this._proc.kill('SIGKILL'); } catch { /* ignore */ }
-      this._proc = null;
-    }
+    if (this._ws) { try { this._ws.close(); } catch {} this._ws = null; }
+    if (this._proc) { try { this._proc.kill('SIGKILL'); } catch {} this._proc = null; }
   }
 }
 
-// Singleton
-let _session = null;
+// ── Singleton map (platform → session) ───────────────────────────────────────
+
+const _sessions = new Map();
 
-export function getSession() { return _session; }
+export function getSession(platform) { return _sessions.get(platform) ?? null; }
 
-export async function startSession(connection, userId) {
-  if (_session) { _session.close(); _session = null; }
-  _session = new XhsLoginSession();
-  await _session.start(connection, userId);
-  return _session;
+export async function startSession(platform, connection, userId) {
+  const existing = _sessions.get(platform);
+  if (existing) { existing.close(); _sessions.delete(platform); }
+  const session = new BrowserLoginSession(platform);
+  _sessions.set(platform, session);
+  await session.start(connection, userId);
+  return session;
 }
 
-export function stopSession() {
-  if (_session) { _session.close(); _session = null; }
+export function stopSession(platform) {
+  const session = _sessions.get(platform);
+  if (session) { session.close(); _sessions.delete(platform); }
 }