@yanhaidao/wecom 2.4.160 → 2.5.110

package/src/transport/bot-ws/inbound.test.ts

@@ -1,290 +0,0 @@
-import { describe, expect, it } from "vitest";
-
-import { mapBotWsFrameToInboundEvent } from "./inbound.js";
-import type { ResolvedBotAccount } from "../../types/index.js";
-
-function createBotAccount(): ResolvedBotAccount {
-  return {
-    accountId: "haidao",
-    configured: true,
-    primaryTransport: "ws",
-    wsConfigured: true,
-    webhookConfigured: false,
-    config: {},
-    ws: {
-      botId: "bot-id",
-      secret: "secret",
-    },
-    token: "",
-    encodingAESKey: "",
-    receiveId: "",
-    botId: "bot-id",
-    secret: "secret",
-  };
-}
-
-describe("mapBotWsFrameToInboundEvent", () => {
-  it("includes quote content in text events", () => {
-    const event = mapBotWsFrameToInboundEvent({
-      account: createBotAccount(),
-      frame: {
-        cmd: "aibot_msg_callback",
-        headers: { req_id: "req-1" },
-        body: {
-          msgid: "msg-1",
-          msgtype: "text",
-          chattype: "group",
-          chatid: "group-1",
-          from: { userid: "user-1" },
-          text: { content: "@daodao 这个线索价值" },
-          quote: {
-            msgtype: "text",
-            text: { content: "原始引用内容" },
-          },
-        },
-      },
-    });
-
-    expect(event.text).toBe("@daodao 这个线索价值\n\n> 原始引用内容");
-  });
-
-  it("extracts attachments from mixed events", () => {
-    const event = mapBotWsFrameToInboundEvent({
-      account: createBotAccount(),
-      frame: {
-        cmd: "aibot_msg_callback",
-        headers: { req_id: "req-2" },
-        body: {
-          msgid: "msg-2",
-          msgtype: "mixed",
-          chattype: "group",
-          chatid: "group-2",
-          from: { userid: "user-2" },
-          mixed: {
-            msg_item: [
-              {
-                msgtype: "text",
-                text: { content: "来看看这张图" },
-              },
-              {
-                msgtype: "image",
-                image: { url: "https://example.com/image.jpg", aeskey: "mock-aes-key" },
-              },
-              {
-                msgtype: "file",
-                file: { url: "https://example.com/doc.pdf", aeskey: "mock-file-key" },
-              },
-              {
-                msgtype: "video",
-                video: { url: "https://example.com/demo.mp4", aeskey: "mock-video-key" },
-              },
-            ],
-          },
-        },
-      },
-    });
-
-    expect(event.attachments).toBeDefined();
-    expect(event.attachments).toHaveLength(3);
-    expect(event.attachments![0]).toEqual({
-      name: "image",
-      remoteUrl: "https://example.com/image.jpg",
-      aesKey: "mock-aes-key",
-    });
-    expect(event.attachments![1]).toEqual({
-      name: "file",
-      remoteUrl: "https://example.com/doc.pdf",
-      aesKey: "mock-file-key",
-    });
-    expect(event.attachments![2]).toEqual({
-      name: "video",
-      remoteUrl: "https://example.com/demo.mp4",
-      aesKey: "mock-video-key",
-    });
-  });
-
-  it("extracts attachment from quote file in text events", () => {
-    const event = mapBotWsFrameToInboundEvent({
-      account: createBotAccount(),
-      frame: {
-        cmd: "aibot_msg_callback",
-        headers: { req_id: "req-3" },
-        body: {
-          msgid: "msg-3",
-          msgtype: "text",
-          chattype: "single",
-          from: { userid: "user-3" },
-          text: { content: "请读取这个引用文件" },
-          quote: {
-            msgtype: "file",
-            file: { url: "https://example.com/quoted.pdf", aeskey: "quoted-file-key" },
-          },
-        },
-      },
-    });
-
-    expect(event.inboundKind).toBe("text");
-    expect(event.attachments).toEqual([
-      {
-        name: "file",
-        remoteUrl: "https://example.com/quoted.pdf",
-        aesKey: "quoted-file-key",
-      },
-    ]);
-  });
-
-  it("maps top-level video to video inbound kind and attachment", () => {
-    const event = mapBotWsFrameToInboundEvent({
-      account: createBotAccount(),
-      frame: {
-        cmd: "aibot_msg_callback",
-        headers: { req_id: "req-4" },
-        body: {
-          msgid: "msg-4",
-          msgtype: "video",
-          chattype: "single",
-          from: { userid: "user-4" },
-          video: { url: "https://example.com/top-video.mp4", aeskey: "top-video-key" },
-        },
-      },
-    });
-
-    expect(event.inboundKind).toBe("video");
-    expect(event.attachments).toEqual([
-      {
-        name: "video",
-        remoteUrl: "https://example.com/top-video.mp4",
-        aesKey: "top-video-key",
-      },
-    ]);
-  });
-
-  it("extracts attachment from quote image in text events", () => {
-    const event = mapBotWsFrameToInboundEvent({
-      account: createBotAccount(),
-      frame: {
-        cmd: "aibot_msg_callback",
-        headers: { req_id: "req-5" },
-        body: {
-          msgid: "msg-5",
-          msgtype: "text",
-          chattype: "single",
-          from: { userid: "user-5" },
-          text: { content: "请看这张引用图片" },
-          quote: {
-            msgtype: "image",
-            image: { url: "https://example.com/quoted.jpg", aeskey: "quoted-image-key" },
-          },
-        },
-      },
-    });
-
-    expect(event.inboundKind).toBe("text");
-    expect(event.attachments).toEqual([
-      {
-        name: "image",
-        remoteUrl: "https://example.com/quoted.jpg",
-        aesKey: "quoted-image-key",
-      },
-    ]);
-  });
-
-  it("extracts attachment from quote video in text events", () => {
-    const event = mapBotWsFrameToInboundEvent({
-      account: createBotAccount(),
-      frame: {
-        cmd: "aibot_msg_callback",
-        headers: { req_id: "req-6" },
-        body: {
-          msgid: "msg-6",
-          msgtype: "voice",
-          chattype: "single",
-          from: { userid: "user-6" },
-          voice: { content: "语音转写内容" },
-          quote: {
-            msgtype: "video",
-            video: { url: "https://example.com/quoted.mp4", aeskey: "quoted-video-key" },
-          },
-        },
-      },
-    });
-
-    expect(event.inboundKind).toBe("voice");
-    expect(event.attachments).toEqual([
-      {
-        name: "video",
-        remoteUrl: "https://example.com/quoted.mp4",
-        aesKey: "quoted-video-key",
-      },
-    ]);
-  });
-
-  it("prioritizes top-level media over quote media in WS events", () => {
-    const event = mapBotWsFrameToInboundEvent({
-      account: createBotAccount(),
-      frame: {
-        cmd: "aibot_msg_callback",
-        headers: { req_id: "req-7" },
-        body: {
-          msgid: "msg-7",
-          msgtype: "image",
-          chattype: "group",
-          chatid: "group-7",
-          from: { userid: "user-7" },
-          image: { url: "https://example.com/top.jpg", aeskey: "top-image-key" },
-          quote: {
-            msgtype: "file",
-            file: { url: "https://example.com/quoted.pdf", aeskey: "quoted-file-key" },
-          },
-        },
-      },
-    });
-
-    // Should only include top-level image, not quote file
-    expect(event.inboundKind).toBe("image");
-    expect(event.attachments).toEqual([
-      {
-        name: "image",
-        remoteUrl: "https://example.com/top.jpg",
-        aesKey: "top-image-key",
-      },
-    ]);
-  });
-
-  it("extracts first image from quote.mixed in text events", () => {
-    const event = mapBotWsFrameToInboundEvent({
-      account: createBotAccount(),
-      frame: {
-        cmd: "aibot_msg_callback",
-        headers: { req_id: "req-8" },
-        body: {
-          msgid: "msg-8",
-          msgtype: "text",
-          chattype: "single",
-          from: { userid: "user-8" },
-          text: { content: "这个引用有图文内容" },
-          quote: {
-            msgtype: "mixed",
-            mixed: {
-              msg_item: [
-                { msgtype: "text", text: { content: "一些文本" } },
-                { msgtype: "image", image: { url: "https://example.com/mixed-img1.jpg", aeskey: "mixed-key-1" } },
-                { msgtype: "image", image: { url: "https://example.com/mixed-img2.jpg", aeskey: "mixed-key-2" } },
-              ],
-            },
-          },
-        },
-      },
-    });
-
-    expect(event.inboundKind).toBe("text");
-    // Should only extract first image from quote.mixed
-    expect(event.attachments).toEqual([
-      {
-        name: "image",
-        remoteUrl: "https://example.com/mixed-img1.jpg",
-        aesKey: "mixed-key-1",
-      },
-    ]);
-  });
-});

