@yanhaidao/wecom 2.3.160 → 2.3.180

package/src/capability/mcp/transport.ts

@@ -0,0 +1,394 @@
+import { generateReqId } from "@wecom/aibot-node-sdk";
+import { getBotWsPushHandle } from "../../runtime.js";
+
+const HTTP_REQUEST_TIMEOUT_MS = 30_000;
+const MCP_CONFIG_FETCH_TIMEOUT_MS = 15_000;
+const MCP_GET_CONFIG_CMD = "aibot_get_mcp_config";
+const MCP_PLUGIN_VERSION = "wecom-dual-plane";
+const LOG_TAG = "[wecom-mcp]";
+
+interface JsonRpcRequest {
+  jsonrpc: "2.0";
+  id?: string;
+  method: string;
+  params?: Record<string, unknown>;
+}
+
+interface JsonRpcResponse {
+  jsonrpc: "2.0";
+  id: number | string;
+  result?: unknown;
+  error?: {
+    code: number;
+    message: string;
+    data?: unknown;
+  };
+}
+
+interface McpSession {
+  sessionId: string | null;
+  initialized: boolean;
+  stateless: boolean;
+}
+
+const CACHE_CLEAR_ERROR_CODES = new Set([-32001, -32002, -32003]);
+
+const mcpConfigCache = new Map<string, Record<string, unknown>>();
+const mcpSessionCache = new Map<string, McpSession>();
+const statelessKeys = new Set<string>();
+const inflightInitRequests = new Map<string, Promise<McpSession>>();
+
+function cacheKey(accountId: string, category: string): string {
+  return `${accountId}::${category}`;
+}
+
+function withTimeout<T>(promise: Promise<T>, timeoutMs: number, message: string): Promise<T> {
+  let timeoutId: ReturnType<typeof setTimeout> | undefined;
+  const timeoutPromise = new Promise<never>((_, reject) => {
+    timeoutId = setTimeout(() => reject(new Error(message)), timeoutMs);
+  });
+  return Promise.race([promise, timeoutPromise]).finally(() => {
+    if (timeoutId) clearTimeout(timeoutId);
+  });
+}
+
+export class McpRpcError extends Error {
+  constructor(
+    public readonly code: number,
+    message: string,
+    public readonly data?: unknown,
+  ) {
+    super(message);
+    this.name = "McpRpcError";
+  }
+}
+
+export class McpHttpError extends Error {
+  constructor(
+    public readonly statusCode: number,
+    message: string,
+  ) {
+    super(message);
+    this.name = "McpHttpError";
+  }
+}
+
+async function fetchMcpConfig(
+  accountId: string,
+  category: string,
+): Promise<Record<string, unknown>> {
+  const handle = getBotWsPushHandle(accountId);
+  if (!handle?.isConnected()) {
+    throw new Error(`当前企微账号 MCP 服务未就绪：account=${accountId} 的 Bot WS 未连接。`);
+  }
+
+  const response = await withTimeout(
+    handle.replyCommand({
+      cmd: MCP_GET_CONFIG_CMD,
+      body: {
+        biz_type: category,
+        plugin_version: MCP_PLUGIN_VERSION,
+      },
+      headers: {
+        req_id: generateReqId("mcp_config"),
+      },
+    }),
+    MCP_CONFIG_FETCH_TIMEOUT_MS,
+    `MCP config fetch timed out after ${MCP_CONFIG_FETCH_TIMEOUT_MS}ms`,
+  );
+
+  const errcode = Number((response as { errcode?: number }).errcode ?? 0);
+  if (errcode !== 0) {
+    throw new Error(
+      `MCP 配置请求失败: errcode=${String((response as { errcode?: number }).errcode)} errmsg=${String((response as { errmsg?: string }).errmsg ?? "unknown")}`,
+    );
+  }
+
+  const body = (response as { body?: { url?: string } }).body;
+  if (!body?.url) {
+    throw new Error(`MCP 配置响应缺少 url 字段 (account=${accountId}, category=${category})`);
+  }
+
+  console.log(`${LOG_TAG} config ready account=${accountId} category=${category} url=${body.url}`);
+  return body as Record<string, unknown>;
+}
+
+async function getMcpUrl(accountId: string, category: string): Promise<string> {
+  const key = cacheKey(accountId, category);
+  const cached = mcpConfigCache.get(key);
+  if (cached?.url) {
+    return String(cached.url);
+  }
+  const body = await fetchMcpConfig(accountId, category);
+  mcpConfigCache.set(key, body);
+  return String(body.url);
+}
+
+async function sendRawJsonRpc(
+  url: string,
+  session: McpSession,
+  body: JsonRpcRequest,
+): Promise<{ rpcResult: unknown; newSessionId: string | null }> {
+  const controller = new AbortController();
+  const timeoutId = setTimeout(() => controller.abort(), HTTP_REQUEST_TIMEOUT_MS);
+  try {
+    const headers: Record<string, string> = {
+      "Content-Type": "application/json",
+      Accept: "application/json, text/event-stream",
+    };
+    if (session.sessionId) {
+      headers["Mcp-Session-Id"] = session.sessionId;
+    }
+
+    const response = await fetch(url, {
+      method: "POST",
+      headers,
+      body: JSON.stringify(body),
+      signal: controller.signal,
+    });
+    const newSessionId = response.headers.get("mcp-session-id");
+
+    if (!response.ok) {
+      throw new McpHttpError(
+        response.status,
+        `MCP HTTP 请求失败: ${response.status} ${response.statusText}`,
+      );
+    }
+
+    const contentLength = response.headers.get("content-length");
+    if (response.status === 204 || contentLength === "0") {
+      return { rpcResult: undefined, newSessionId };
+    }
+
+    const contentType = response.headers.get("content-type") ?? "";
+    if (contentType.includes("text/event-stream")) {
+      return {
+        rpcResult: await parseSseResponse(response),
+        newSessionId,
+      };
+    }
+
+    const text = await response.text();
+    if (!text.trim()) {
+      return { rpcResult: undefined, newSessionId };
+    }
+
+    const rpc = JSON.parse(text) as JsonRpcResponse;
+    if (rpc.error) {
+      throw new McpRpcError(
+        rpc.error.code,
+        `MCP 调用错误 [${rpc.error.code}]: ${rpc.error.message}`,
+        rpc.error.data,
+      );
+    }
+    return { rpcResult: rpc.result, newSessionId };
+  } catch (error) {
+    if (error instanceof DOMException && error.name === "AbortError") {
+      throw new Error(`MCP 请求超时 (${HTTP_REQUEST_TIMEOUT_MS}ms)`);
+    }
+    throw error;
+  } finally {
+    clearTimeout(timeoutId);
+  }
+}
+
+async function initializeSession(
+  accountId: string,
+  category: string,
+  url: string,
+): Promise<McpSession> {
+  const key = cacheKey(accountId, category);
+  const session: McpSession = { sessionId: null, initialized: false, stateless: false };
+
+  const initializeRequest: JsonRpcRequest = {
+    jsonrpc: "2.0",
+    id: generateReqId("mcp_init"),
+    method: "initialize",
+    params: {
+      protocolVersion: "2025-03-26",
+      capabilities: {},
+      clientInfo: { name: "wecom_mcp", version: MCP_PLUGIN_VERSION },
+    },
+  };
+
+  const initResult = await sendRawJsonRpc(url, session, initializeRequest);
+  if (initResult.newSessionId) {
+    session.sessionId = initResult.newSessionId;
+  }
+  if (!session.sessionId) {
+    session.stateless = true;
+    session.initialized = true;
+    statelessKeys.add(key);
+    mcpSessionCache.set(key, session);
+    return session;
+  }
+
+  const notifyRequest: JsonRpcRequest = {
+    jsonrpc: "2.0",
+    method: "notifications/initialized",
+  };
+  const notifyResult = await sendRawJsonRpc(url, session, notifyRequest);
+  if (notifyResult.newSessionId) {
+    session.sessionId = notifyResult.newSessionId;
+  }
+  session.initialized = true;
+  mcpSessionCache.set(key, session);
+  return session;
+}
+
+async function getOrCreateSession(
+  accountId: string,
+  category: string,
+  url: string,
+): Promise<McpSession> {
+  const key = cacheKey(accountId, category);
+  if (statelessKeys.has(key)) {
+    const cached = mcpSessionCache.get(key);
+    if (cached) return cached;
+  }
+
+  const cached = mcpSessionCache.get(key);
+  if (cached?.initialized) {
+    return cached;
+  }
+
+  const inflight = inflightInitRequests.get(key);
+  if (inflight) {
+    return inflight;
+  }
+
+  const promise = initializeSession(accountId, category, url).finally(() => {
+    inflightInitRequests.delete(key);
+  });
+  inflightInitRequests.set(key, promise);
+  return promise;
+}
+
+async function rebuildSession(
+  accountId: string,
+  category: string,
+  url: string,
+): Promise<McpSession> {
+  const key = cacheKey(accountId, category);
+  const inflight = inflightInitRequests.get(key);
+  if (inflight) return inflight;
+  const promise = initializeSession(accountId, category, url).finally(() => {
+    inflightInitRequests.delete(key);
+  });
+  inflightInitRequests.set(key, promise);
+  return promise;
+}
+
+async function parseSseResponse(response: Response): Promise<unknown> {
+  const text = await response.text();
+  const lines = text.split("\n");
+  let currentParts: string[] = [];
+  let lastEventData = "";
+
+  for (const line of lines) {
+    if (line.startsWith("data: ")) {
+      currentParts.push(line.slice(6));
+      continue;
+    }
+    if (line.startsWith("data:")) {
+      currentParts.push(line.slice(5));
+      continue;
+    }
+    if (line.trim() === "" && currentParts.length > 0) {
+      lastEventData = currentParts.join("\n").trim();
+      currentParts = [];
+    }
+  }
+  if (currentParts.length > 0) {
+    lastEventData = currentParts.join("\n").trim();
+  }
+  if (!lastEventData) {
+    throw new Error("SSE 响应中未包含有效数据");
+  }
+
+  const rpc = JSON.parse(lastEventData) as JsonRpcResponse;
+  if (rpc.error) {
+    throw new McpRpcError(
+      rpc.error.code,
+      `MCP 调用错误 [${rpc.error.code}]: ${rpc.error.message}`,
+      rpc.error.data,
+    );
+  }
+  return rpc.result;
+}
+
+export function clearWecomMcpCategoryCache(accountId: string, category: string): void {
+  const key = cacheKey(accountId, category);
+  console.log(`${LOG_TAG} clear cache account=${accountId} category=${category}`);
+  mcpConfigCache.delete(key);
+  mcpSessionCache.delete(key);
+  statelessKeys.delete(key);
+  inflightInitRequests.delete(key);
+}
+
+export function clearWecomMcpAccountCache(accountId: string): void {
+  const prefix = `${accountId}::`;
+  for (const key of [...mcpConfigCache.keys()]) {
+    if (key.startsWith(prefix)) mcpConfigCache.delete(key);
+  }
+  for (const key of [...mcpSessionCache.keys()]) {
+    if (key.startsWith(prefix)) mcpSessionCache.delete(key);
+  }
+  for (const key of [...statelessKeys]) {
+    if (key.startsWith(prefix)) statelessKeys.delete(key);
+  }
+  for (const key of [...inflightInitRequests.keys()]) {
+    if (key.startsWith(prefix)) inflightInitRequests.delete(key);
+  }
+}
+
+export interface McpToolInfo {
+  name: string;
+  description?: string;
+  inputSchema?: Record<string, unknown>;
+}
+
+export async function sendJsonRpc(
+  accountId: string,
+  category: string,
+  method: string,
+  params?: Record<string, unknown>,
+): Promise<unknown> {
+  const url = await getMcpUrl(accountId, category);
+  const body: JsonRpcRequest = {
+    jsonrpc: "2.0",
+    id: generateReqId("mcp_rpc"),
+    method,
+    ...(params !== undefined ? { params } : {}),
+  };
+
+  let session = await getOrCreateSession(accountId, category, url);
+
+  try {
+    const result = await sendRawJsonRpc(url, session, body);
+    if (result.newSessionId) {
+      session.sessionId = result.newSessionId;
+    }
+    return result.rpcResult;
+  } catch (error) {
+    if (error instanceof McpRpcError && CACHE_CLEAR_ERROR_CODES.has(error.code)) {
+      clearWecomMcpCategoryCache(accountId, category);
+    }
+    if (session.stateless) {
+      throw error;
+    }
+    if (error instanceof McpHttpError && error.statusCode === 404) {
+      mcpSessionCache.delete(cacheKey(accountId, category));
+      session = await rebuildSession(accountId, category, url);
+      const result = await sendRawJsonRpc(url, session, body);
+      if (result.newSessionId) {
+        session.sessionId = result.newSessionId;
+      }
+      return result.rpcResult;
+    }
+    console.error(
+      `${LOG_TAG} rpc failed account=${accountId} category=${category} method=${method} error=${error instanceof Error ? error.message : String(error)}`,
+    );
+    throw error;
+  }
+}

