@voko/lite 0.3.1

package/src/workers/agent-worker.js

@@ -0,0 +1,378 @@
+/**
+ * Agent Worker - 每个 Agent 的独立 WuKongIM 连接进程
+ *
+ * 通过 fork 启动，每个 worker 有独立的 V8 实例和 WKSDK.shared() 实例
+ * 用法: node agent-worker.js <agentId> <configJson>
+ */
+
+const fs = require('fs');
+const path = require('path');
+
+// Worker 专用错误日志（独立于主进程日志，避免编码问题）
+// 优先使用主进程传入的 WORKER_LOG_PATH 环境变量，兼容旧版硬编码路径
+const workerLogPath = process.env.WORKER_LOG_PATH || path.join(require('os').homedir(), 'AppData', 'Roaming', 'voko', 'agent-worker.log');
+function formatTime() {
+  const now = new Date();
+  const y = now.getFullYear();
+  const m = String(now.getMonth() + 1).padStart(2, '0');
+  const d = String(now.getDate()).padStart(2, '0');
+  const h = String(now.getHours()).padStart(2, '0');
+  const min = String(now.getMinutes()).padStart(2, '0');
+  const s = String(now.getSeconds()).padStart(2, '0');
+  return `${y}-${m}-${d} ${h}:${min}:${s}`;
+}
+function workerLog(...args) {
+  const line = `[${formatTime()}] ` + args.map(a => typeof a === 'object' ? JSON.stringify(a) : String(a)).join(' ') + '\n';
+  try { fs.appendFileSync(workerLogPath, line); } catch (e) { /* ignore */ }
+  console.log(...args);
+}
+function workerErr(...args) {
+  const line = `[${formatTime()}] [ERR] ` + args.map(a => typeof a === 'object' ? JSON.stringify(a) : String(a)).join(' ') + '\n';
+  try { fs.appendFileSync(workerLogPath, line); } catch (e) { /* ignore */ }
+  console.error(...args);
+}
+
+// 获取命令行参数
+const agentId = process.argv[2];
+let config = {};
+try {
+  config = JSON.parse(process.argv[3] || '{}');
+} catch (e) {
+  workerErr(`[${agentId || 'unknown'}] 解析 config JSON 失败:`, e.message, '原始参数:', process.argv[3]);
+  process.exit(1);
+}
+
+const { uid, token, serverUrl } = config;
+
+
+if (!agentId || !uid || !token || !serverUrl) {
+  workerErr(`[${agentId || 'unknown'}] Worker 启动参数不足`, { agentId, uid, token, serverUrl });
+  process.exit(1);
+}
+
+// Node.js 环境没有全局 WebSocket，用 ws 模块 polyfill
+if (typeof WebSocket === 'undefined') {
+  try {
+    const WS = require('ws');
+    global.WebSocket = WS;
+  } catch (err) {
+    workerErr(`[${agentId}] 加载 ws 模块失败:`, err.message, err.stack);
+    process.exit(1);
+  }
+} else {
+}
+
+// 加载 SDK
+let WKSDK, MessageText, MessageImage, Channel, ChannelTypePerson;
+let MessageFile;
+try {
+  const wukongim = require('wukongimjssdk');
+  WKSDK = wukongim.WKSDK;
+  MessageText = wukongim.MessageText;
+  MessageImage = wukongim.MessageImage;
+  Channel = wukongim.Channel;
+  ChannelTypePerson = wukongim.ChannelTypePerson;
+  try {
+    MessageFile = require('./message-content').MessageFile;
+  } catch (err) {
+    workerErr(`[${agentId}] 加载 MessageFile 失败:`, err.message);
+  }
+} catch (err) {
+  workerErr(`[${agentId}] 加载 wukongimjssdk 失败:`, err.message, err.stack);
+  process.exit(1);
+}
+
+// 初始化 SDK
+let sdk;
+try {
+  sdk = WKSDK.shared();
+  sdk.config.uid = uid;
+  sdk.config.token = token;
+  sdk.config.addr = serverUrl;
+  sdk.config.autoReconnect = true; // SDK 内置自动重连，退出手动管理避免冲突
+
+  // 注册自定义文件消息类型（contentType=4），确保收发都能正确编解码
+  if (MessageFile) {
+    try {
+      WKSDK.shared().messageContentManager.register(4, MessageFile);
+    } catch (err) {
+      workerErr(`[${agentId}] 注册 MessageFile 失败:`, err.message);
+    }
+  }
+} catch (err) {
+  workerErr(`[${agentId}] WKSDK 初始化失败:`, err.message, err.stack);
+  process.exit(1);
+}
+
+// 状态映射（与 SDK ConnectStatus 枚举一致）
+// ConnectStatus: 0=Disconnect 1=Connected 2=Connecting 3=ConnectFail 4=ConnectKick
+const statusMap = {
+  0: 'disconnected',
+  1: 'connected',
+  2: 'connecting',
+  3: 'connectFail',
+  4: 'kicked'
+};
+
+// 连接状态监听（仅上报，重连由 SDK 内置的 autoReconnect 处理）
+sdk.connectManager.addConnectStatusListener((status) => {
+  const statusText = statusMap[status] || 'unknown';
+  workerLog(`[${agentId}] 连接状态: ${statusText}`);
+  process.send({
+    type: 'status',
+    agentId,
+    status: statusText,
+    statusCode: status
+  });
+});
+
+// 消息监听
+sdk.chatManager.addMessageListener((rawMsg) => {
+  // 如果是自己发的消息，跳过（避免回声）
+  if (rawMsg.fromUID === uid) {
+    return;
+  }
+
+  // contentType 99 是命令消息，跳过不处理
+  if (rawMsg.contentType === 99) {
+    return;
+  }
+
+
+  // 提取消息字段
+  const fromUid = rawMsg.fromUID;
+  const toUid = rawMsg.toUID || uid;
+  const channelId = rawMsg.channel?.channelID || rawMsg.channelID;
+  const channelType = rawMsg.channelType ?? 1;
+  const msgId = rawMsg.messageID || 'unknown';
+  const timestamp = rawMsg.timestamp || Date.now() / 1000;
+
+  // 根据 contentType 提取内容
+  let content = '';
+  let contentObj = rawMsg.content;
+  // 如果 content 是 { contentObj: { url, name, size, type } } 结构，提取出来
+  if (rawMsg.content?.contentObj) {
+    contentObj = rawMsg.content.contentObj;
+  }
+  const contentType = rawMsg.contentType ?? 1;
+
+  // 检测是否是文件（有 name, url, size 字段的对象）
+  const isFileObject = contentObj?.name && contentObj?.url;
+
+  if (contentType === 4 || isFileObject) {
+    // 文件消息：统一提取 name, url, size, type 用于图形化显示
+    content = JSON.stringify({
+      name: contentObj?.name || contentObj?.fileName || '',
+      url: contentObj?.url || '',
+      size: contentObj?.size || 0,
+      type: contentObj?.type || ''
+    });
+  } else if (contentType === 1) {
+    // 文本消息 - 支持 text 和 content 字段
+    content = contentObj?.text || contentObj?.content || '';
+  } else if (contentType === 2) {
+    // 图片消息
+    content = contentObj?.url || rawMsg.content?.url || '';
+  } else if (contentType === 3) {
+    // 视频消息
+    content = contentObj?.url || '[视频]';
+  } else {
+    // 其他类型，尝试提取有意义的字段
+    const c = rawMsg.content;
+    if (typeof c === 'object' && c !== null) {
+      content = c.fileName || c.url || c.text || JSON.stringify(c);
+    } else {
+      content = String(c || '');
+    }
+  }
+  // 确保 content 是字符串
+  if (typeof content !== 'string') {
+    content = JSON.stringify(content);
+  }
+
+  process.send({
+    type: 'message',
+    agentId,
+    data: {
+      fromUid,
+      toUid,
+      channelId,
+      channelType,
+      content: typeof content === 'string' ? content : String(content),
+      contentType,
+      messageId: msgId,
+      timestamp,
+      messageSeq: rawMsg.messageSeq,
+      clientMsgNo: rawMsg.clientMsgNo,
+      noPersist: rawMsg.header?.noPersist ? 1 : 0,
+      redDot: rawMsg.header?.reddot ? 1 : 0,
+      syncOnce: rawMsg.header?.syncOnce ? 1 : 0
+    }
+  });
+});
+
+// 响应主进程的主动探测
+process.on('message', (msg) => {
+  if (msg && msg.type === 'ping') {
+    const st = sdk.connectManager.status;
+    const connected = st === 1;
+    process.send({ type: 'pong', agentId, connected, statusCode: st });
+  }
+});
+
+// 连接
+try {
+  sdk.connect();
+} catch (err) {
+  workerErr(`[${agentId}] sdk.connect() 失败:`, err.message, err.stack);
+  process.exit(1);
+}
+
+// SDK bug: connectManager 连接成功时不触发 notifyConnectStatusListeners(1)
+// 持续轮询 SDK 内部状态，状态变化时实时上报给主进程
+// 不设超时：SDK autoReconnect 自动重连后仍能检测到
+let lastStatus = -1;
+const connCheckTimer = setInterval(() => {
+  const st = sdk.connectManager.status;
+  if (st === lastStatus) return; // 状态未变，跳过
+  lastStatus = st;
+
+  if (st === 1) { // ConnectStatus.Connected
+    workerLog(`[${agentId}] 连接状态: connected`);
+    process.send({ type: 'status', agentId, status: 'connected', statusCode: 1 });
+  } else if (st === 0) { // Disconnect
+    workerLog(`[${agentId}] 连接状态: disconnected`);
+    process.send({ type: 'status', agentId, status: 'disconnected', statusCode: 0 });
+  } else if (st === 3) { // ConnectFail
+    workerLog(`[${agentId}] 连接状态: connect_fail`);
+    process.send({ type: 'status', agentId, status: 'connect_fail', statusCode: 3 });
+  } else if (st === 4) { // ConnectKick — SDK 不会自动重连，停止轮询
+    workerLog(`[${agentId}] 连接状态: kicked`);
+    process.send({ type: 'status', agentId, status: 'kicked', statusCode: 4 });
+    clearInterval(connCheckTimer);
+  }
+}, 500);
+
+// 等待服务端回执的发送队列（clientSeq → 消息元数据）
+const pendingSends = new Map();
+
+// 监听发送回执（服务端确认后提供真实 messageSeq）
+sdk.chatManager.addMessageStatusListener((sendack) => {
+  const pending = pendingSends.get(sendack.clientSeq);
+  if (pending) {
+    process.send({
+      type: 'sent',
+      agentId: pending.agentId,
+      channelId: pending.channelId,
+      localMsgId: pending.localMsgId,
+      messageId: sendack.messageID,
+      messageSeq: sendack.messageSeq,
+      clientMsgNo: pending.clientMsgNo,
+      success: true
+    });
+    pendingSends.delete(sendack.clientSeq);
+  }
+});
+
+// 接收主进程指令
+process.on('message', (msg) => {
+  console.log(`[${agentId}] 收到指令:`, JSON.stringify(msg).substring(0, 100));
+
+  if (msg.type === 'send') {
+    // 发送消息
+    const { channelId, content, messageType, localMsgId } = msg;
+    (async () => {
+      try {
+        const channel = new Channel(channelId, ChannelTypePerson);
+        let message;
+        if (messageType === 'image') {
+          // 发送图片（远程 URL，不上传）
+          message = new MessageImage();
+          message.url = content;
+        } else if (messageType === 'file') {
+          // 发送文件消息：content 为 JSON 字符串 {url, name, size, type}
+          if (MessageFile) {
+            let filePayload = {};
+            try {
+              filePayload = JSON.parse(content || '{}');
+            } catch (_) { /* 非 JSON 则视为 url */ }
+            message = new MessageFile();
+            message.url = filePayload.url || content;
+            message.name = filePayload.name || '';
+            message.size = filePayload.size || 0;
+            message.type = filePayload.type || '';
+          } else {
+            // 兜底：按文本发送 JSON（兼容旧版本）
+            message = new MessageText(content);
+          }
+        } else {
+          // 发送文本
+          message = new MessageText(content);
+        }
+        const result = await sdk.chatManager.send(message, channel);
+
+        // SDK 在 send() 内部已生成 clientMsgNo 和 clientSeq
+        const clientSeq = result.clientSeq;
+        const clientMsgNo = result.clientMsgNo || null;
+
+        // 加入确认队列（等待服务端回执后更新 messageSeq）
+        pendingSends.set(clientSeq, { agentId, channelId, localMsgId: localMsgId || result.messageID, clientMsgNo });
+
+        // 立即返回（clientMsgNo 已知，messageSeq 待回执）
+        process.send({
+          type: 'sent',
+          agentId,
+          channelId,
+          localMsgId: localMsgId || result.messageID,
+          messageId: result.messageID,
+          messageSeq: null,
+          clientMsgNo,
+          success: true
+        });
+      } catch (e) {
+        workerErr(`[${agentId}] 消息发送失败:`, e);
+        process.send({
+          type: 'sent',
+          agentId,
+          channelId,
+          success: false,
+          error: e.message
+        });
+      }
+    })();
+  }
+
+  if (msg.type === 'disconnect') {
+    sdk.disconnect();
+  }
+});
+
+// 进程退出时断开连接
+process.on('exit', () => {
+  sdk.disconnect();
+});
+
+process.on('uncaughtException', (err) => {
+  workerErr(`[${agentId}] 未捕获异常:`, err.message, err.stack);
+});
+
+process.on('unhandledRejection', (reason, promise) => {
+  workerErr(`[${agentId}] 未处理的 Promise 拒绝:`, reason);
+});
+
+// 定期检查父进程是否还活着（Windows 上 kill 不触发 SIGTERM，需主动检测）
+if (process.platform === 'win32') {
+  const parentCheck = setInterval(() => {
+    try {
+      // 信号 0 = 仅探测进程是否存在，不发送实际信号
+      process.kill(process.ppid, 0);
+    } catch (_) {
+      workerLog(`[${agentId}] 主进程已退出，worker 自动终止`);
+      clearInterval(parentCheck);
+      sdk.disconnect();
+      process.exit(0);
+    }
+  }, 3000);
+}
+
+workerLog(`[${agentId}] Worker 已启动，等待连接...`);