package/src/transport/bot-ws/inbound.ts

@@ -1,163 +0,0 @@
-import type { BaseMessage, EventMessage, WsFrame } from "@wecom/aibot-node-sdk";
-
-import { buildInboundBody } from "../bot-webhook/message-shape.js";
-import type {
-  ResolvedBotAccount,
-  UnifiedInboundEvent,
-  WecomBotInboundMessage,
-  WecomInboundKind,
-} from "../../types/index.js";
-
-function resolveInboundKind(message: BaseMessage | EventMessage): WecomInboundKind {
-  if (message.msgtype === "event") {
-    const eventType = String((message as EventMessage).event?.eventtype ?? "").trim();
-    if (eventType === "enter_chat") return "welcome";
-    if (eventType === "template_card_event") return "template-card-event";
-    return "event";
-  }
-  switch (message.msgtype) {
-    case "image":
-      return "image";
-    case "file":
-      return "file";
-    case "voice":
-      return "voice";
-    case "video":
-      return "video";
-    case "mixed":
-      return "mixed";
-    default:
-      return "text";
-  }
-}
-
-function pushAttachment(
-  list: NonNullable<UnifiedInboundEvent["attachments"]>,
-  name: "image" | "file" | "video",
-  remoteUrl?: string,
-  aesKey?: string,
-): void {
-  if (!remoteUrl) {
-    return;
-  }
-  list.push({ name, remoteUrl, aesKey });
-}
-
-function resolveEventText(message: BaseMessage | EventMessage, account: ResolvedBotAccount): string {
-  if (message.msgtype !== "event") {
-    return buildInboundBody(message as WecomBotInboundMessage);
-  }
-
-  const event = message as EventMessage;
-  if (event.event?.eventtype === "enter_chat" && account.config.welcomeText) {
-    return account.config.welcomeText;
-  }
-  return `[event:${String(event.event?.eventtype ?? "unknown")}]`;
-}
-
-export function mapBotWsFrameToInboundEvent(params: {
-  account: ResolvedBotAccount;
-  frame: WsFrame<BaseMessage | EventMessage>;
-}): UnifiedInboundEvent {
-  const { account, frame } = params;
-  const body = frame.body;
-  if (!body) {
-    throw new Error("Bot WS frame body is required");
-  }
-  const peerKind = body.chattype === "group" ? "group" : "direct";
-  const senderId = body.from?.userid ?? "unknown";
-  const peerId = peerKind === "group" ? body.chatid ?? senderId : senderId;
-  const inboundKind = resolveInboundKind(body);
-
-  let attachments: UnifiedInboundEvent["attachments"];
-  const collected: NonNullable<UnifiedInboundEvent["attachments"]> = [];
-  if (body.msgtype === "image") {
-    pushAttachment(collected, "image", (body as any).image?.url, (body as any).image?.aeskey);
-  } else if (body.msgtype === "file") {
-    pushAttachment(collected, "file", (body as any).file?.url, (body as any).file?.aeskey);
-  } else if (body.msgtype === "video") {
-    pushAttachment(collected, "video", (body as any).video?.url, (body as any).video?.aeskey);
-  } else if (body.msgtype === "mixed") {
-    const items = (body as any).mixed?.msg_item;
-    if (Array.isArray(items)) {
-      for (const item of items) {
-        const itemType = String(item.msgtype ?? "").toLowerCase();
-        if (itemType === "image") {
-          pushAttachment(collected, "image", item.image?.url, item.image?.aeskey);
-        } else if (itemType === "file") {
-          pushAttachment(collected, "file", item.file?.url, item.file?.aeskey);
-        } else if (itemType === "video") {
-          pushAttachment(collected, "video", item.video?.url, item.video?.aeskey);
-        }
-      }
-    }
-  }
-
-  // 新增支持：如果没有顶层媒体，尝试从引用中提取媒体附件
-  // 优先级：quote.image/file/video 优先，其次 quote.mixed 中第一个图片
-  if (collected.length === 0) {
-    const quote = (body as any).quote;
-    if (quote) {
-      const quoteType = String(quote.msgtype ?? "").toLowerCase();
-      // 处理单个媒体类型的引用
-      if (quoteType === "image") {
-        pushAttachment(collected, "image", quote.image?.url, quote.image?.aeskey);
-      } else if (quoteType === "file") {
-        pushAttachment(collected, "file", quote.file?.url, quote.file?.aeskey);
-      } else if (quoteType === "video") {
-        pushAttachment(collected, "video", quote.video?.url, quote.video?.aeskey);
-      } 
-      // 处理图文混合类型：只提取第一个图片以保持与 webhook 一致
-      else if (quoteType === "mixed" && Array.isArray(quote.mixed?.msg_item)) {
-        for (const item of quote.mixed.msg_item) {
-          const itemType = String(item.msgtype ?? "").toLowerCase();
-          if (itemType === "image") {
-            pushAttachment(collected, "image", item.image?.url, item.image?.aeskey);
-            break;
-          }
-        }
-      }
-    }
-  }
-
-  if (collected.length > 0) {
-    attachments = collected;
-  }
-
-  return {
-    accountId: account.accountId,
-    capability: "bot",
-    transport: "bot-ws",
-    inboundKind,
-    messageId: body.msgid,
-    conversation: {
-      accountId: account.accountId,
-      peerKind,
-      peerId,
-      senderId,
-    },
-    senderName: senderId,
-    text: resolveEventText(body, account),
-    timestamp: typeof body.create_time === "number" ? body.create_time : Date.now(),
-    raw: {
-      transport: "bot-ws",
-      command: frame.cmd,
-      headers: frame.headers,
-      body,
-      envelopeType: "ws",
-    },
-    replyContext: {
-      transport: "bot-ws",
-      accountId: account.accountId,
-      reqId: frame.headers.req_id,
-      raw: {
-        transport: "bot-ws",
-        command: frame.cmd,
-        headers: frame.headers,
-        body,
-        envelopeType: "ws",
-      },
-    },
-    ...(attachments && { attachments }),
-  };
-}