package/src/channel.ts

@@ -27,16 +27,22 @@ const meta = {
   selectionLabel: "WeCom (企业微信)",
   docsPath: "/channels/wecom",
   docsLabel: "企业微信",
-  blurb: "企业微信官方推荐三方插件，默认 Bot WS 配置简单，支持主动发消息与 Agent 全能力。",
+  blurb:
+    "企业微信官方推荐三方插件，默认 Bot WS 配置简单，支持主动发消息与 Agent 全能力。",
   selectionDocsPrefix: "文档：",
   aliases: ["wechatwork", "wework", "qywx", "企微", "企业微信"],
   order: 85,
   quickstartAllowFrom: true,
 };
 
-function resolveAccountInboundPath(account: ResolvedWecomAccount): string | undefined {
+function resolveAccountInboundPath(
+  account: ResolvedWecomAccount,
+): string | undefined {
   const derivedPaths = resolveDerivedPathSummary(account.accountId);
-  if (account.bot?.primaryTransport === "webhook" && account.bot.webhookConfigured) {
+  if (
+    account.bot?.primaryTransport === "webhook" &&
+    account.bot.webhookConfigured
+  ) {
     return derivedPaths.botWebhook[0];
   }
   if (account.agent?.callbackConfigured) {
@@ -51,7 +57,9 @@ function normalizeWecomMessagingTarget(raw: string): string | undefined {
   if (/^wecom-agent:/i.test(trimmed)) {
     return trimmed;
   }
-  return trimmed.replace(/^(wecom|wechatwork|wework|qywx):/i, "").trim() || undefined;
+  return (
+    trimmed.replace(/^(wecom|wechatwork|wework|qywx):/i, "").trim() || undefined
+  );
 }
 
 export const wecomPlugin: ChannelPlugin<ResolvedWecomAccount> = {
@@ -68,9 +76,6 @@ export const wecomPlugin: ChannelPlugin<ResolvedWecomAccount> = {
     blockStreaming: false,
   },
   reload: { configPrefixes: ["channels.wecom"] },
-  // NOTE: We intentionally avoid Zod -> JSON Schema conversion at plugin-load time.
-  // Some OpenClaw runtime environments load plugin modules via jiti in a way that can
-  // surface zod `toJSONSchema()` binding issues (e.g. `this` undefined leading to `_zod` errors).
   // A permissive schema keeps config UX working while preventing startup failures.
   configSchema: {
     schema: {
@@ -81,8 +86,10 @@ export const wecomPlugin: ChannelPlugin<ResolvedWecomAccount> = {
   },
   config: {
     listAccountIds: (cfg) => listWecomAccountIds(cfg as OpenClawConfig),
-    resolveAccount: (cfg, accountId) => resolveWecomAccount({ cfg: cfg as OpenClawConfig, accountId }),
-    defaultAccountId: (cfg) => resolveDefaultWecomAccountId(cfg as OpenClawConfig),
+    resolveAccount: (cfg, accountId) =>
+      resolveWecomAccount({ cfg: cfg as OpenClawConfig, accountId }),
+    defaultAccountId: (cfg) =>
+      resolveDefaultWecomAccountId(cfg as OpenClawConfig),
     setAccountEnabled: ({ cfg, accountId, enabled }) =>
       setAccountEnabledInConfigSection({
         cfg: cfg as OpenClawConfig,
@@ -126,9 +133,15 @@ export const wecomPlugin: ChannelPlugin<ResolvedWecomAccount> = {
       };
     },
     resolveAllowFrom: ({ cfg, accountId }) => {
-      const account = resolveWecomAccount({ cfg: cfg as OpenClawConfig, accountId });
+      const account = resolveWecomAccount({
+        cfg: cfg as OpenClawConfig,
+        accountId,
+      });
       // 与其他渠道保持一致：直接返回 allowFrom，空则允许所有人
-      const allowFrom = account.agent?.config.dm?.allowFrom ?? account.bot?.config.dm?.allowFrom ?? [];
+      const allowFrom =
+        account.agent?.config.dm?.allowFrom ??
+        account.bot?.config.dm?.allowFrom ??
+        [];
       return allowFrom.map((entry) => String(entry));
     },
     formatAllowFrom: ({ allowFrom }) =>
@@ -170,20 +183,31 @@ export const wecomPlugin: ChannelPlugin<ResolvedWecomAccount> = {
       transport: (snapshot as { transport?: string }).transport ?? null,
       ownerId: (snapshot as { ownerId?: string }).ownerId ?? null,
       health: (snapshot as { health?: string }).health ?? "idle",
-      ownerDriftAt: (snapshot as { ownerDriftAt?: number | null }).ownerDriftAt ?? null,
+      ownerDriftAt:
+        (snapshot as { ownerDriftAt?: number | null }).ownerDriftAt ?? null,
       connected: (snapshot as { connected?: boolean }).connected,
       authenticated: (snapshot as { authenticated?: boolean }).authenticated,
       lastStartAt: snapshot.lastStartAt ?? null,
       lastStopAt: snapshot.lastStopAt ?? null,
       lastError: snapshot.lastError ?? null,
-      lastErrorAt: (snapshot as { lastErrorAt?: number | null }).lastErrorAt ?? null,
+      lastErrorAt:
+        (snapshot as { lastErrorAt?: number | null }).lastErrorAt ?? null,
       lastInboundAt: snapshot.lastInboundAt ?? null,
       lastOutboundAt: snapshot.lastOutboundAt ?? null,
-      recentInboundSummary: (snapshot as { recentInboundSummary?: string | null }).recentInboundSummary ?? null,
-      recentOutboundSummary: (snapshot as { recentOutboundSummary?: string | null }).recentOutboundSummary ?? null,
-      recentIssueCategory: (snapshot as { recentIssueCategory?: string | null }).recentIssueCategory ?? null,
-      recentIssueSummary: (snapshot as { recentIssueSummary?: string | null }).recentIssueSummary ?? null,
-      transportSessions: (snapshot as { transportSessions?: string[] }).transportSessions ?? [],
+      recentInboundSummary:
+        (snapshot as { recentInboundSummary?: string | null })
+          .recentInboundSummary ?? null,
+      recentOutboundSummary:
+        (snapshot as { recentOutboundSummary?: string | null })
+          .recentOutboundSummary ?? null,
+      recentIssueCategory:
+        (snapshot as { recentIssueCategory?: string | null })
+          .recentIssueCategory ?? null,
+      recentIssueSummary:
+        (snapshot as { recentIssueSummary?: string | null })
+          .recentIssueSummary ?? null,
+      transportSessions:
+        (snapshot as { transportSessions?: string[] }).transportSessions ?? [],
       probe: snapshot.probe,
       lastProbeAt: snapshot.lastProbeAt ?? null,
     }),
@@ -199,25 +223,43 @@ export const wecomPlugin: ChannelPlugin<ResolvedWecomAccount> = {
         enabled: account.enabled,
         configured: account.configured && !conflict,
         webhookPath: resolveAccountInboundPath(account),
-        primaryTransport: account.bot?.primaryTransport ?? (account.agent ? "agent-callback" : null),
-        transport: (runtime as { transport?: string } | undefined)?.transport ?? null,
+        primaryTransport:
+          account.bot?.primaryTransport ??
+          (account.agent ? "agent-callback" : null),
+        transport:
+          (runtime as { transport?: string } | undefined)?.transport ?? null,
         ownerId: (runtime as { ownerId?: string } | undefined)?.ownerId ?? null,
         health: (runtime as { health?: string } | undefined)?.health ?? "idle",
-        ownerDriftAt: (runtime as { ownerDriftAt?: number | null } | undefined)?.ownerDriftAt ?? null,
+        ownerDriftAt:
+          (runtime as { ownerDriftAt?: number | null } | undefined)
+            ?.ownerDriftAt ?? null,
         connected: (runtime as { connected?: boolean } | undefined)?.connected,
-        authenticated: (runtime as { authenticated?: boolean } | undefined)?.authenticated,
+        authenticated: (runtime as { authenticated?: boolean } | undefined)
+          ?.authenticated,
         running: runtime?.running ?? false,
         lastStartAt: runtime?.lastStartAt ?? null,
         lastStopAt: runtime?.lastStopAt ?? null,
         lastError: runtime?.lastError ?? conflict?.message ?? null,
-        lastErrorAt: (runtime as { lastErrorAt?: number | null } | undefined)?.lastErrorAt ?? null,
+        lastErrorAt:
+          (runtime as { lastErrorAt?: number | null } | undefined)
+            ?.lastErrorAt ?? null,
         lastInboundAt: runtime?.lastInboundAt ?? null,
         lastOutboundAt: runtime?.lastOutboundAt ?? null,
-        recentInboundSummary: (runtime as { recentInboundSummary?: string | null } | undefined)?.recentInboundSummary ?? null,
-        recentOutboundSummary: (runtime as { recentOutboundSummary?: string | null } | undefined)?.recentOutboundSummary ?? null,
-        recentIssueCategory: (runtime as { recentIssueCategory?: string | null } | undefined)?.recentIssueCategory ?? null,
-        recentIssueSummary: (runtime as { recentIssueSummary?: string | null } | undefined)?.recentIssueSummary ?? null,
-        transportSessions: (runtime as { transportSessions?: string[] } | undefined)?.transportSessions ?? [],
+        recentInboundSummary:
+          (runtime as { recentInboundSummary?: string | null } | undefined)
+            ?.recentInboundSummary ?? null,
+        recentOutboundSummary:
+          (runtime as { recentOutboundSummary?: string | null } | undefined)
+            ?.recentOutboundSummary ?? null,
+        recentIssueCategory:
+          (runtime as { recentIssueCategory?: string | null } | undefined)
+            ?.recentIssueCategory ?? null,
+        recentIssueSummary:
+          (runtime as { recentIssueSummary?: string | null } | undefined)
+            ?.recentIssueSummary ?? null,
+        transportSessions:
+          (runtime as { transportSessions?: string[] } | undefined)
+            ?.transportSessions ?? [],
         dmPolicy: account.bot?.config.dm?.policy ?? "pairing",
       };
     },