package/src/workers/message-content.js

@@ -0,0 +1,51 @@
+/**
+ * 自定义 IM 消息内容类
+ *
+ * WuKongIM JS SDK 内置：MessageText(1)、MessageImage(2)、MessageVideo(3) 等。
+ * 文件消息（contentType=4）没有现成类，因此参考 chatroom 项目实现自定义 MessageFile。
+ */
+
+const { MessageContent } = require('wukongimjssdk');
+
+// WuKongIM SDK 当前版本未导出 file 常量，文件消息统一使用 contentType = 4
+const MESSAGE_CONTENT_TYPE_FILE = 4;
+
+/**
+ * 文件消息内容类
+ * encodeJSON / decodeJSON 会被 SDK 用于序列化/反序列化 payload
+ */
+class MessageFile extends MessageContent {
+  constructor() {
+    super();
+    this.url = '';
+    this.name = '';
+    this.size = 0;
+    this.type = '';
+  }
+
+  get contentType() {
+    return MESSAGE_CONTENT_TYPE_FILE;
+  }
+
+  encodeJSON() {
+    return {
+      url: this.url,
+      name: this.name,
+      size: this.size,
+      type: this.type
+    };
+  }
+
+  decodeJSON(json) {
+    this.url = json.url || '';
+    this.name = json.name || '';
+    this.size = json.size || 0;
+    this.type = json.type || '';
+  }
+
+  get conversationDigest() {
+    return `[文件] ${this.name || ''}`;
+  }
+}
+
+module.exports = { MessageFile };