import type { Client } from "@buape/carbon"; import { ChannelType, MessageType } from "@buape/carbon"; import { Routes } from "discord-api-types/v10"; import { beforeEach, describe, expect, it, vi } from "vitest"; const sendMock = vi.fn(); const reactMock = vi.fn(); const updateLastRouteMock = vi.fn(); const dispatchMock = vi.fn(); const readAllowFromStoreMock = vi.fn(); const upsertPairingRequestMock = vi.fn(); vi.mock("./send.js", () => ({ sendMessageDiscord: (...args: unknown[]) => sendMock(...args), reactMessageDiscord: async (...args: unknown[]) => { reactMock(...args); }, })); vi.mock("../auto-reply/reply/dispatch-from-config.js", () => ({ dispatchReplyFromConfig: (...args: unknown[]) => dispatchMock(...args), })); vi.mock("../pairing/pairing-store.js", () => ({ readProviderAllowFromStore: (...args: unknown[]) => readAllowFromStoreMock(...args), upsertProviderPairingRequest: (...args: unknown[]) => upsertPairingRequestMock(...args), })); vi.mock("../config/sessions.js", async (importOriginal) => { const actual = await importOriginal(); return { ...actual, resolveStorePath: vi.fn(() => "/tmp/clawdbot-sessions.json"), updateLastRoute: (...args: unknown[]) => updateLastRouteMock(...args), resolveSessionKey: vi.fn(), }; }); beforeEach(() => { sendMock.mockReset().mockResolvedValue(undefined); updateLastRouteMock.mockReset(); dispatchMock.mockReset().mockImplementation(async ({ dispatcher }) => { dispatcher.sendFinalReply({ text: "hi" }); return { queuedFinal: true, counts: { final: 1 } }; }); readAllowFromStoreMock.mockReset().mockResolvedValue([]); upsertPairingRequestMock .mockReset() .mockResolvedValue({ code: "PAIRCODE", created: true }); vi.resetModules(); }); describe("discord tool result dispatch", () => { it("sends status replies with responsePrefix", async () => { const { createDiscordMessageHandler } = await import("./monitor.js"); const cfg = { agents: { defaults: { model: "anthropic/claude-opus-4-5", workspace: "/tmp/clawd", }, }, session: { store: "/tmp/clawdbot-sessions.json" }, messages: { responsePrefix: "PFX" }, discord: { dm: { enabled: true, policy: "open" } }, } as ReturnType; const runtimeError = vi.fn(); const handler = createDiscordMessageHandler({ cfg, discordConfig: cfg.discord, accountId: "default", token: "token", runtime: { log: vi.fn(), error: runtimeError, exit: (code: number): never => { throw new Error(`exit ${code}`); }, }, botUserId: "bot-id", guildHistories: new Map(), historyLimit: 0, mediaMaxBytes: 10_000, textLimit: 2000, replyToMode: "off", dmEnabled: true, groupDmEnabled: false, }); const client = { fetchChannel: vi.fn().mockResolvedValue({ type: ChannelType.DM, name: "dm", }), } as unknown as Client; await handler( { message: { id: "m1", content: "/status", channelId: "c1", timestamp: new Date().toISOString(), type: MessageType.Default, attachments: [], embeds: [], mentionedEveryone: false, mentionedUsers: [], mentionedRoles: [], author: { id: "u1", bot: false, username: "Ada" }, }, author: { id: "u1", bot: false, username: "Ada" }, guild_id: null, }, client, ); expect(runtimeError).not.toHaveBeenCalled(); expect(sendMock).toHaveBeenCalledTimes(1); expect(sendMock.mock.calls[0]?.[1]).toMatch(/^PFX /); }, 30_000); it("caches channel info lookups between messages", async () => { const { createDiscordMessageHandler } = await import("./monitor.js"); const cfg = { agents: { defaults: { model: "anthropic/claude-opus-4-5", workspace: "/tmp/clawd", }, }, session: { store: "/tmp/clawdbot-sessions.json" }, discord: { dm: { enabled: true, policy: "open" } }, } as ReturnType; const handler = createDiscordMessageHandler({ cfg, discordConfig: cfg.discord, accountId: "default", token: "token", runtime: { log: vi.fn(), error: vi.fn(), exit: (code: number): never => { throw new Error(`exit ${code}`); }, }, botUserId: "bot-id", guildHistories: new Map(), historyLimit: 0, mediaMaxBytes: 10_000, textLimit: 2000, replyToMode: "off", dmEnabled: true, groupDmEnabled: false, }); const fetchChannel = vi.fn().mockResolvedValue({ type: ChannelType.DM, name: "dm", }); const client = { fetchChannel } as unknown as Client; const baseMessage = { content: "hello", channelId: "cache-channel-1", timestamp: new Date().toISOString(), type: MessageType.Default, attachments: [], embeds: [], mentionedEveryone: false, mentionedUsers: [], mentionedRoles: [], author: { id: "u-cache", bot: false, username: "Ada" }, }; await handler( { message: { ...baseMessage, id: "m-cache-1" }, author: baseMessage.author, guild_id: null, }, client, ); await handler( { message: { ...baseMessage, id: "m-cache-2" }, author: baseMessage.author, guild_id: null, }, client, ); expect(fetchChannel).toHaveBeenCalledTimes(1); }); it("includes forwarded message snapshots in body", async () => { const { createDiscordMessageHandler } = await import("./monitor.js"); let capturedBody = ""; dispatchMock.mockImplementationOnce(async ({ ctx, dispatcher }) => { capturedBody = ctx.Body ?? ""; dispatcher.sendFinalReply({ text: "ok" }); return { queuedFinal: true, counts: { final: 1 } }; }); const cfg = { agents: { defaults: { model: "anthropic/claude-opus-4-5", workspace: "/tmp/clawd", }, }, session: { store: "/tmp/clawdbot-sessions.json" }, discord: { dm: { enabled: true, policy: "open" } }, } as ReturnType; const handler = createDiscordMessageHandler({ cfg, discordConfig: cfg.discord, accountId: "default", token: "token", runtime: { log: vi.fn(), error: vi.fn(), exit: (code: number): never => { throw new Error(`exit ${code}`); }, }, botUserId: "bot-id", guildHistories: new Map(), historyLimit: 0, mediaMaxBytes: 10_000, textLimit: 2000, replyToMode: "off", dmEnabled: true, groupDmEnabled: false, }); const client = { fetchChannel: vi.fn().mockResolvedValue({ type: ChannelType.DM, name: "dm", }), } as unknown as Client; await handler( { message: { id: "m-forward-1", content: "", channelId: "c-forward-1", timestamp: new Date().toISOString(), type: MessageType.Default, attachments: [], embeds: [], mentionedEveryone: false, mentionedUsers: [], mentionedRoles: [], author: { id: "u1", bot: false, username: "Ada" }, rawData: { message_snapshots: [ { message: { content: "forwarded hello", embeds: [], attachments: [], author: { id: "u2", username: "Bob", discriminator: "0", }, }, }, ], }, }, author: { id: "u1", bot: false, username: "Ada" }, guild_id: null, }, client, ); expect(capturedBody).toContain("[Forwarded message from @Bob]"); expect(capturedBody).toContain("forwarded hello"); }); it("uses channel id allowlists for non-thread channels with categories", async () => { const { createDiscordMessageHandler } = await import("./monitor.js"); let capturedCtx: { SessionKey?: string } | undefined; dispatchMock.mockImplementationOnce(async ({ ctx, dispatcher }) => { capturedCtx = ctx; dispatcher.sendFinalReply({ text: "hi" }); return { queuedFinal: true, counts: { final: 1 } }; }); const cfg = { agents: { defaults: { model: "anthropic/claude-opus-4-5", workspace: "/tmp/clawd", }, }, session: { store: "/tmp/clawdbot-sessions.json" }, discord: { dm: { enabled: true, policy: "open" }, guilds: { "*": { requireMention: false, channels: { c1: { allow: true } }, }, }, }, routing: { allowFrom: [] }, } as ReturnType; const handler = createDiscordMessageHandler({ cfg, discordConfig: cfg.discord, accountId: "default", token: "token", runtime: { log: vi.fn(), error: vi.fn(), exit: (code: number): never => { throw new Error(`exit ${code}`); }, }, botUserId: "bot-id", guildHistories: new Map(), historyLimit: 0, mediaMaxBytes: 10_000, textLimit: 2000, replyToMode: "off", dmEnabled: true, groupDmEnabled: false, guildEntries: { "*": { requireMention: false, channels: { c1: { allow: true } } }, }, }); const client = { fetchChannel: vi.fn().mockResolvedValue({ type: ChannelType.GuildText, name: "general", parentId: "category-1", }), rest: { get: vi.fn() }, } as unknown as Client; await handler( { message: { id: "m-category", content: "hello", channelId: "c1", timestamp: new Date().toISOString(), type: MessageType.Default, attachments: [], embeds: [], mentionedEveryone: false, mentionedUsers: [], mentionedRoles: [], author: { id: "u1", bot: false, username: "Ada", tag: "Ada#1" }, }, author: { id: "u1", bot: false, username: "Ada", tag: "Ada#1" }, member: { displayName: "Ada" }, guild: { id: "g1", name: "Guild" }, guild_id: "g1", }, client, ); expect(capturedCtx?.SessionKey).toBe("agent:main:discord:channel:c1"); }); it("replies with pairing code and sender id when dmPolicy is pairing", async () => { const { createDiscordMessageHandler } = await import("./monitor.js"); const cfg = { agents: { defaults: { model: "anthropic/claude-opus-4-5", workspace: "/tmp/clawd", }, }, session: { store: "/tmp/clawdbot-sessions.json" }, discord: { dm: { enabled: true, policy: "pairing", allowFrom: [] } }, } as ReturnType; const handler = createDiscordMessageHandler({ cfg, discordConfig: cfg.discord, accountId: "default", token: "token", runtime: { log: vi.fn(), error: vi.fn(), exit: (code: number): never => { throw new Error(`exit ${code}`); }, }, botUserId: "bot-id", guildHistories: new Map(), historyLimit: 0, mediaMaxBytes: 10_000, textLimit: 2000, replyToMode: "off", dmEnabled: true, groupDmEnabled: false, }); const client = { fetchChannel: vi.fn().mockResolvedValue({ type: ChannelType.DM, name: "dm", }), } as unknown as Client; await handler( { message: { id: "m1", content: "hello", channelId: "c1", timestamp: new Date().toISOString(), type: MessageType.Default, attachments: [], embeds: [], mentionedEveryone: false, mentionedUsers: [], mentionedRoles: [], author: { id: "u2", bot: false, username: "Ada" }, }, author: { id: "u2", bot: false, username: "Ada" }, guild_id: null, }, client, ); expect(dispatchMock).not.toHaveBeenCalled(); expect(upsertPairingRequestMock).toHaveBeenCalled(); expect(sendMock).toHaveBeenCalledTimes(1); expect(String(sendMock.mock.calls[0]?.[1] ?? "")).toContain( "Your Discord user id: u2", ); expect(String(sendMock.mock.calls[0]?.[1] ?? "")).toContain( "Pairing code: PAIRCODE", ); }, 10000); it("accepts guild messages when mentionPatterns match", async () => { const { createDiscordMessageHandler } = await import("./monitor.js"); const cfg = { agents: { defaults: { model: "anthropic/claude-opus-4-5", workspace: "/tmp/clawd", }, }, session: { store: "/tmp/clawdbot-sessions.json" }, discord: { dm: { enabled: true, policy: "open" }, groupPolicy: "open", guilds: { "*": { requireMention: true } }, }, messages: { responsePrefix: "PFX", groupChat: { mentionPatterns: ["\\bclawd\\b"] }, }, } as ReturnType; const handler = createDiscordMessageHandler({ cfg, discordConfig: cfg.discord, accountId: "default", token: "token", runtime: { log: vi.fn(), error: vi.fn(), exit: (code: number): never => { throw new Error(`exit ${code}`); }, }, botUserId: "bot-id", guildHistories: new Map(), historyLimit: 0, mediaMaxBytes: 10_000, textLimit: 2000, replyToMode: "off", dmEnabled: true, groupDmEnabled: false, guildEntries: { "*": { requireMention: true } }, }); const client = { fetchChannel: vi.fn().mockResolvedValue({ type: ChannelType.GuildText, name: "general", }), } as unknown as Client; await handler( { message: { id: "m2", content: "clawd: hello", channelId: "c1", timestamp: new Date().toISOString(), type: MessageType.Default, attachments: [], embeds: [], mentionedEveryone: false, mentionedUsers: [], mentionedRoles: [], author: { id: "u1", bot: false, username: "Ada" }, }, author: { id: "u1", bot: false, username: "Ada" }, member: { nickname: "Ada" }, guild: { id: "g1", name: "Guild" }, guild_id: "g1", }, client, ); expect(dispatchMock).toHaveBeenCalledTimes(1); expect(sendMock).toHaveBeenCalledTimes(1); }, 10000); it("forks thread sessions and injects starter context", async () => { const { createDiscordMessageHandler } = await import("./monitor.js"); let capturedCtx: | { SessionKey?: string; ParentSessionKey?: string; ThreadStarterBody?: string; ThreadLabel?: string; } | undefined; dispatchMock.mockImplementationOnce(async ({ ctx, dispatcher }) => { capturedCtx = ctx; dispatcher.sendFinalReply({ text: "hi" }); return { queuedFinal: true, counts: { final: 1 } }; }); const cfg = { agents: { defaults: { model: "anthropic/claude-opus-4-5", workspace: "/tmp/clawd", }, }, session: { store: "/tmp/clawdbot-sessions.json" }, messages: { responsePrefix: "PFX" }, discord: { dm: { enabled: true, policy: "open" }, groupPolicy: "open", guilds: { "*": { requireMention: false } }, }, } as ReturnType; const handler = createDiscordMessageHandler({ cfg, discordConfig: cfg.discord, accountId: "default", token: "token", runtime: { log: vi.fn(), error: vi.fn(), exit: (code: number): never => { throw new Error(`exit ${code}`); }, }, botUserId: "bot-id", guildHistories: new Map(), historyLimit: 0, mediaMaxBytes: 10_000, textLimit: 2000, replyToMode: "off", dmEnabled: true, groupDmEnabled: false, guildEntries: { "*": { requireMention: false } }, }); const threadChannel = { type: ChannelType.GuildText, name: "thread-name", parentId: "p1", parent: { id: "p1", name: "general" }, isThread: () => true, fetchStarterMessage: async () => ({ content: "starter message", author: { tag: "Alice#1", username: "Alice" }, createdTimestamp: Date.now(), }), }; const client = { fetchChannel: vi.fn().mockResolvedValue({ type: ChannelType.GuildText, name: "thread-name", }), rest: { get: vi.fn().mockResolvedValue({ content: "starter message", author: { id: "u1", username: "Alice", discriminator: "0001" }, timestamp: new Date().toISOString(), }), }, } as unknown as Client; await handler( { message: { id: "m4", content: "thread reply", channelId: "t1", channel: threadChannel, timestamp: new Date().toISOString(), type: MessageType.Default, attachments: [], embeds: [], mentionedEveryone: false, mentionedUsers: [], mentionedRoles: [], author: { id: "u2", bot: false, username: "Bob", tag: "Bob#2" }, }, author: { id: "u2", bot: false, username: "Bob", tag: "Bob#2" }, member: { displayName: "Bob" }, guild: { id: "g1", name: "Guild" }, guild_id: "g1", }, client, ); expect(capturedCtx?.SessionKey).toBe("agent:main:discord:channel:t1"); expect(capturedCtx?.ParentSessionKey).toBe("agent:main:discord:channel:p1"); expect(capturedCtx?.ThreadStarterBody).toContain("starter message"); expect(capturedCtx?.ThreadLabel).toContain("Discord thread #general"); }); it("treats forum threads as distinct sessions without channel payloads", async () => { const { createDiscordMessageHandler } = await import("./monitor.js"); let capturedCtx: | { SessionKey?: string; ParentSessionKey?: string; ThreadStarterBody?: string; ThreadLabel?: string; } | undefined; dispatchMock.mockImplementationOnce(async ({ ctx, dispatcher }) => { capturedCtx = ctx; dispatcher.sendFinalReply({ text: "hi" }); return { queuedFinal: true, counts: { final: 1 } }; }); const cfg = { agent: { model: "anthropic/claude-opus-4-5", workspace: "/tmp/clawd" }, session: { store: "/tmp/clawdbot-sessions.json" }, discord: { dm: { enabled: true, policy: "open" }, groupPolicy: "open", guilds: { "*": { requireMention: false } }, }, routing: { allowFrom: [] }, } as ReturnType; const handler = createDiscordMessageHandler({ cfg, discordConfig: cfg.discord, accountId: "default", token: "token", runtime: { log: vi.fn(), error: vi.fn(), exit: (code: number): never => { throw new Error(`exit ${code}`); }, }, botUserId: "bot-id", guildHistories: new Map(), historyLimit: 0, mediaMaxBytes: 10_000, textLimit: 2000, replyToMode: "off", dmEnabled: true, groupDmEnabled: false, guildEntries: { "*": { requireMention: false } }, }); const fetchChannel = vi .fn() .mockResolvedValueOnce({ type: ChannelType.PublicThread, name: "topic-1", parentId: "forum-1", }) .mockResolvedValueOnce({ type: ChannelType.GuildForum, name: "support", }); const restGet = vi.fn().mockResolvedValue({ content: "starter message", author: { id: "u1", username: "Alice", discriminator: "0001" }, timestamp: new Date().toISOString(), }); const client = { fetchChannel, rest: { get: restGet, }, } as unknown as Client; await handler( { message: { id: "m6", content: "thread reply", channelId: "t1", timestamp: new Date().toISOString(), type: MessageType.Default, attachments: [], embeds: [], mentionedEveryone: false, mentionedUsers: [], mentionedRoles: [], author: { id: "u2", bot: false, username: "Bob", tag: "Bob#2" }, }, author: { id: "u2", bot: false, username: "Bob", tag: "Bob#2" }, member: { displayName: "Bob" }, guild: { id: "g1", name: "Guild" }, guild_id: "g1", }, client, ); expect(capturedCtx?.SessionKey).toBe("agent:main:discord:channel:t1"); expect(capturedCtx?.ParentSessionKey).toBe( "agent:main:discord:channel:forum-1", ); expect(capturedCtx?.ThreadStarterBody).toContain("starter message"); expect(capturedCtx?.ThreadLabel).toContain("Discord thread #support"); expect(restGet).toHaveBeenCalledWith(Routes.channelMessage("t1", "t1")); }); it("scopes thread sessions to the routed agent", async () => { const { createDiscordMessageHandler } = await import("./monitor.js"); let capturedCtx: | { SessionKey?: string; ParentSessionKey?: string; } | undefined; dispatchMock.mockImplementationOnce(async ({ ctx, dispatcher }) => { capturedCtx = ctx; dispatcher.sendFinalReply({ text: "hi" }); return { queuedFinal: true, counts: { final: 1 } }; }); const cfg = { agents: { defaults: { model: "anthropic/claude-opus-4-5", workspace: "/tmp/clawd", }, }, session: { store: "/tmp/clawdbot-sessions.json" }, messages: { responsePrefix: "PFX" }, discord: { dm: { enabled: true, policy: "open" }, groupPolicy: "open", guilds: { "*": { requireMention: false } }, }, bindings: [ { agentId: "support", match: { provider: "discord", guildId: "g1" } }, ], } as ReturnType; const handler = createDiscordMessageHandler({ cfg, discordConfig: cfg.discord, accountId: "default", token: "token", runtime: { log: vi.fn(), error: vi.fn(), exit: (code: number): never => { throw new Error(`exit ${code}`); }, }, botUserId: "bot-id", guildHistories: new Map(), historyLimit: 0, mediaMaxBytes: 10_000, textLimit: 2000, replyToMode: "off", dmEnabled: true, groupDmEnabled: false, guildEntries: { "*": { requireMention: false } }, }); const threadChannel = { type: ChannelType.GuildText, name: "thread-name", parentId: "p1", parent: { id: "p1", name: "general" }, isThread: () => true, }; const client = { fetchChannel: vi.fn().mockResolvedValue({ type: ChannelType.GuildText, name: "thread-name", }), rest: { get: vi.fn().mockResolvedValue({ content: "starter message", author: { id: "u1", username: "Alice", discriminator: "0001" }, timestamp: new Date().toISOString(), }), }, } as unknown as Client; await handler( { message: { id: "m5", content: "thread reply", channelId: "t1", channel: threadChannel, timestamp: new Date().toISOString(), type: MessageType.Default, attachments: [], embeds: [], mentionedEveryone: false, mentionedUsers: [], mentionedRoles: [], author: { id: "u2", bot: false, username: "Bob", tag: "Bob#2" }, }, author: { id: "u2", bot: false, username: "Bob", tag: "Bob#2" }, member: { displayName: "Bob" }, guild: { id: "g1", name: "Guild" }, guild_id: "g1", }, client, ); expect(capturedCtx?.SessionKey).toBe("agent:support:discord:channel:t1"); expect(capturedCtx?.ParentSessionKey).toBe( "agent:support:discord:channel:p1", ); }); });