import type { PersistedContextFile } from "./details.js";

type TextContent = { type: "text"; text: string };

function escapeXml(value: string): string {
	return value
		.replaceAll("&", "&")
		.replaceAll('"', """)
		.replaceAll("<", "&lt;")
		.replaceAll(">", "&gt;");
}

export function appendAgentsContext<TContent extends { type: string }>(
	content: TContent[],
	files: PersistedContextFile[],
): Array<TContent | TextContent> {
	if (!files.length) return content;
	const appendix = [
		"<subdirectory_agents_context>",
		...files.map(
			(file) =>
				`<agents_file path="${escapeXml(file.path)}">\n${escapeXml(file.content)}\n</agents_file>`,
		),
		"</subdirectory_agents_context>",
	].join("\n");
	return [...content, { type: "text", text: appendix }];
}