@ag-eco/agentplate-cli 0.13.0

package/src/tools/trellis/commands/diff.ts

@@ -0,0 +1,150 @@
+import { join } from "node:path";
+import type { Command } from "commander";
+import { c, errorOut, humanOut, icons, jsonOut } from "../output.ts";
+import { readJsonl } from "../store.ts";
+import type { Prompt, Section } from "../types.ts";
+import { ExitError } from "../types.ts";
+
+export interface SectionChange {
+	section: string;
+	type: "added" | "removed" | "modified" | "unchanged";
+}
+
+export function diffSections(fromSections: Section[], toSections: Section[]): SectionChange[] {
+	const changes: SectionChange[] = [];
+	const fromMap = new Map(fromSections.map((s) => [s.name, s.body]));
+	const toMap = new Map(toSections.map((s) => [s.name, s.body]));
+
+	// Check all names in both versions
+	const allNames = new Set([...fromMap.keys(), ...toMap.keys()]);
+
+	for (const name of allNames) {
+		const fromBody = fromMap.get(name);
+		const toBody = toMap.get(name);
+
+		if (fromBody === undefined && toBody !== undefined) {
+			changes.push({ section: name, type: "added" });
+		} else if (fromBody !== undefined && toBody === undefined) {
+			changes.push({ section: name, type: "removed" });
+		} else if (fromBody !== toBody) {
+			changes.push({ section: name, type: "modified" });
+		} else {
+			changes.push({ section: name, type: "unchanged" });
+		}
+	}
+
+	return changes;
+}
+
+export default async function diff(args: string[], json: boolean): Promise<void> {
+	const cwd = process.cwd();
+	const promptsPath = join(cwd, ".trellis", "prompts.jsonl");
+
+	if (args.includes("--help") || args.includes("-h")) {
+		humanOut(`Usage: tl diff <name> <v1> <v2> [options]
+
+Options:
+  --json    Output as JSON`);
+		return;
+	}
+
+	const positional = args.filter((a) => !a.startsWith("--"));
+	if (positional.length < 3) {
+		if (json) {
+			jsonOut({ success: false, command: "diff", error: "Usage: tl diff <name> <v1> <v2>" });
+		} else {
+			errorOut("Usage: tl diff <name> <v1> <v2>");
+		}
+		throw new ExitError(1);
+	}
+
+	const [name, v1Str, v2Str] = positional as [string, string, string];
+	const v1 = Number.parseInt(v1Str, 10);
+	const v2 = Number.parseInt(v2Str, 10);
+
+	if (Number.isNaN(v1) || Number.isNaN(v2)) {
+		errorOut("Versions must be integers");
+		throw new ExitError(1);
+	}
+
+	const allRecords = await readJsonl<Prompt>(promptsPath);
+
+	// Find by name and version
+	const from = allRecords.find((p) => p.name === name && p.version === v1);
+	const to = allRecords.find((p) => p.name === name && p.version === v2);
+
+	if (!from) {
+		if (json) {
+			jsonOut({ success: false, command: "diff", error: `Prompt '${name}@${v1}' not found` });
+		} else {
+			errorOut(`Prompt '${name}@${v1}' not found`);
+		}
+		throw new ExitError(1);
+	}
+
+	if (!to) {
+		if (json) {
+			jsonOut({ success: false, command: "diff", error: `Prompt '${name}@${v2}' not found` });
+		} else {
+			errorOut(`Prompt '${name}@${v2}' not found`);
+		}
+		throw new ExitError(1);
+	}
+
+	const changes = diffSections(from.sections, to.sections);
+	const nonTrivial = changes.filter((ch) => ch.type !== "unchanged");
+
+	if (json) {
+		jsonOut({
+			success: true,
+			command: "diff",
+			name,
+			from: v1,
+			to: v2,
+			changes: nonTrivial,
+		});
+		return;
+	}
+
+	humanOut(`${c.bold(name)}: v${v1} → v${v2}`);
+	humanOut("");
+
+	if (nonTrivial.length === 0) {
+		humanOut(c.dim("No section changes."));
+		return;
+	}
+
+	for (const change of changes) {
+		if (change.type === "unchanged") continue;
+
+		const icon =
+			change.type === "added"
+				? c.green("+")
+				: change.type === "removed"
+					? c.red("-")
+					: icons.blocked;
+		humanOut(`  ${icon} ${change.section} (${change.type})`);
+
+		if (change.type === "modified") {
+			const fromBody = from.sections.find((s) => s.name === change.section)?.body ?? "";
+			const toBody = to.sections.find((s) => s.name === change.section)?.body ?? "";
+			const fromLines = fromBody.split("\n").length;
+			const toLines = toBody.split("\n").length;
+			humanOut(c.dim(`    ${fromLines} → ${toLines} lines`));
+		}
+	}
+}
+
+export function registerDiffCommand(program: Command): void {
+	program
+		.command("diff")
+		.description("Section-aware diff between two prompt versions")
+		.argument("<name>", "Prompt name")
+		.argument("<v1>", "First version number")
+		.argument("<v2>", "Second version number")
+		.action(async (name: string, v1: string, v2: string) => {
+			const json: boolean = program.opts().json ?? false;
+			const args = [name, v1, v2, ...(json ? ["--json"] : [])];
+			await diff(args, json);
+		});
+}