package/src/transport/bot-ws/media.test.ts

@@ -1,44 +0,0 @@
-import type { WSClient } from "@wecom/aibot-node-sdk";
-import { fetchRemoteMedia } from "openclaw/plugin-sdk/media-runtime";
-import { beforeEach, describe, expect, it, vi } from "vitest";
-import { uploadAndSendBotWsMedia } from "./media.js";
-
-vi.mock("openclaw/plugin-sdk/media-runtime", () => ({
-  assertLocalMediaAllowed: vi.fn(),
-  detectMime: vi.fn(),
-  fetchRemoteMedia: vi.fn(),
-}));
-
-describe("uploadAndSendBotWsMedia", () => {
-  const fetchRemoteMediaMock = vi.mocked(fetchRemoteMedia);
-
-  beforeEach(() => {
-    fetchRemoteMediaMock.mockReset();
-    fetchRemoteMediaMock.mockResolvedValue({
-      buffer: Buffer.from("png"),
-      contentType: "image/png",
-      fileName: "sample.png",
-    } as never);
-  });
-
-  it("passes the configured maxBytes to outbound media loading", async () => {
-    const wsClient = {
-      uploadMedia: vi.fn().mockResolvedValue({ media_id: "media-1" }),
-      sendMediaMessage: vi.fn().mockResolvedValue({ headers: { req_id: "req-1" } }),
-    } as unknown as WSClient;
-
-    await uploadAndSendBotWsMedia({
-      wsClient,
-      chatId: "hidao",
-      mediaUrl: "https://example.com/sample.png",
-      maxBytes: 42 * 1024 * 1024,
-    });
-
-    expect(fetchRemoteMediaMock).toHaveBeenCalledWith(
-      expect.objectContaining({
-        url: "https://example.com/sample.png",
-        maxBytes: 42 * 1024 * 1024,
-      }),
-    );
-  });
-});