@yanhaidao/wecom 2.4.160 → 2.5.110

package/src/capability/mcp/index.ts

@@ -1,10 +0,0 @@
-export { createWeComMcpToolFactory } from "./tool.js";
-export {
-  clearWecomMcpAccountCache,
-  clearWecomMcpCategoryCache,
-  McpHttpError,
-  McpRpcError,
-  sendJsonRpc,
-  type McpToolInfo,
-} from "./transport.js";
-export { cleanSchemaForGemini } from "./schema.js";

package/src/capability/mcp/schema.ts

@@ -1,107 +0,0 @@
-const GEMINI_UNSUPPORTED_KEYWORDS = new Set([
-  "patternProperties",
-  "additionalProperties",
-  "$schema",
-  "$id",
-  "$ref",
-  "$defs",
-  "definitions",
-  "examples",
-  "minLength",
-  "maxLength",
-  "minimum",
-  "maximum",
-  "multipleOf",
-  "pattern",
-  "format",
-  "minItems",
-  "maxItems",
-  "uniqueItems",
-  "minProperties",
-  "maxProperties",
-]);
-
-export function cleanSchemaForGemini(schema: unknown): unknown {
-  if (!schema || typeof schema !== "object") return schema;
-  if (Array.isArray(schema)) return schema.map(cleanSchemaForGemini);
-
-  const obj = schema as Record<string, unknown>;
-  const defs: Record<string, unknown> = {
-    ...(obj.$defs && typeof obj.$defs === "object" ? (obj.$defs as Record<string, unknown>) : {}),
-    ...(obj.definitions && typeof obj.definitions === "object"
-      ? (obj.definitions as Record<string, unknown>)
-      : {}),
-  };
-  return cleanWithDefs(obj, defs, new Set());
-}
-
-function cleanWithDefs(
-  schema: unknown,
-  defs: Record<string, unknown>,
-  refStack: Set<string>,
-): unknown {
-  if (!schema || typeof schema !== "object") return schema;
-  if (Array.isArray(schema)) return schema.map((item) => cleanWithDefs(item, defs, refStack));
-
-  const obj = schema as Record<string, unknown>;
-  if (obj.$defs && typeof obj.$defs === "object") {
-    Object.assign(defs, obj.$defs as Record<string, unknown>);
-  }
-  if (obj.definitions && typeof obj.definitions === "object") {
-    Object.assign(defs, obj.definitions as Record<string, unknown>);
-  }
-
-  if (typeof obj.$ref === "string") {
-    const ref = obj.$ref;
-    if (refStack.has(ref)) return {};
-    const match = ref.match(/^#\/(?:\$defs|definitions)\/(.+)$/);
-    if (match?.[1] && defs[match[1]]) {
-      const nextStack = new Set(refStack);
-      nextStack.add(ref);
-      return cleanWithDefs(defs[match[1]], defs, nextStack);
-    }
-    return {};
-  }
-
-  const cleaned: Record<string, unknown> = {};
-  for (const [key, value] of Object.entries(obj)) {
-    if (GEMINI_UNSUPPORTED_KEYWORDS.has(key)) continue;
-    if (key === "const") {
-      cleaned.enum = [value];
-      continue;
-    }
-    if (key === "properties" && value && typeof value === "object" && !Array.isArray(value)) {
-      cleaned[key] = Object.fromEntries(
-        Object.entries(value as Record<string, unknown>).map(([propKey, propValue]) => [
-          propKey,
-          cleanWithDefs(propValue, defs, refStack),
-        ]),
-      );
-      continue;
-    }
-    if (key === "items" && value) {
-      cleaned[key] = Array.isArray(value)
-        ? value.map((item) => cleanWithDefs(item, defs, refStack))
-        : cleanWithDefs(value, defs, refStack);
-      continue;
-    }
-    if ((key === "anyOf" || key === "oneOf" || key === "allOf") && Array.isArray(value)) {
-      const nonNull = value.filter((variant) => {
-        if (!variant || typeof variant !== "object") return true;
-        return (variant as Record<string, unknown>).type !== "null";
-      });
-      if (nonNull.length === 1) {
-        const single = cleanWithDefs(nonNull[0], defs, refStack);
-        if (single && typeof single === "object" && !Array.isArray(single)) {
-          Object.assign(cleaned, single as Record<string, unknown>);
-        }
-      } else {
-        cleaned[key] = nonNull.map((variant) => cleanWithDefs(variant, defs, refStack));
-      }
-      continue;
-    }
-    cleaned[key] = value;
-  }
-
-  return cleaned;
-}

package/src/capability/mcp/tool.ts

@@ -1,174 +0,0 @@
-import type {
-  OpenClawPluginToolContext,
-  OpenClawPluginToolFactory,
-} from "openclaw/plugin-sdk/core";
-import { resolveWecomSourceSnapshot } from "../../runtime/source-registry.js";
-import { cleanSchemaForGemini } from "./schema.js";
-import { clearWecomMcpCategoryCache, sendJsonRpc, type McpToolInfo } from "./transport.js";
-
-type WecomMcpParams = {
-  action: "list" | "call";
-  category: string;
-  method?: string;
-  args?: string | Record<string, unknown>;
-};
-
-const BIZ_CACHE_CLEAR_ERROR_CODES = new Set([850002]);
-
-function textResult<TDetails>(data: TDetails) {
-  return {
-    content: [{ type: "text" as const, text: JSON.stringify(data, null, 2) }],
-    details: data,
-  };
-}
-
-function errorResult(error: unknown) {
-  if (error && typeof error === "object" && "errcode" in error) {
-    const errcode = Number((error as { errcode?: number }).errcode ?? 0);
-    const errmsg = String((error as { errmsg?: string }).errmsg ?? `错误码: ${errcode}`);
-    return textResult({ error: errmsg, errcode });
-  }
-  return textResult({
-    error: error instanceof Error ? error.message : String(error),
-  });
-}
-
-function parseArgs(args: string | Record<string, unknown> | undefined): Record<string, unknown> {
-  if (!args) return {};
-  if (typeof args === "object") return args;
-  try {
-    return JSON.parse(args) as Record<string, unknown>;
-  } catch (error) {
-    const detail = error instanceof SyntaxError ? error.message : String(error);
-    throw new Error(`args 不是合法的 JSON: ${args} (${detail})`);
-  }
-}
-
-function extractToolAccountId(ctx: OpenClawPluginToolContext): string | undefined {
-  const explicit = String((ctx as { accountId?: string }).accountId ?? "").trim();
-  if (explicit) return explicit;
-  const agentAccountId = String(ctx.agentAccountId ?? "").trim();
-  return agentAccountId || undefined;
-}
-
-async function handleList(accountId: string, category: string): Promise<unknown> {
-  const result = (await sendJsonRpc(accountId, category, "tools/list")) as
-    | { tools?: McpToolInfo[] }
-    | undefined;
-  const tools = result?.tools ?? [];
-  return {
-    accountId,
-    category,
-    count: tools.length,
-    tools: tools.map((tool) => ({
-      name: tool.name,
-      description: tool.description ?? "",
-      inputSchema: tool.inputSchema ? cleanSchemaForGemini(tool.inputSchema) : undefined,
-    })),
-  };
-}
-
-function checkBizErrorAndClearCache(result: unknown, accountId: string, category: string): void {
-  if (!result || typeof result !== "object") return;
-  const content = (result as { content?: Array<{ type: string; text?: string }> }).content;
-  if (!Array.isArray(content)) return;
-  for (const item of content) {
-    if (item.type !== "text" || !item.text) continue;
-    try {
-      const parsed = JSON.parse(item.text) as { errcode?: number };
-      if (typeof parsed.errcode === "number" && BIZ_CACHE_CLEAR_ERROR_CODES.has(parsed.errcode)) {
-        clearWecomMcpCategoryCache(accountId, category);
-        return;
-      }
-    } catch {
-      // Ignore non-JSON content.
-    }
-  }
-}
-
-async function handleCall(
-  accountId: string,
-  category: string,
-  method: string,
-  args: Record<string, unknown>,
-): Promise<unknown> {
-  const result = await sendJsonRpc(accountId, category, "tools/call", {
-    name: method,
-    arguments: args,
-  });
-  checkBizErrorAndClearCache(result, accountId, category);
-  return result;
-}
-
-export function createWeComMcpToolFactory(): OpenClawPluginToolFactory {
-  return (toolContext: OpenClawPluginToolContext) => {
-    if (toolContext.messageChannel !== "wecom") {
-      return null;
-    }
-    const accountId = extractToolAccountId(toolContext);
-    const source = resolveWecomSourceSnapshot({
-      accountId,
-      sessionKey: toolContext.sessionKey,
-      sessionId: toolContext.sessionId,
-    });
-    if (!source || source.source !== "bot-ws") {
-      return null;
-    }
-
-    return {
-      name: "wecom_mcp",
-      label: "WeCom MCP",
-      description:
-        "企业微信 Bot WS MCP 工具。仅在 WeCom Bot WS 会话中可用，用于列出和调用企业微信 MCP 能力。",
-      parameters: {
-        type: "object" as const,
-        properties: {
-          action: {
-            type: "string",
-            enum: ["list", "call"],
-            description: "操作类型：list 或 call",
-          },
-          category: {
-            type: "string",
-            description: "MCP 品类，如 contact、todo、meeting、doc",
-          },
-          method: {
-            type: "string",
-            description: "action=call 时要调用的工具方法名",
-          },
-          args: {
-            type: "string",
-            description: "action=call 时传入的 JSON 字符串参数，默认 {}",
-          },
-        },
-        required: ["action", "category"],
-      },
-      async execute(_toolCallId: string, rawParams: unknown) {
-        try {
-          const params = rawParams as WecomMcpParams;
-          const effectiveAccountId = extractToolAccountId(toolContext);
-          if (!effectiveAccountId) {
-            throw new Error("当前会话缺少 WeCom accountId，无法调用 wecom_mcp。");
-          }
-
-          if (params.action === "list") {
-            return textResult(await handleList(effectiveAccountId, params.category));
-          }
-          if (!params.method) {
-            return textResult({ error: "action=call 时必须提供 method" });
-          }
-          return textResult(
-            await handleCall(
-              effectiveAccountId,
-              params.category,
-              params.method,
-              parseArgs(params.args),
-            ),
-          );
-        } catch (error) {
-          return errorResult(error);
-        }
-      },
-    };
-  };
-}

package/src/capability/mcp/transport.ts

@@ -1,394 +0,0 @@
-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;
-  }
-}