@voko/lite 0.3.1

package/src/server/wechat-handler.js

@@ -0,0 +1,439 @@
+/**
+ * VOKO Desktop - WeChat (iLink) 消息处理器
+ * 通过长轮询接收 WeChat 消息事件
+ *
+ * 用途：
+ * - 当 agent 需要主人介入时，通过微信通知主人
+ * - 主人回复后，将消息转发给对应 agent 的 session
+ */
+
+const https = require('https');
+
+class WechatHandler {
+  constructor(config, options = {}) {
+    // iLink 配置
+    this.baseUrl = config.baseUrl || 'https://ilinkai.weixin.qq.com';
+    this.botToken = config.botToken;
+    this.ownerUserId = config.ownerUserId;  // 主人微信 user_id
+    this.pollInterval = config.pollInterval || 35000;  // 长轮询超时 35秒
+
+    // 回调函数
+    this.onOwnerReply = options.onOwnerReply || null;
+    this.getInterventionByParentMsgId = options.getInterventionByParentMsgId || null;
+    this.getLatestPendingIntervention = options.getLatestPendingIntervention || null;
+    this.getPendingWechatCount = options.getPendingWechatCount || null;
+    this.isEnabled = options.isEnabled || (() => true);
+    this.onSessionExpired = options.onSessionExpired || null;
+
+    // 轮询状态
+    this.getUpdatesBuf = '';
+    this.pollTimer = null;
+    this.isPolling = false;
+    this.enabled = false;
+    this.pollIntervalMs = config.pollIntervalMs || 1000;  // 默认1秒
+
+    // 扫码登录状态
+    this.qrCode = null;
+    this.qrCodeToken = null;  // 保存 token 用于轮询
+    this.qrCodeStatus = null;
+    this.accountId = null;
+  }
+
+  async start() {
+    if (this.enabled) {
+      console.log('[Wechat] 已经启动');
+      return;
+    }
+
+    if (!this.botToken || !this.ownerUserId) {
+      console.log('[Wechat] 缺少 botToken 或 ownerUserId，跳过启动');
+      return;
+    }
+
+    this.enabled = true;
+    console.log('[Wechat] 启动长轮询，超时', this.pollInterval, 'ms');
+    this.scheduleNextPoll();
+  }
+
+  scheduleNextPoll() {
+    if (!this.enabled) return;
+
+    this.pollTimer = setTimeout(async () => {
+      await this.pollOnce();
+      this.scheduleNextPoll();
+    }, this.pollIntervalMs);  // 使用配置的时间间隔
+  }
+
+  async pollOnce() {
+    if (!this.isEnabled()) {
+      return;
+    }
+
+    const pendingCount = this.getPendingWechatCount ? this.getPendingWechatCount() : 1;
+    if (pendingCount === 0) {
+      return;
+    }
+
+    try {
+      this.isPolling = true;
+      await this.fetchUpdates();
+    } catch (err) {
+      console.error('[Wechat] 长轮询错误:', err.message);
+    } finally {
+      this.isPolling = false;
+    }
+  }
+
+  async fetchUpdates() {
+    return new Promise((resolve, reject) => {
+      const postData = JSON.stringify({
+        get_updates_buf: this.getUpdatesBuf || '',
+        base_info: {
+          channel_version: '1.0.2'
+        }
+      });
+
+      const url = new URL(`${this.baseUrl}/ilink/bot/getupdates`);
+      const options = {
+        hostname: url.hostname,
+        port: 443,
+        path: url.pathname,
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json',
+          'Authorization': `Bearer ${this.botToken}`,
+          'AuthorizationType': 'ilink_bot_token',
+          'Content-Length': Buffer.byteLength(postData)
+        }
+      };
+
+      const req = https.request(options, (res) => {
+        let data = '';
+        res.on('data', chunk => data += chunk);
+        res.on('end', () => {
+          console.log('[Wechat] getupdates response length:', data.length, 'content:', data.substring(0, 300));
+          try {
+            const json = JSON.parse(data);
+            // 如果没有 ret 字段，检查是否有 msgs 或 sync_buf（iLink 成功响应）
+            if (json.ret === 0 || (json.ret === undefined && (json.msgs || json.sync_buf))) {
+              if (json.msgs && json.msgs.length > 0) {
+                console.log('[Wechat] 收到消息, msgs count:', json.msgs.length);
+                this.getUpdatesBuf = json.get_updates_buf || '';
+                this.processUpdates(json.msgs);
+              } else {
+                // 更新游标，即使没有消息
+                this.getUpdatesBuf = json.get_updates_buf || this.getUpdatesBuf;
+              }
+              resolve();
+            } else if (json.ret == null) {
+              // ret 为空、undefined、null 都视为心跳/空响应，继续轮询
+              console.log('[Wechat] 长轮询空响应或心跳，继续轮询');
+              resolve();
+            } else if (json.ret === -14 || json.ret === -2) {
+              console.log('[Wechat] 会话过期或无效，准备重新登录, ret:', json.ret, 'errmsg:', json.errmsg);
+              if (this.onSessionExpired) {
+                this.onSessionExpired();
+              }
+              resolve();  // 不拒绝，继续下一次轮询
+            } else {
+              console.error('[Wechat] 长轮询 API 错误, ret:', json.ret, 'errmsg:', json.errmsg);
+              resolve();  // 其他错误也继续轮询
+            }
+          } catch (e) {
+            reject(e);
+          }
+        });
+      });
+
+      req.on('error', (err) => {
+        console.error('[Wechat] 长轮询请求失败:', err.message);
+        reject(err);
+      });
+
+      // 设置超时，避免长时间挂起
+      req.setTimeout(40000, () => {
+        req.destroy();
+        console.log('[Wechat] 长轮询超时，40秒');
+        resolve();  // 超时不算错误，继续下一次轮询
+      });
+
+      req.write(postData);
+      req.end();
+    });
+  }
+
+  // 检查是否是会话过期错误（需要重新登录）
+  isSessionExpiredError(ret, errmsg) {
+    // -14 = session expired, -2 = 也可能是会话问题
+    return ret === -14 || (ret === -2 && errmsg === 'unknown error');
+  }
+
+  async processUpdates(msgs) {
+    for (const msg of msgs) {
+      if (msg.message_type === 1) {  // USER 消息
+        await this.handleMessage(msg);
+      }
+    }
+  }
+
+  async handleMessage(msg) {
+    const fromUserId = msg.from_user_id;
+    const messageId = msg.message_id?.toString();
+    const text = this.extractText(msg);
+    const contextToken = msg.context_token;
+
+    console.log('[Wechat] 收到消息, context_token:', contextToken, 'messageId:', messageId, 'fromUserId:', fromUserId, 'text:', text);
+
+    // 只处理来自主人的消息
+    if (fromUserId !== this.ownerUserId) {
+      return;
+    }
+
+    if (contextToken) {
+      await this.handleReply(contextToken, text, messageId);
+    } else {
+      console.log('[Wechat] 消息无 context_token，跳过处理');
+    }
+  }
+
+  extractText(msg) {
+    if (!msg.item_list) return '';
+    for (const item of msg.item_list) {
+      if (item.type === 1 && item.text_item) {
+        return item.text_item.text || '';
+      }
+    }
+    return '';
+  }
+
+  async handleReply(contextToken, content, replyMessageId) {
+    if (this.getInterventionByParentMsgId) {
+      // 1. 优先尝试 context_token 精确匹配
+      let intervention = this.getInterventionByParentMsgId(contextToken);
+      if (intervention) {
+        console.log('[Wechat] context_token 精确匹配到干预记录:', intervention.id);
+        if (this.onOwnerReply) {
+          this.onOwnerReply(intervention, content, replyMessageId);
+        }
+        return;
+      }
+
+      // 2. Fallback: ownerUserId + 最新 pending 记录匹配
+      // iLink 的 context_token 是服务器生成的，与我们的 client_id 不同
+      // 因此无法精确匹配时，使用 fallback 机制
+      console.log('[Wechat] context_token 未匹配，尝试 fallback 匹配');
+      if (this.getLatestPendingIntervention) {
+        intervention = this.getLatestPendingIntervention();
+        if (intervention) {
+          console.log('[Wechat] fallback 匹配到最新 pending 记录:', intervention.id);
+          if (this.onOwnerReply) {
+            this.onOwnerReply(intervention, content, replyMessageId);
+          }
+          return;
+        }
+        console.log('[Wechat] 无 pending 记录，跳过');
+      } else {
+        console.log('[Wechat] getLatestPendingIntervention 未提供，跳过');
+      }
+    }
+  }
+
+  /**
+   * 发送消息给微信用户（带记录，用于后续匹配）
+   */
+  async sendMessageToOwnerWithTracking(content, visitorId, sessionKey) {
+    return new Promise((resolve, reject) => {
+      const clientId = `msg_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
+
+      const postData = JSON.stringify({
+        msg: {
+          from_user_id: '',
+          to_user_id: this.ownerUserId,
+          client_id: clientId,
+          message_type: 2,  // BOT
+          message_state: 2,  // FINISH
+          context_token: '',  // 主动推送时为空
+          item_list: [
+            { type: 1, text_item: { text: content } }
+          ]
+        },
+        base_info: {
+          channel_version: '1.0.2'
+        }
+      });
+
+      const url = new URL(`${this.baseUrl}/ilink/bot/sendmessage`);
+      const options = {
+        hostname: url.hostname,
+        port: 443,
+        path: url.pathname,
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json',
+          'Authorization': `Bearer ${this.botToken}`,
+          'AuthorizationType': 'ilink_bot_token',
+          'Content-Length': Buffer.byteLength(postData)
+        }
+      };
+
+      const req = https.request(options, (res) => {
+        let data = '';
+        res.on('data', chunk => data += chunk);
+        res.on('end', () => {
+          try {
+            const json = JSON.parse(data);
+            if (json.ret === 0) {
+              console.log('[Wechat] 发送消息给主人成功, clientId:', clientId);
+              resolve({ messageId: clientId, sentMessageId: clientId });
+            } else if (json.ret === -2) {
+              // 会话过期/无效，标记为永久失败，不重试
+              console.error('[Wechat] 发送消息失败, ret:', json.ret, 'errmsg:', json.errmsg, '(会话过期，不重试)');
+              reject(new Error('SESSION_EXPIRED: ' + (json.errmsg || 'session expired')));
+            } else {
+              console.error('[Wechat] 发送消息失败, ret:', json.ret, 'errmsg:', json.errmsg);
+              reject(new Error(json.errmsg || 'Wechat API error'));
+            }
+          } catch (e) {
+            reject(e);
+          }
+        });
+      });
+
+      req.on('error', (err) => {
+        console.error('[Wechat] 发送请求失败:', err.message);
+        reject(new Error('请求失败: ' + err.message));
+      });
+      req.write(postData);
+      req.end();
+    });
+  }
+
+  /**
+   * 获取登录二维码
+   */
+  async fetchQrCode() {
+    return new Promise((resolve, reject) => {
+      const url = new URL(`${this.baseUrl}/ilink/bot/get_bot_qrcode`);
+      url.searchParams.set('bot_type', '3');
+
+      const postData = JSON.stringify({
+        local_token_list: this.botToken ? [this.botToken] : []
+      });
+
+      const options = {
+        hostname: url.hostname,
+        port: 443,
+        path: url.pathname + url.search,
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json',
+          'Authorization': `Bearer ${this.botToken || ''}`,
+          'AuthorizationType': 'ilink_bot_token',
+          'Content-Length': Buffer.byteLength(postData)
+        }
+      };
+
+      const req = https.request(options, (res) => {
+        let data = '';
+        res.on('data', chunk => data += chunk);
+        res.on('end', () => {
+          try {
+            const json = JSON.parse(data);
+            // qrcode_img_content 是实际的图片 URL 或 base64 数据
+            // qrcode 只是 token，不是图片
+            if (json.ret === 0 && (json.qrcode_img_content || json.qrcode)) {
+              this.qrCodeToken = json.qrcode;  // 保存 token
+              this.qrCode = json.qrcode_img_content || json.qrcode;  // 返回 URL 或 token
+              resolve(this.qrCode);
+            } else {
+              reject(new Error(json.errmsg || '获取二维码失败'));
+            }
+          } catch (e) {
+            reject(e);
+          }
+        });
+      });
+
+      req.on('error', reject);
+      req.write(postData);
+      req.end();
+    });
+  }
+
+  /**
+   * 轮询扫码状态
+   */
+  async pollQrCodeStatus() {
+    if (!this.qrCodeToken) {
+      return { status: 'expired' };
+    }
+
+    return new Promise((resolve, reject) => {
+      const url = new URL(`${this.baseUrl}/ilink/bot/get_qrcode_status`);
+      url.searchParams.set('qrcode', this.qrCodeToken);
+
+      const options = {
+        hostname: url.hostname,
+        port: 443,
+        path: url.pathname + url.search,
+        method: 'GET',
+        headers: {
+          'Authorization': `Bearer ${this.botToken || ''}`,
+          'AuthorizationType': 'ilink_bot_token'
+        }
+      };
+
+      const req = https.request(options, (res) => {
+        let data = '';
+        res.on('data', chunk => data += chunk);
+        res.on('end', () => {
+          try {
+            const json = JSON.parse(data);
+            if (json.ret === 0) {
+              this.qrCodeStatus = json.status;
+              if (json.status === 'confirmed' && json.bot_token) {
+                this.botToken = json.bot_token;
+                this.accountId = json.ilink_bot_id || this.accountId;
+                this.ownerUserId = json.ilink_user_id || this.ownerUserId;
+                resolve({
+                  status: 'confirmed',
+                  bot_token: json.bot_token,
+                  owner_user_id: json.ilink_user_id,
+                  account_id: json.ilink_bot_id
+                });
+              } else {
+                resolve(json);
+              }
+            } else {
+              resolve({ status: 'expired' });
+            }
+          } catch (e) {
+            reject(e);
+          }
+        });
+      });
+
+      req.on('error', reject);
+      req.end();
+    });
+  }
+
+  stop() {
+    this.enabled = false;
+    if (this.pollTimer) {
+      clearTimeout(this.pollTimer);
+      this.pollTimer = null;
+    }
+    console.log('[Wechat] 轮询已停止');
+  }
+
+  updateConfig(config) {
+    if (config.baseUrl) this.baseUrl = config.baseUrl;
+    if (config.botToken) this.botToken = config.botToken;
+    if (config.ownerUserId) this.ownerUserId = config.ownerUserId;
+    if (config.pollInterval) this.pollInterval = config.pollInterval;
+    console.log('[Wechat] 配置已更新');
+  }
+}
+
+module.exports = WechatHandler;