@vui-design/openclaw-plugin-feishu-progress 0.4.1 → 0.4.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vui-design/openclaw-plugin-feishu-progress",
-  "version": "0.4.1",
+  "version": "0.4.2",
   "description": "飞书任务进度卡片插件 — 在 AI 执行复杂多步骤任务时，实时更新飞书进度卡片",
   "keywords": [
     "openclaw-plugin",

package/src/feishu-client.ts

@@ -11,6 +11,8 @@ interface TokenCache {
 
 export class FeishuClient {
   private tokenCache: TokenCache | null = null;
+  // 并发刷新去重：多个调用同时过期时只发一次请求
+  private refreshPromise: Promise<string> | null = null;
 
   constructor(private config: FeishuConfig) {}
 
@@ -24,6 +26,14 @@ export class FeishuClient {
     if (this.tokenCache && Date.now() < this.tokenCache.expiresAt) {
       return this.tokenCache.token;
     }
+    if (this.refreshPromise) return this.refreshPromise;
+    this.refreshPromise = this.doRefresh().finally(() => {
+      this.refreshPromise = null;
+    });
+    return this.refreshPromise;
+  }
+
+  private async doRefresh(): Promise<string> {
     const res = await fetch(
       `${this.baseUrl}/open-apis/auth/v3/tenant_access_token/internal`,
       {
@@ -38,7 +48,12 @@ export class FeishuClient {
     if (!res.ok) {
       throw new Error(`飞书获取 token 失败: HTTP ${res.status}`);
     }
-    const data = (await res.json()) as { tenant_access_token: string; expire: number; code: number; msg: string };
+    const data = (await res.json()) as {
+      tenant_access_token: string;
+      expire: number;
+      code: number;
+      msg: string;
+    };
     if (data.code !== 0) {
       throw new Error(`飞书获取 token 失败: ${data.msg} (code=${data.code})`);
     }
@@ -78,7 +93,11 @@ export class FeishuClient {
     if (!res.ok) {
       throw new Error(`飞书发送消息失败: HTTP ${res.status}`);
     }
-    const data = (await res.json()) as { code: number; msg: string; data: { message_id: string } };
+    const data = (await res.json()) as {
+      code: number;
+      msg: string;
+      data: { message_id: string };
+    };
     if (data.code !== 0) {
       throw new Error(`飞书发送消息失败: ${data.msg} (code=${data.code})`);
     }

package/src/index.ts

@@ -99,12 +99,21 @@ function extractSnippet(text: string): string {
   return (lines[lines.length - 1] ?? '').slice(0, 50);
 }
 
+// 从 sessionKey 提取飞书用户 ID
+// sessionKey 格式：agent:{agentId}:feishu:direct:{userId}
+function chatIdFromSessionKey(sessionKey: string | undefined): string | undefined {
+  if (!sessionKey) return undefined;
+  const m = sessionKey.match(/^agent:[^:]+:feishu:(?:direct|group):(.+)$/);
+  return m?.[1];
+}
+
 // ── 主插件 ────────────────────────────────────────────────────────────────
 
 interface RunState {
-  /** 飞书接收方 ID（ou_xxx / oc_xxx） */
   chatId: string;
-  /** 当前轮次累积的 AI 输出文本 */
+  /** 用于 agent_end 精确匹配，避免按 chatId 误删并发 run */
+  agentId: string;
+  sessionKey: string;
   assistantBuffer: string;
   /** 已发送过的 snippet，防止重复推送 */
   sentSnippets: Set<string>;
@@ -135,50 +144,54 @@ export default {
       domain: cfg.domain ?? 'feishu',
     });
 
-    // ── 关联数据 ────────────────────────────────────────────────────────────
-
-    // accountId ("main") → 飞书裸 ID（ou_xxx / oc_xxx）
-    // 用 sessionKey 细化：sessionKey → chatId（隔离多对话）
-    const sessionKeyChatMap = new Map<string, string>();
-    // accountId → chatId（兜底，在没有 sessionKey 时使用）
+    // accountId ("main") → 飞书裸 ID，兜底用（sessionKey 解析失败时）
     const accountChatMap = new Map<string, string>();
 
     // runId → RunState
     const runStateMap = new Map<string, RunState>();
 
+    // 统一清理入口
+    function cleanupRun(runId: string): void {
+      const state = runStateMap.get(runId);
+      if (!state) return;
+      clearInterval(state.heartbeatTimer);
+      runStateMap.delete(runId);
+    }
+
     // ── Hooks ───────────────────────────────────────────────────────────────
 
-    // 收到飞书消息 → 记录 chatId，同时按 sessionKey 和 accountId 双索引
+    // 收到飞书消息 → 记录 accountId → chatId（兜底索引）
     api.on('message_received', async (_event: any, ctx: any) => {
       if (ctx.channelId !== 'feishu') return;
       if (!ctx.conversationId) return;
+      // OpenClaw 内部格式 "user:ou_xxx"，剥离前缀
       const rawId = ctx.conversationId.includes(':')
         ? ctx.conversationId.split(':').pop()!
         : ctx.conversationId;
       if (ctx.accountId) accountChatMap.set(ctx.accountId, rawId);
-      // sessionKey 在 message_received ctx 里也可能存在
-      if (ctx.sessionKey) sessionKeyChatMap.set(ctx.sessionKey, rawId);
     });
 
     // llm_input → 建立 runId → RunState，启动心跳
-    // 此时 ctx.agentId 和 ctx.sessionKey 均可用，可精确找到 chatId
     api.on('llm_input', async (event: any, ctx: any) => {
       if (!event.runId) return;
       if (runStateMap.has(event.runId)) return; // 同一 run 只初始化一次
 
-      // sessionKey 精确匹配优先，其次 agentId 兜底
+      // 从 sessionKey 精确解析 chatId；无法解析时退回 accountChatMap
       const chatId =
-        (ctx.sessionKey ? sessionKeyChatMap.get(ctx.sessionKey) : undefined) ??
+        chatIdFromSessionKey(ctx.sessionKey) ??
         (ctx.agentId ? accountChatMap.get(ctx.agentId) : undefined);
       if (!chatId) return;
 
       const startedAt = Date.now();
       const state: RunState = {
         chatId,
+        agentId: ctx.agentId ?? '',
+        sessionKey: ctx.sessionKey ?? '',
         assistantBuffer: '',
         sentSnippets: new Set(),
         startedAt,
         lastSentAt: startedAt,
+        // 10s 间隔确保 55s 阈值能准时触发
         heartbeatTimer: setInterval(() => {
           if (!runStateMap.has(event.runId)) return;
           const idleSec = (Date.now() - state.lastSentAt) / 1000;
@@ -187,7 +200,7 @@ export default {
             state.lastSentAt = Date.now();
             client.sendText(chatId, `⏳ 处理中，已用时约 ${elapsed} 秒...`).catch(() => {});
           }
-        }, 30_000),
+        }, 10_000),
       };
 
       runStateMap.set(event.runId, state);
@@ -212,7 +225,7 @@ export default {
 
         let label: string;
         if (snippet && !state.sentSnippets.has(snippet)) {
-          // AI 文本去重：同一段文字不重复发送
+          // AI 文本优先，且去重
           state.sentSnippets.add(snippet);
           label = snippet;
         } else {
@@ -231,22 +244,16 @@ export default {
 
       // lifecycle:end/error → 清理
       if (stream === 'lifecycle' && (data?.phase === 'end' || data?.phase === 'error')) {
-        clearInterval(state.heartbeatTimer);
-        runStateMap.delete(runId);
+        cleanupRun(runId);
       }
     });
 
-    // agent_end hook 作为兜底清理（防止 lifecycle 事件丢失时内存泄漏）
+    // agent_end → 兜底清理（按 sessionKey/agentId 精确匹配，不按 chatId）
     api.on('agent_end', async (_event: any, ctx: any) => {
-      // 通过 sessionKey 找到对应的 runId 并清理
       for (const [runId, state] of runStateMap) {
-        if (
-          (ctx.sessionKey && sessionKeyChatMap.get(ctx.sessionKey) === state.chatId) ||
-          (ctx.agentId && accountChatMap.get(ctx.agentId) === state.chatId)
-        ) {
-          clearInterval(state.heartbeatTimer);
-          runStateMap.delete(runId);
-        }
+        const matchBySession = ctx.sessionKey && state.sessionKey === ctx.sessionKey;
+        const matchByAgent = !ctx.sessionKey && ctx.agentId && state.agentId === ctx.agentId;
+        if (matchBySession || matchByAgent) cleanupRun(runId);
       }
     });
   },