@wecom/wecom-openclaw-plugin 2026.4.2 → 2026.4.3-beta.0

package/dist/cjs/src/state-manager.js

@@ -1,158 +0,0 @@
-'use strict';
-
-var reqidStore = require('./reqid-store.js');
-var _const = require('./const.js');
-
-/**
- * 企业微信全局状态管理模块
- *
- * 负责管理 WSClient 实例、消息状态（带 TTL 清理）、ReqId 存储
- * 解决全局 Map 的内存泄漏问题
- */
-// ============================================================================
-// WSClient 实例管理
-// ============================================================================
-/** WSClient 实例管理 */
-const wsClientInstances = new Map();
-/**
- * 获取指定账户的 WSClient 实例
- */
-function getWeComWebSocket(accountId) {
-    return wsClientInstances.get(accountId) ?? null;
-}
-/**
- * 设置指定账户的 WSClient 实例
- */
-function setWeComWebSocket(accountId, client) {
-    wsClientInstances.set(accountId, client);
-}
-/** 消息状态管理 */
-const messageStates = new Map();
-/** 定期清理定时器 */
-let cleanupTimer = null;
-/**
- * 启动消息状态定期清理（自动 TTL 清理 + 容量限制）
- */
-function startMessageStateCleanup() {
-    if (cleanupTimer)
-        return;
-    cleanupTimer = setInterval(() => {
-        pruneMessageStates();
-    }, _const.MESSAGE_STATE_CLEANUP_INTERVAL_MS);
-    // 允许进程退出时不阻塞
-    if (cleanupTimer && typeof cleanupTimer === "object" && "unref" in cleanupTimer) {
-        cleanupTimer.unref();
-    }
-}
-/**
- * 停止消息状态定期清理
- */
-function stopMessageStateCleanup() {
-    if (cleanupTimer) {
-        clearInterval(cleanupTimer);
-        cleanupTimer = null;
-    }
-}
-/**
- * 清理过期和超量的消息状态条目
- */
-function pruneMessageStates() {
-    const now = Date.now();
-    // 1. 清理过期条目
-    for (const [key, entry] of messageStates) {
-        if (now - entry.createdAt >= _const.MESSAGE_STATE_TTL_MS) {
-            messageStates.delete(key);
-        }
-    }
-    // 2. 容量限制：如果仍超过最大条目数，按时间淘汰最旧的
-    if (messageStates.size > _const.MESSAGE_STATE_MAX_SIZE) {
-        const sorted = [...messageStates.entries()].sort((a, b) => a[1].createdAt - b[1].createdAt);
-        const toRemove = sorted.slice(0, messageStates.size - _const.MESSAGE_STATE_MAX_SIZE);
-        for (const [key] of toRemove) {
-            messageStates.delete(key);
-        }
-    }
-}
-/**
- * 设置消息状态
- */
-function setMessageState(messageId, state) {
-    messageStates.set(messageId, {
-        state,
-        createdAt: Date.now(),
-    });
-}
-/**
- * 删除消息状态
- */
-function deleteMessageState(messageId) {
-    messageStates.delete(messageId);
-}
-// ============================================================================
-// ReqId 持久化存储管理（按 accountId 隔离）
-// ============================================================================
-/**
- * ReqId 持久化存储管理
- * 参考 createPersistentDedupe 模式：内存 + 磁盘双层、文件锁、原子写入、TTL 过期、防抖写入
- * 重启后可从磁盘恢复，确保主动推送消息时能获取到 reqId
- */
-const reqIdStores = new Map();
-function getOrCreateReqIdStore(accountId) {
-    let store = reqIdStores.get(accountId);
-    if (!store) {
-        store = reqidStore.createPersistentReqIdStore();
-        reqIdStores.set(accountId, store);
-    }
-    return store;
-}
-// ============================================================================
-// ReqId 操作函数
-// ============================================================================
-/**
- * 设置 chatId 对应的 reqId（写入内存 + 防抖写磁盘）
- */
-function setReqIdForChat(chatId, reqId, accountId = "default") {
-    getOrCreateReqIdStore(accountId).set(chatId, reqId);
-}
-/**
- * 启动时预热 reqId 缓存（从磁盘加载到内存）
- *
- * 注意：由于移除了磁盘存储，此函数现在只返回 0（无预热条目）
- */
-async function warmupReqIdStore(accountId = "default", log) {
-    // 由于移除了磁盘存储，不再需要预热过程
-    log?.("[WeCom] reqid-store warmup: no-op (disk storage removed)");
-    return 0;
-}
-// ============================================================================
-// 全局 cleanup（断开连接时释放所有资源）
-// ============================================================================
-/**
- * 清理指定账户的所有资源
- */
-async function cleanupAccount(accountId) {
-    // 1. 断开 WSClient
-    const wsClient = wsClientInstances.get(accountId);
-    if (wsClient) {
-        try {
-            wsClient.disconnect();
-        }
-        catch {
-            // 忽略断开连接时的错误
-        }
-        wsClientInstances.delete(accountId);
-    }
-    // 2. 由于移除了磁盘存储，不再需要 flush reqId 存储
-    // 注意：不删除 store，因为重连后可能还需要
-}
-
-exports.cleanupAccount = cleanupAccount;
-exports.deleteMessageState = deleteMessageState;
-exports.getWeComWebSocket = getWeComWebSocket;
-exports.setMessageState = setMessageState;
-exports.setReqIdForChat = setReqIdForChat;
-exports.setWeComWebSocket = setWeComWebSocket;
-exports.startMessageStateCleanup = startMessageStateCleanup;
-exports.stopMessageStateCleanup = stopMessageStateCleanup;
-exports.warmupReqIdStore = warmupReqIdStore;
-//# sourceMappingURL=state-manager.js.map

