@sunnoy/wecom 2.1.0 → 2.2.1

package/wecom/constants.js

@@ -26,11 +26,91 @@ export const REQID_FLUSH_DEBOUNCE_MS = 1_000;
 export const PENDING_REPLY_TTL_MS = 5 * 60 * 1000;
 export const PENDING_REPLY_MAX_SIZE = 50;
 
+// WeCom stream messages expire ~6 minutes after creation. Rotate the stream
+// before hitting that hard limit so the user never sees a dead stream.
+export const STREAM_MAX_LIFETIME_MS = 5 * 60 * 1000;
+
+export const IMAGE_MAX_BYTES = 10 * 1024 * 1024;
+export const VIDEO_MAX_BYTES = 10 * 1024 * 1024;
+export const VOICE_MAX_BYTES = 2 * 1024 * 1024;
+export const FILE_MAX_BYTES = 20 * 1024 * 1024;
+export const ABSOLUTE_MAX_BYTES = FILE_MAX_BYTES;
+
 export const DEFAULT_MEDIA_MAX_MB = 5;
 export const TEXT_CHUNK_LIMIT = 4000;
-export const DEFAULT_WELCOME_MESSAGE = ["你好，我是 AI 助手。", "", "可用命令：", "/new", "/compact", "/help", "/status"].join(
-  "\n",
-);
+export const DEFAULT_WELCOME_MESSAGES = [
+  [
+    "新的一天，元气满满！🌞",
+    "",
+    "你可以通过斜杠指令管理会话：",
+    "/new 新建对话",
+    "/compact 压缩对话",
+    "/help 帮助",
+    "/status 查看状态",
+    "/reasoning stream 打开思考动画",
+  ].join("\n"),
+  [
+    "终于唤醒我啦，我已经准备就绪！😄",
+    "",
+    "试试这些常用指令：",
+    "/new 新建对话",
+    "/compact 压缩对话",
+    "/help 帮助",
+    "/status 查看状态",
+    "/reasoning stream 打开思考动画",
+  ].join("\n"),
+  [
+    "欢迎回来，准备开始今天的工作吧！✨",
+    "",
+    "会话管理指令：",
+    "/new 新建对话",
+    "/compact 压缩对话",
+    "/help 帮助",
+    "/status 查看状态",
+    "/reasoning stream 打开思考动画",
+  ].join("\n"),
+  [
+    "嗨，我已经在线！🤖",
+    "",
+    "你可以先试试这些命令：",
+    "/new 新建对话",
+    "/compact 压缩对话",
+    "/help 帮助",
+    "/status 查看状态",
+    "/reasoning stream 打开思考动画",
+  ].join("\n"),
+  [
+    "今天也一起高效开工吧！🚀",
+    "",
+    "先来看看这些指令：",
+    "/new 新建对话",
+    "/compact 压缩对话",
+    "/help 帮助",
+    "/status 查看状态",
+    "/reasoning stream 打开思考动画",
+  ].join("\n"),
+  [
+    "叮咚，你的数字助手已就位！🎉",
+    "",
+    "常用操作给你备好了：",
+    "/new 新建对话",
+    "/compact 压缩对话",
+    "/help 帮助",
+    "/status 查看状态",
+    "/reasoning stream 打开思考动画",
+  ].join("\n"),
+  [
+    "灵感加载完成，随时可以开聊！💡",
+    "",
+    "你可以这样开始：",
+    "/new 新建对话",
+    "/compact 压缩对话",
+    "/help 帮助",
+    "/status 查看状态",
+    "/reasoning stream 打开思考动画",
+  ].join("\n"),
+];
+export const DEFAULT_WELCOME_MESSAGE = DEFAULT_WELCOME_MESSAGES[0];
 
 export const MEDIA_CACHE_DIR = join(process.env.HOME || "/tmp", ".openclaw", "media", "wecom");
 

package/wecom/mcp-config.js

