pi-roundtable 0.26.2 → 0.27.0

package/src/testing.ts

@@ -68,6 +68,8 @@ import {
 	type ConversationTurns,
 	conversationTurns,
 } from "./core/routing/conversation-turns.ts";
+import { MessageObservers } from "./core/routing/message-observers.ts";
+import { PeerBots, type PeerBotsConfig } from "./core/routing/peer-bots.ts";
 import { surfacePort } from "./core/routing/surface-port.ts";
 import {
 	AGENTS,
@@ -258,6 +260,8 @@ export interface TestPluginOptions {
 	apiKeys?: Readonly<Record<string, string>>;
 	/** How long the router holds a bare forward for the message that follows it; the host option `conversations.forwardJoinMs`. */
 	forwardJoinMs?: number;
+	/** The peer bots the router answers, as the host option `peerBots` (`discord.peerBots`); default none. */
+	peerBots?: PeerBotsConfig;
 	/** The addresses `context.listener(id)` reports; a plugin that reads one gets `undefined` for any not listed. */
 	listeners?: readonly ListenerConfig[];
 }
@@ -384,6 +388,10 @@ export async function testPlugin(
 	const conversations: ConversationPort = {
 		handle: async (message) =>
 			(router ?? unlinked("conversations")).handle(message),
+		messageDeleted: async (channel, id, space) =>
+			(router ?? unlinked("conversations")).messageDeleted(channel, id, space),
+		messageEdited: async (message) =>
+			(router ?? unlinked("conversations")).messageEdited(message),
 		background: async (turn) =>
 			(router ?? unlinked("conversations")).background(turn),
 		runsAs: async (turn) => (router ?? unlinked("conversations")).runsAs(turn),
@@ -407,6 +415,7 @@ export async function testPlugin(
 			? [...registry.surfaces, ...(options.surfaces ?? [])]
 			: unlinked("surfaces"),
 	);
+	const observers = new MessageObservers(logger, () => surfaces.roomSurfaces());
 	const sink = {
 		turnStarted: (turn: TurnEvent) => {
 			events.push({ name: "turnStarted", turn });
@@ -538,6 +547,7 @@ export async function testPlugin(
 		}),
 		surfaces,
 		turns,
+		observe: (filter, observer) => observers.add(filter, observer),
 		database: () => {
 			if (!options.database) throw new PluginError("no database is configured");
 			return options.database;
@@ -602,6 +612,8 @@ export async function testPlugin(
 				: undefined),
 		queue,
 		logger,
+		observers,
+		...(options.peerBots ? { peerBots: new PeerBots(options.peerBots) } : {}),
 		...(options.forwardJoinMs === undefined
 			? {}
 			: { forwardJoinMs: options.forwardJoinMs }),
@@ -640,7 +652,12 @@ export async function testPlugin(
 		throw error;
 	}
 	for (const service of registry.services) await service.start?.();
-	for (const surface of options.surfaces ?? []) await surface.start(deliver);
+	for (const surface of options.surfaces ?? [])
+		await surface.start(deliver, {
+			deleted: (channel, id, space) =>
+				void conversations.messageDeleted?.(channel, id, space),
+			edited: (message) => void conversations.messageEdited?.(message),
+		});
 	let stopped = false;
 	const files: { channel: ChannelKey; file: ReplyFile }[] = [];
 	const images: { channel: ChannelKey; image: ClientPart }[] = [];