import type { ToolCall } from "../types";
import { getDialectDefinition } from "./factory";
import type { Dialect, InbandTool } from "./types";

const INTENT_PLACEHOLDER = "…";

export function renderToolExamples(tool: InbandTool, dialect: Dialect, intentField?: string): string {
	const examples = tool.examples;
	if (!examples?.length) return "";
	const definition = getDialectDefinition(dialect);
	const renderCall = (args: Record<string, unknown>): string => {
		// When intent tracing injects `i` into the schema, examples must show a
		// placeholder so the model learns to emit it. Keep it first, matching the
		// schema injection order.
		const finalArgs = intentField ? { [intentField]: INTENT_PLACEHOLDER, ...args } : args;
		const call: ToolCall = {
			type: "toolCall",
			id: "example",
			name: tool.name,
			arguments: finalArgs,
		};
		return `<example>\n${definition.renderToolCall(call, { tools: [tool], example: true }).trim()}\n</example>`;
	};
	const parts = examples.map(ex => {
		const head = ex.caption ? `# ${ex.caption}\n` : "";
		if ("call" in ex) return head + renderCall(ex.call);
		if ("good" in ex) {
			return `${head}WRONG:\n${renderCall(ex.bad)}\nRIGHT:\n${renderCall(ex.good)}`;
		}
		return head.trimEnd() + (ex.note ? `\n${ex.note}` : "");
	});
	return `<examples>\n${parts.join("\n")}\n</examples>`;
}