package/dist/cjs/src/state-manager.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"state-manager.js","sources":["../../../../../src/state-manager.ts"],"sourcesContent":["/**\n * 企业微信全局状态管理模块\n *\n * 负责管理 WSClient 实例、消息状态（带 TTL 清理）、ReqId 存储\n * 解决全局 Map 的内存泄漏问题\n */\n\nimport type { WSClient } from \"@wecom/aibot-node-sdk\";\nimport type { MessageState } from \"./interface.js\";\nimport { createPersistentReqIdStore, type PersistentReqIdStore } from \"./reqid-store.js\";\nimport {\n  MESSAGE_STATE_TTL_MS,\n  MESSAGE_STATE_CLEANUP_INTERVAL_MS,\n  MESSAGE_STATE_MAX_SIZE,\n} from \"./const.js\";\n\n// ============================================================================\n// WSClient 实例管理\n// ============================================================================\n\n/** WSClient 实例管理 */\nconst wsClientInstances = new Map<string, WSClient>();\n\n/**\n * 获取指定账户的 WSClient 实例\n */\nexport function getWeComWebSocket(accountId: string): WSClient | null {\n  return wsClientInstances.get(accountId) ?? null;\n}\n\n/**\n * 设置指定账户的 WSClient 实例\n */\nexport function setWeComWebSocket(accountId: string, client: WSClient): void {\n  wsClientInstances.set(accountId, client);\n}\n\n/**\n * 删除指定账户的 WSClient 实例\n */\nexport function deleteWeComWebSocket(accountId: string): void {\n  wsClientInstances.delete(accountId);\n}\n\n// ============================================================================\n// 消息状态管理（带 TTL 清理，防止内存泄漏）\n// ============================================================================\n\n/** 消息状态条目（带创建时间戳，用于 TTL 清理） */\ninterface MessageStateEntry {\n  state: MessageState;\n  createdAt: number;\n}\n\n/** 消息状态管理 */\nconst messageStates = new Map<string, MessageStateEntry>();\n\n/** 定期清理定时器 */\nlet cleanupTimer: ReturnType<typeof setInterval> | null = null;\n\n/**\n * 启动消息状态定期清理（自动 TTL 清理 + 容量限制）\n */\nexport function startMessageStateCleanup(): void {\n  if (cleanupTimer) return;\n\n  cleanupTimer = setInterval(() => {\n    pruneMessageStates();\n  }, MESSAGE_STATE_CLEANUP_INTERVAL_MS);\n\n  // 允许进程退出时不阻塞\n  if (cleanupTimer && typeof cleanupTimer === \"object\" && \"unref\" in cleanupTimer) {\n    cleanupTimer.unref();\n  }\n}\n\n/**\n * 停止消息状态定期清理\n */\nexport function stopMessageStateCleanup(): void {\n  if (cleanupTimer) {\n    clearInterval(cleanupTimer);\n    cleanupTimer = null;\n  }\n}\n\n/**\n * 清理过期和超量的消息状态条目\n */\nfunction pruneMessageStates(): void {\n  const now = Date.now();\n\n  // 1. 清理过期条目\n  for (const [key, entry] of messageStates) {\n    if (now - entry.createdAt >= MESSAGE_STATE_TTL_MS) {\n      messageStates.delete(key);\n    }\n  }\n\n  // 2. 容量限制：如果仍超过最大条目数，按时间淘汰最旧的\n  if (messageStates.size > MESSAGE_STATE_MAX_SIZE) {\n    const sorted = [...messageStates.entries()].sort((a, b) => a[1].createdAt - b[1].createdAt);\n    const toRemove = sorted.slice(0, messageStates.size - MESSAGE_STATE_MAX_SIZE);\n    for (const [key] of toRemove) {\n      messageStates.delete(key);\n    }\n  }\n}\n\n/**\n * 设置消息状态\n */\nexport function setMessageState(messageId: string, state: MessageState): void {\n  messageStates.set(messageId, {\n    state,\n    createdAt: Date.now(),\n  });\n}\n\n/**\n * 获取消息状态\n */\nexport function getMessageState(messageId: string): MessageState | undefined {\n  const entry = messageStates.get(messageId);\n  if (!entry) return undefined;\n\n  // 检查 TTL\n  if (Date.now() - entry.createdAt >= MESSAGE_STATE_TTL_MS) {\n    messageStates.delete(messageId);\n    return undefined;\n  }\n  return entry.state;\n}\n\n/**\n * 删除消息状态\n */\nexport function deleteMessageState(messageId: string): void {\n  messageStates.delete(messageId);\n}\n\n/**\n * 清空所有消息状态\n */\nexport function clearAllMessageStates(): void {\n  messageStates.clear();\n}\n\n// ============================================================================\n// ReqId 持久化存储管理（按 accountId 隔离）\n// ============================================================================\n\n/**\n * ReqId 持久化存储管理\n * 参考 createPersistentDedupe 模式：内存 + 磁盘双层、文件锁、原子写入、TTL 过期、防抖写入\n * 重启后可从磁盘恢复，确保主动推送消息时能获取到 reqId\n */\nconst reqIdStores = new Map<string, PersistentReqIdStore>();\n\nfunction getOrCreateReqIdStore(accountId: string): PersistentReqIdStore {\n  let store = reqIdStores.get(accountId);\n  if (!store) {\n    store = createPersistentReqIdStore(accountId);\n    reqIdStores.set(accountId, store);\n  }\n  return store;\n}\n\n// ============================================================================\n// ReqId 操作函数\n// ============================================================================\n\n/**\n * 设置 chatId 对应的 reqId（写入内存 + 防抖写磁盘）\n */\nexport function setReqIdForChat(chatId: string, reqId: string, accountId = \"default\"): void {\n  getOrCreateReqIdStore(accountId).set(chatId, reqId);\n}\n\n/**\n * 获取 chatId 对应的 reqId（异步：优先内存，miss 时查磁盘并回填内存）\n */\nexport async function getReqIdForChatAsync(chatId: string, accountId = \"default\"): Promise<string | undefined> {\n  return getOrCreateReqIdStore(accountId).get(chatId);\n}\n\n/**\n * 获取 chatId 对应的 reqId（同步：仅内存，保留向后兼容）\n */\nexport function getReqIdForChat(chatId: string, accountId = \"default\"): string | undefined {\n  return getOrCreateReqIdStore(accountId).getSync(chatId);\n}\n\n/**\n * 删除 chatId 对应的 reqId\n */\nexport function deleteReqIdForChat(chatId: string, accountId = \"default\"): void {\n  getOrCreateReqIdStore(accountId).delete(chatId);\n}\n\n/**\n * 启动时预热 reqId 缓存（从磁盘加载到内存）\n * \n * 注意：由于移除了磁盘存储，此函数现在只返回 0（无预热条目）\n */\nexport async function warmupReqIdStore(\n  accountId = \"default\",\n  log?: (...args: unknown[]) => void,\n): Promise<number> {\n  // 由于移除了磁盘存储，不再需要预热过程\n  log?.(\"[WeCom] reqid-store warmup: no-op (disk storage removed)\");\n  return 0;\n}\n\n/**\n * 立即将 reqId 数据刷写到磁盘（用于优雅退出）\n * \n * 注意：由于移除了磁盘存储，此函数现在是无操作\n */\nexport async function flushReqIdStore(accountId = \"default\"): Promise<void> {\n  // 由于移除了磁盘存储，不再需要刷写操作\n}\n\n// ============================================================================\n// 全局 cleanup（断开连接时释放所有资源）\n// ============================================================================\n\n/**\n * 清理指定账户的所有资源\n */\nexport async function cleanupAccount(accountId: string): Promise<void> {\n  // 1. 断开 WSClient\n  const wsClient = wsClientInstances.get(accountId);\n  if (wsClient) {\n    try {\n      wsClient.disconnect();\n    } catch {\n      // 忽略断开连接时的错误\n    }\n    wsClientInstances.delete(accountId);\n  }\n\n  // 2. 由于移除了磁盘存储，不再需要 flush reqId 存储\n  // 注意：不删除 store，因为重连后可能还需要\n}\n\n/**\n * 清理所有资源（用于进程退出）\n */\nexport async function cleanupAll(): Promise<void> {\n  // 停止定期清理\n  stopMessageStateCleanup();\n\n  // 清理所有 WSClient\n  for (const [accountId, wsClient] of wsClientInstances) {\n    try {\n      wsClient.disconnect();\n    } catch {\n      // 忽略\n    }\n  }\n  wsClientInstances.clear();\n\n  // 由于移除了磁盘存储，不再需要 flush 所有 reqId 存储\n\n  // 清空消息状态\n  clearAllMessageStates();\n}\n"],"names":["MESSAGE_STATE_CLEANUP_INTERVAL_MS","MESSAGE_STATE_TTL_MS","MESSAGE_STATE_MAX_SIZE","createPersistentReqIdStore"],"mappings":";;;;;AAAA;;;;;AAKG;AAWH;AACA;AACA;AAEA;AACA,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAoB;AAErD;;AAEG;AACG,SAAU,iBAAiB,CAAC,SAAiB,EAAA;IACjD,OAAO,iBAAiB,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,IAAI;AACjD;AAEA;;AAEG;AACG,SAAU,iBAAiB,CAAC,SAAiB,EAAE,MAAgB,EAAA;AACnE,IAAA,iBAAiB,CAAC,GAAG,CAAC,SAAS,EAAE,MAAM,CAAC;AAC1C;AAmBA;AACA,MAAM,aAAa,GAAG,IAAI,GAAG,EAA6B;AAE1D;AACA,IAAI,YAAY,GAA0C,IAAI;AAE9D;;AAEG;SACa,wBAAwB,GAAA;AACtC,IAAA,IAAI,YAAY;QAAE;AAElB,IAAA,YAAY,GAAG,WAAW,CAAC,MAAK;AAC9B,QAAA,kBAAkB,EAAE;IACtB,CAAC,EAAEA,wCAAiC,CAAC;;IAGrC,IAAI,YAAY,IAAI,OAAO,YAAY,KAAK,QAAQ,IAAI,OAAO,IAAI,YAAY,EAAE;QAC/E,YAAY,CAAC,KAAK,EAAE;IACtB;AACF;AAEA;;AAEG;SACa,uBAAuB,GAAA;IACrC,IAAI,YAAY,EAAE;QAChB,aAAa,CAAC,YAAY,CAAC;QAC3B,YAAY,GAAG,IAAI;IACrB;AACF;AAEA;;AAEG;AACH,SAAS,kBAAkB,GAAA;AACzB,IAAA,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE;;IAGtB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,aAAa,EAAE;QACxC,IAAI,GAAG,GAAG,KAAK,CAAC,SAAS,IAAIC,2BAAoB,EAAE;AACjD,YAAA,aAAa,CAAC,MAAM,CAAC,GAAG,CAAC;QAC3B;IACF;;AAGA,IAAA,IAAI,aAAa,CAAC,IAAI,GAAGC,6BAAsB,EAAE;AAC/C,QAAA,MAAM,MAAM,GAAG,CAAC,GAAG,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AAC3F,QAAA,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,aAAa,CAAC,IAAI,GAAGA,6BAAsB,CAAC;AAC7E,QAAA,KAAK,MAAM,CAAC,GAAG,CAAC,IAAI,QAAQ,EAAE;AAC5B,YAAA,aAAa,CAAC,MAAM,CAAC,GAAG,CAAC;QAC3B;IACF;AACF;AAEA;;AAEG;AACG,SAAU,eAAe,CAAC,SAAiB,EAAE,KAAmB,EAAA;AACpE,IAAA,aAAa,CAAC,GAAG,CAAC,SAAS,EAAE;QAC3B,KAAK;AACL,QAAA,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;AACtB,KAAA,CAAC;AACJ;AAiBA;;AAEG;AACG,SAAU,kBAAkB,CAAC,SAAiB,EAAA;AAClD,IAAA,aAAa,CAAC,MAAM,CAAC,SAAS,CAAC;AACjC;AASA;AACA;AACA;AAEA;;;;AAIG;AACH,MAAM,WAAW,GAAG,IAAI,GAAG,EAAgC;AAE3D,SAAS,qBAAqB,CAAC,SAAiB,EAAA;IAC9C,IAAI,KAAK,GAAG,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC;IACtC,IAAI,CAAC,KAAK,EAAE;AACV,QAAA,KAAK,GAAGC,qCAA0B,CAAU,CAAC;AAC7C,QAAA,WAAW,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC;IACnC;AACA,IAAA,OAAO,KAAK;AACd;AAEA;AACA;AACA;AAEA;;AAEG;AACG,SAAU,eAAe,CAAC,MAAc,EAAE,KAAa,EAAE,SAAS,GAAG,SAAS,EAAA;IAClF,qBAAqB,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC;AACrD;AAuBA;;;;AAIG;AACI,eAAe,gBAAgB,CACpC,SAAS,GAAG,SAAS,EACrB,GAAkC,EAAA;;AAGlC,IAAA,GAAG,GAAG,0DAA0D,CAAC;AACjE,IAAA,OAAO,CAAC;AACV;AAWA;AACA;AACA;AAEA;;AAEG;AACI,eAAe,cAAc,CAAC,SAAiB,EAAA;;IAEpD,MAAM,QAAQ,GAAG,iBAAiB,CAAC,GAAG,CAAC,SAAS,CAAC;IACjD,IAAI,QAAQ,EAAE;AACZ,QAAA,IAAI;YACF,QAAQ,CAAC,UAAU,EAAE;QACvB;AAAE,QAAA,MAAM;;QAER;AACA,QAAA,iBAAiB,CAAC,MAAM,CAAC,SAAS,CAAC;IACrC;;;AAIF;;;;;;;;;;;;"}