import type { AgentMessage } from "@mariozechner/pi-agent-core";

import type { AssistantMessage, ToolResultMessage, UserMessage } from "@mariozechner/pi-ai";

import type { AssistantMessage, UserMessage } from "@mariozechner/pi-ai";

import { ZERO_USAGE_FIXTURE } from "./usage-fixtures.js";

export function castAgentMessage(message: unknown): AgentMessage {

...overrides,

};

}

export function makeAgentToolResultMessage(

overrides: Partial<ToolResultMessage> &

Pick<ToolResultMessage, "toolCallId" | "toolName" | "content">,

): ToolResultMessage {

const { toolCallId, toolName, content, ...rest } = overrides;

return {

role: "toolResult",

toolCallId,

toolName,

content,

isError: false,

timestamp: 0,

...rest,

};

}

import { mkdirSync, writeFileSync, type RmOptions } from "node:fs";

import { mkdirSync, type RmOptions } from "node:fs";

import { mkdir, mkdtemp, rm } from "node:fs/promises";

import { tmpdir } from "node:os";

import path from "node:path";

createTempDirSync,

};

}

export function createBundledPluginPublicSurfaceFixture(params: {

createTempDirSync: (prefix: string) => string;

marker: string;

prefix: string;

}) {

const rootDir = params.createTempDirSync(params.prefix);

mkdirSync(path.join(rootDir, "demo"), { recursive: true });

writeFileSync(

path.join(rootDir, "demo", "api.js"),

`export const marker = ${JSON.stringify(params.marker)};\n`,

"utf8",

);

return rootDir;

}

export function createThrowingBundledPluginPublicSurfaceFixture(params: {

createTempDirSync: (prefix: string) => string;

prefix: string;

}) {

const rootDir = params.createTempDirSync(params.prefix);

mkdirSync(path.join(rootDir, "bad"), { recursive: true });

writeFileSync(

path.join(rootDir, "bad", "api.js"),

`throw new Error("plugin load failure");\n`,

"utf8",

);

return rootDir;

}

return pluginRoot;

}

export async function writeBundleTextFiles(

rootDir: string,

files: Readonly<Record<string, string>>,

) {

await Promise.all(

Object.entries(files).map(async ([relativePath, contents]) => {

const filePath = path.join(rootDir, relativePath);

await fs.mkdir(path.dirname(filePath), { recursive: true });

await fs.writeFile(filePath, contents, "utf-8");

}),

);

}

export function createEnabledPluginEntries(pluginIds: readonly string[]) {

return Object.fromEntries(pluginIds.map((pluginId) => [pluginId, { enabled: true }]));

}