@@ -0,0 +1,146 @@
+import os from "node:os";
+import path from "node:path";
+import { mkdir, readFile, rename, writeFile } from "node:fs/promises";
+import { generateReqId } from "@wecom/aibot-node-sdk";
+import { logger } from "../logger.js";
+
+const MCP_GET_CONFIG_CMD = "aibot_get_mcp_config";
+const MCP_CONFIG_FETCH_TIMEOUT_MS = 15_000;
+const MCP_CONFIG_KEY = "doc";
+const DEFAULT_MCP_TRANSPORT = "streamable-http";
+
+let mcpConfigWriteQueue = Promise.resolve();
+
+function withTimeout(promise, timeoutMs, message) {
+  if (!timeoutMs || !Number.isFinite(timeoutMs) || timeoutMs <= 0) {
+    return promise;
+  }
+
+  let timer = null;
+  const timeout = new Promise((_, reject) => {
+    timer = setTimeout(() => reject(new Error(message ?? `Timed out after ${timeoutMs}ms`)), timeoutMs);
+  });
+
+  promise.catch(() => {});
+
+  return Promise.race([promise, timeout]).finally(() => {
+    if (timer) {
+      clearTimeout(timer);
+    }
+  });
+}
+
+function getWecomConfigPath() {
+  return path.join(os.homedir(), ".openclaw", "wecomConfig", "config.json");
+}
+
+function resolveMcpTransport(body = {}) {
+  const candidate = String(
+    body.transport_type ??
+      body.transportType ??
+      body.config_type ??
+      body.configType ??
+      body.type ??
+      "",
+  )
+    .trim()
+    .toLowerCase();
+
+  return candidate || DEFAULT_MCP_TRANSPORT;
+}
+
+async function readJsonFile(filePath, fallback = {}) {
+  try {
+    const raw = await readFile(filePath, "utf8");
+    return JSON.parse(raw);
+  } catch (error) {
+    if (error?.code === "ENOENT") {
+      return fallback;
+    }
+    throw error;
+  }
+}
+
+async function writeJsonFileAtomically(filePath, value) {
+  const dir = path.dirname(filePath);
+  await mkdir(dir, { recursive: true });
+  const tempPath = `${filePath}.${process.pid}.${Date.now()}.tmp`;
+  await writeFile(tempPath, `${JSON.stringify(value, null, 2)}\n`, "utf8");
+  await rename(tempPath, filePath);
+}
+
+export async function fetchMcpConfig(wsClient) {
+  if (!wsClient || typeof wsClient.reply !== "function") {
+    throw new Error("WS client does not support MCP config requests");
+  }
+
+  const reqId = generateReqId("mcp_config");
+  const response = await withTimeout(
+    wsClient.reply({ headers: { req_id: reqId } }, { biz_type: MCP_CONFIG_KEY }, MCP_GET_CONFIG_CMD),
+    MCP_CONFIG_FETCH_TIMEOUT_MS,
+    `MCP config fetch timed out after ${MCP_CONFIG_FETCH_TIMEOUT_MS}ms`,
+  );
+
+  if (response?.errcode && response.errcode !== 0) {
+    throw new Error(`MCP config request failed: errcode=${response.errcode}, errmsg=${response.errmsg ?? "unknown"}`);
+  }
+
+  const body = response?.body;
+  if (!body?.url) {
+    throw new Error("MCP config response missing required 'url' field");
+  }
+
+  return {
+    key: MCP_CONFIG_KEY,
+    type: resolveMcpTransport(body),
+    url: body.url,
+    isAuthed: body.is_authed,
+  };
+}
+
+export async function saveMcpConfig(config, runtime) {
+  const configPath = getWecomConfigPath();
+
+  const saveTask = mcpConfigWriteQueue.then(async () => {
+    const current = await readJsonFile(configPath, {});
+    if (!current.mcpConfig || typeof current.mcpConfig !== "object") {
+      current.mcpConfig = {};
+    }
+
+    current.mcpConfig[config.key || MCP_CONFIG_KEY] = {
+      type: config.type,
+      url: config.url,
+    };
+
+    await writeJsonFileAtomically(configPath, current);
+    runtime?.log?.(`[WeCom] MCP config saved to ${configPath}`);
+  });
+
+  mcpConfigWriteQueue = saveTask.catch(() => {});
+  return saveTask;
+}
+
+export async function fetchAndSaveMcpConfig(wsClient, accountId, runtime) {
+  try {
+    runtime?.log?.(`[${accountId}] Fetching MCP config...`);
+    const config = await fetchMcpConfig(wsClient);
+    runtime?.log?.(
+      `[${accountId}] MCP config fetched: url=${config.url}, type=${config.type}, is_authed=${config.isAuthed ?? "N/A"}`,
+    );
+    await saveMcpConfig(config, runtime);
+  } catch (error) {
+    if (typeof wsClient?.reply !== "function") {
+      logger.debug?.(`[${accountId}] Skipping MCP config fetch because WS client has no reply() support`);
+      return;
+    }
+    runtime?.error?.(`[${accountId}] Failed to fetch/save MCP config: ${String(error)}`);
+  }
+}
+
+export const mcpConfigTesting = {
+  getWecomConfigPath,
+  resolveMcpTransport,
+  resetWriteQueue() {
+    mcpConfigWriteQueue = Promise.resolve();
+  },
+};