pi-skill/extensions/completion-report.ts

// ABOUTME: Completion Report Viewer — opens a GUI browser window showing work summary, file diffs, and rollback controls.
// ABOUTME: Gathers git diff data, renders interactive report with per-file rollback capability.

import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
import { Text } from "@mariozechner/pi-tui";
import { Type } from "@sinclair/typebox";
import { readFileSync, writeFileSync, existsSync, mkdirSync } from "node:fs";
import { join, dirname } from "node:path";
import { homedir } from "node:os";
import { execSync } from "node:child_process";
import { fileURLToPath } from "node:url";
import { createServer, type Server, type IncomingMessage, type ServerResponse } from "node:http";
import { outputLine } from "./lib/output-box.ts";
import { applyExtensionDefaults } from "./lib/themeMap.ts";
import { generateCompletionReportHTML, type ReportData, type ChangedFile } from "./lib/completion-report-html.ts";
import { createCompletionReportStandaloneExport, saveStandaloneExport } from "./lib/viewer-standalone-export.ts";
import { upsertPersistedReport } from "./lib/report-index.ts";
import { registerActiveViewer, clearActiveViewer, notifyViewerOpen } from "./lib/viewer-session.ts";

// ── Types ────────────────────────────────────────────────────────────

interface ReportResult {
	action: "done" | "rollback" | "closed";
	rolledBackFiles: string[];
}

// ── Git Helpers ──────────────────────────────────────────────────────

function execGit(cmd: string, cwd: string): string {
	try {
		return execSync(cmd, { cwd, encoding: "utf-8", maxBuffer: 10 * 1024 * 1024 }).trim();
	} catch {
		return "";
	}
}

function isGitRepo(cwd: string): boolean {
	return execGit("git rev-parse --is-inside-work-tree", cwd) === "true";
}

/**
 * Auto-detect the best base ref to diff against.
 * Priority:
 * 1. Explicit base_ref parameter
 * 2. If there are staged/unstaged changes, diff against HEAD
 * 3. HEAD~1 (last commit)
 */
function resolveBaseRef(cwd: string, explicitRef?: string): string {
	if (explicitRef) return explicitRef;

	// Check if there are uncommitted changes (staged or unstaged)
	const status = execGit("git status --porcelain", cwd);
	if (status.length > 0) {
		return "HEAD";
	}

	// Default to last commit
	return "HEAD~1";
}

/**
 * Parse `git diff --numstat` output into file stats.
 */
function parseNumstat(output: string): Array<{ path: string; additions: number; deletions: number }> {
	if (!output.trim()) return [];
	return output.split("\n").filter(Boolean).map((line) => {
		const [add, del, ...pathParts] = line.split("\t");
		const path = pathParts.join("\t"); // handle paths with tabs (renames show as old\tnew)
		return {
			path: path.replace(/.*=> /, "").replace(/[{}]/g, "").trim(),
			additions: add === "-" ? 0 : parseInt(add, 10),
			deletions: del === "-" ? 0 : parseInt(del, 10),
		};
	});
}

/**
 * Detect file status (added, modified, deleted, renamed).
 */
function getFileStatuses(cwd: string, baseRef: string): Map<string, { status: ChangedFile["status"]; oldPath?: string }> {
	const statusMap = new Map<string, { status: ChangedFile["status"]; oldPath?: string }>();

	// For uncommitted changes
	if (baseRef === "HEAD") {
		// Unstaged changes
		const unstaged = execGit("git diff --name-status", cwd);
		for (const line of unstaged.split("\n").filter(Boolean)) {
			const [status, ...parts] = line.split("\t");
			const filePath = parts[parts.length - 1];
			if (status.startsWith("R")) {
				statusMap.set(filePath, { status: "renamed", oldPath: parts[0] });
			} else if (status === "A") {
				statusMap.set(filePath, { status: "added" });
			} else if (status === "D") {
				statusMap.set(filePath, { status: "deleted" });
			} else {
				statusMap.set(filePath, { status: "modified" });
			}
		}

		// Staged changes
		const staged = execGit("git diff --cached --name-status", cwd);
		for (const line of staged.split("\n").filter(Boolean)) {
			const [status, ...parts] = line.split("\t");
			const filePath = parts[parts.length - 1];
			if (!statusMap.has(filePath)) {
				if (status.startsWith("R")) {
					statusMap.set(filePath, { status: "renamed", oldPath: parts[0] });
				} else if (status === "A") {
					statusMap.set(filePath, { status: "added" });
				} else if (status === "D") {
					statusMap.set(filePath, { status: "deleted" });
				} else {
					statusMap.set(filePath, { status: "modified" });
				}
			}
		}

		// Untracked files
		const untracked = execGit("git ls-files --others --exclude-standard", cwd);
		for (const filePath of untracked.split("\n").filter(Boolean)) {
			if (!statusMap.has(filePath)) {
				statusMap.set(filePath, { status: "added" });
			}
		}
	} else {
		// Committed changes
		const output = execGit(`git diff --name-status ${baseRef}`, cwd);
		for (const line of output.split("\n").filter(Boolean)) {
			const [status, ...parts] = line.split("\t");
			const filePath = parts[parts.length - 1];
			if (status.startsWith("R")) {
				statusMap.set(filePath, { status: "renamed", oldPath: parts[0] });
			} else if (status === "A") {
				statusMap.set(filePath, { status: "added" });
			} else if (status === "D") {
				statusMap.set(filePath, { status: "deleted" });
			} else {
				statusMap.set(filePath, { status: "modified" });
			}
		}
	}

	return statusMap;
}

/**
 * Gather all data needed for the completion report.
 */
function shouldSuppressReportFile(filePath: string): boolean {
	const normalized = filePath.replace(/\\/g, "/");
	return normalized.startsWith(".context/test-exports/") ||
		normalized.startsWith(".context/reports/") ||
		normalized === "agent/extensions/lib/marked.min.js";
}

function summarizeSuppressedFile(filePath: string): string {
	return [
		"@@ -0,0 +1,1 @@",
		`+Diff preview suppressed for generated or bulky artifact: ${filePath}`,
		"+Use copy/save/export or open the file directly if you need to inspect the full contents.",
	].join("\n");
}

function gatherReportData(cwd: string, title: string, summary: string, baseRef: string): ReportData {
	const resolvedRef = resolveBaseRef(cwd, baseRef);

	// Get diff stats
	let numstatOutput: string;
	if (resolvedRef === "HEAD") {
		// Combine staged + unstaged + untracked
		const unstaged = execGit("git diff --numstat", cwd);
		const staged = execGit("git diff --cached --numstat", cwd);
		numstatOutput = [unstaged, staged].filter(Boolean).join("\n");
	} else {
		numstatOutput = execGit(`git diff --numstat ${resolvedRef}`, cwd);
	}

	const stats = parseNumstat(numstatOutput);
	const statuses = getFileStatuses(cwd, resolvedRef);

	// Get per-file diffs
	const files: ChangedFile[] = [];

	for (const stat of stats) {
		const statusInfo = statuses.get(stat.path) || { status: "modified" as const };
		let diff: string;

		if (resolvedRef === "HEAD") {
			// Try unstaged first, then staged
			diff = execGit(`git diff -- "${stat.path}"`, cwd);
			if (!diff) {
				diff = execGit(`git diff --cached -- "${stat.path}"`, cwd);
			}
		} else {
			diff = execGit(`git diff ${resolvedRef} -- "${stat.path}"`, cwd);
		}

		files.push({
			path: stat.path,
			status: statusInfo.status,
			additions: stat.additions,
			deletions: stat.deletions,
			diff: shouldSuppressReportFile(stat.path) ? summarizeSuppressedFile(stat.path) : diff,
			oldPath: statusInfo.oldPath,
		});
	}

	// Also add untracked files if diffing against HEAD
	if (resolvedRef === "HEAD") {
		const untracked = execGit("git ls-files --others --exclude-standard", cwd);
		for (const filePath of untracked.split("\n").filter(Boolean)) {
			if (!files.some((f) => f.path === filePath)) {
				if (shouldSuppressReportFile(filePath)) {
					files.push({
						path: filePath,
						status: "added",
						additions: 1,
						deletions: 0,
						diff: summarizeSuppressedFile(filePath),
					});
					continue;
				}

				// Read file content to show as "all added"
				let content = "";
				try {
					content = readFileSync(join(cwd, filePath), "utf-8");
				} catch {
					content = "(binary or unreadable file)";
				}
				const lines = content.split("\n");
				const diff = lines.map((l) => `+${l}`).join("\n");
				files.push({
					path: filePath,
					status: "added",
					additions: lines.length,
					deletions: 0,
					diff: `@@ -0,0 +1,${lines.length} @@\n${diff}`,
				});
			}
		}
	}

	// Sort: modified first, then added, then deleted, then renamed
	const statusOrder: Record<string, number> = { modified: 0, added: 1, deleted: 2, renamed: 3 };
	files.sort((a, b) => (statusOrder[a.status] ?? 9) - (statusOrder[b.status] ?? 9));

	const totalAdditions = files.reduce((sum, f) => sum + f.additions, 0);
	const totalDeletions = files.reduce((sum, f) => sum + f.deletions, 0);

	// Read task markdown if it exists
	let taskMarkdown: string | undefined;
	const todoPath = join(cwd, ".context", "todo.md");
	if (existsSync(todoPath)) {
		try {
			taskMarkdown = readFileSync(todoPath, "utf-8");
		} catch {}
	}

	return {
		title,
		summary,
		files,
		baseRef: resolvedRef,
		totalAdditions,
		totalDeletions,
		taskMarkdown,
	};
}

// ── HTTP Server ──────────────────────────────────────────────────────

function startReportServer(
	report: ReportData,
	cwd: string,
): Promise<{ port: number; server: Server; waitForResult: () => Promise<ReportResult> }> {
	return new Promise((resolveSetup) => {
		let resolveResult: (result: ReportResult) => void;
		let settled = false;
		const settle = (result: ReportResult) => {
			if (settled) return;
			settled = true;
			resolveResult!(result);
		};
		const resultPromise = new Promise<ReportResult>((res) => {
			resolveResult = res;
		});

		const server = createServer((req: IncomingMessage, res: ServerResponse) => {
			res.setHeader("Access-Control-Allow-Origin", "*");
			res.setHeader("Access-Control-Allow-Methods", "GET, POST, OPTIONS");
			res.setHeader("Access-Control-Allow-Headers", "Content-Type");

			if (req.method === "OPTIONS") {
				res.writeHead(204);
				res.end();
				return;
			}

			const url = new URL(req.url || "/", `http://localhost`);

			// Serve the main HTML page
			if (req.method === "GET" && url.pathname === "/") {
				const port = (server.address() as any)?.port || 0;
				const html = generateCompletionReportHTML({ report, port });
				res.writeHead(200, { "Content-Type": "text/html; charset=utf-8" });
				res.end(html);
				return;
			}

			// Serve the logo image
			if (req.method === "GET" && url.pathname === "/logo.png") {
				try {
					const logoPath = join(dirname(fileURLToPath(import.meta.url)), "assets", "agent-logo.png");
					const logoData = readFileSync(logoPath);
					res.writeHead(200, { "Content-Type": "image/png", "Cache-Control": "public, max-age=3600" });
					res.end(logoData);
				} catch {
					res.writeHead(404);
					res.end();
				}
				return;
			}

			// Handle rollback
			if (req.method === "POST" && url.pathname === "/rollback") {
				let body = "";
				req.on("data", (chunk) => { body += chunk; });
				req.on("end", () => {
					try {
						const data = JSON.parse(body);
						const files: string[] = data.files || [];
						const baseRef: string = data.baseRef || "HEAD";
						const errors: string[] = [];

						for (const filePath of files) {
							try {
								if (baseRef === "HEAD") {
									// For uncommitted changes, checkout from HEAD
									execSync(`git checkout HEAD -- "${filePath}"`, { cwd, encoding: "utf-8" });
								} else {
									// For committed changes, checkout from the base ref
									execSync(`git checkout ${baseRef} -- "${filePath}"`, { cwd, encoding: "utf-8" });
								}
							} catch (err: any) {
								errors.push(`${filePath}: ${err.message}`);
							}
						}

						if (errors.length > 0) {
							res.writeHead(200, { "Content-Type": "application/json" });
							res.end(JSON.stringify({ ok: false, error: errors.join("; ") }));
						} else {
							res.writeHead(200, { "Content-Type": "application/json" });
							res.end(JSON.stringify({ ok: true }));
						}
					} catch {
						res.writeHead(400, { "Content-Type": "application/json" });
						res.end(JSON.stringify({ error: "Invalid JSON" }));
					}
				});
				return;
			}

			// Handle result (done)
			if (req.method === "POST" && url.pathname === "/result") {
				let body = "";
				req.on("data", (chunk) => { body += chunk; });
				req.on("end", () => {
					try {
						const data = JSON.parse(body);
						res.writeHead(200, { "Content-Type": "application/json" });
						res.end(JSON.stringify({ ok: true }));
						settle({
							action: data.action || "done",
							rolledBackFiles: data.rolledBackFiles || [],
						});
					} catch {
						res.writeHead(400, { "Content-Type": "application/json" });
						res.end(JSON.stringify({ error: "Invalid JSON" }));
					}
				});
				return;
			}

			// Handle save to desktop
			if (req.method === "POST" && url.pathname === "/save") {
				let body = "";
				req.on("data", (chunk) => { body += chunk; });
				req.on("end", () => {
					try {
						const data = JSON.parse(body);
						const desktop = join(homedir(), "Desktop");
						if (!existsSync(desktop)) mkdirSync(desktop, { recursive: true });
						const ts = new Date().toISOString().replace(/[:.]/g, "-").slice(0, 19);
						const fileName = `report-${ts}.md`;
						const filePath = join(desktop, fileName);
						writeFileSync(filePath, data.content, "utf-8");
						res.writeHead(200, { "Content-Type": "application/json" });
						res.end(JSON.stringify({ ok: true, message: `Saved to ~/Desktop/${fileName}` }));
					} catch (err: any) {
						res.writeHead(500, { "Content-Type": "application/json" });
						res.end(JSON.stringify({ error: err.message }));
					}
				});
				return;
			}

			if (req.method === "POST" && url.pathname === "/export-standalone") {
				try {
					const html = createCompletionReportStandaloneExport(report);
					const saved = saveStandaloneExport({ filePrefix: "report-readonly", html });
					res.writeHead(200, { "Content-Type": "application/json" });
					res.end(JSON.stringify({ ok: true, message: `Standalone export saved to ~/Desktop/${saved.fileName}` }));
				} catch (err: any) {
					res.writeHead(500, { "Content-Type": "application/json" });
					res.end(JSON.stringify({ error: err.message }));
				}
				return;
			}

			// 404
			res.writeHead(404);
			res.end("Not found");
		});

		server.on("close", () => {
			settle({ action: "closed", rolledBackFiles: [] });
		});

		server.listen(0, "127.0.0.1", () => {
			const addr = server.address() as any;
			resolveSetup({
				port: addr.port,
				server,
				waitForResult: () => resultPromise,
			});
		});
	});
}

function openBrowser(url: string): void {
	try {
		execSync(`open "${url}"`, { stdio: "ignore" });
	} catch {
		try {
			execSync(`xdg-open "${url}"`, { stdio: "ignore" });
		} catch {
			try {
				execSync(`start "${url}"`, { stdio: "ignore" });
			} catch {}
		}
	}
}

// ── Tool Parameters ──────────────────────────────────────────────────

const ShowReportParams = Type.Object({
	title: Type.Optional(Type.String({ description: "Title for the report (default: 'Completion Report')" })),
	summary: Type.Optional(Type.String({ description: "Markdown summary of the work done" })),
	base_ref: Type.Optional(Type.String({ description: "Git ref to diff against (default: auto-detect — HEAD for uncommitted changes, HEAD~1 for committed)" })),
});

// ── Extension ────────────────────────────────────────────────────────

export default function (pi: ExtensionAPI) {
	let activeServer: Server | null = null;
	let activeSession: { kind: "report"; title: string; url: string; server: Server; onClose: () => void } | null = null;

	function cleanupServer() {
		const server = activeServer;
		activeServer = null;
		if (server) {
			try { server.close(); } catch {}
		}
		if (activeSession) {
			clearActiveViewer(activeSession);
			activeSession = null;
		}
	}

	// ── show_report tool ─────────────────────────────────────────────

	pi.registerTool({
		name: "show_report",
		label: "Show Report",
		description:
			"Open a completion report viewer in the browser. Shows a summary of work done, " +
			"files changed with unified diffs, and per-file rollback controls.\n\n" +
			"Automatically gathers git diff data from the working directory. " +
			"Includes task completion data from .context/todo.md if available.\n\n" +
			"The user can review diffs, rollback individual files or all changes, " +
			"copy the report, or save it to the desktop.",
		parameters: ShowReportParams,

		async execute(_toolCallId, params, _signal, _onUpdate, ctx) {
			const {
				title = "Completion Report",
				summary = "",
				base_ref,
			} = params as { title?: string; summary?: string; base_ref?: string };

			const cwd = ctx.cwd || process.cwd();

			// Check if we're in a git repo
			if (!isGitRepo(cwd)) {
				return {
					content: [{ type: "text" as const, text: "Error: Not a git repository. The completion report requires git to gather file changes." }],
				};
			}

			// Gather report data
			const report = gatherReportData(cwd, title, summary, base_ref || "");

			if (report.files.length === 0) {
				return {
					content: [{ type: "text" as const, text: "No file changes detected. Nothing to report." }],
				};
			}

			// Clean up any previous server
			cleanupServer();

			// Start server and open browser
			const { port, server, waitForResult } = await startReportServer(report, cwd);
			activeServer = server;

			const url = `http://127.0.0.1:${port}`;
			activeSession = {
				kind: "report",
				title,
				url,
				server,
				onClose: () => {
					activeServer = null;
					activeSession = null;
				},
			};
			registerActiveViewer(activeSession);
			openBrowser(url);
			notifyViewerOpen(ctx, activeSession);

			// Wait for user to close the report
			try {
				const result = await waitForResult();

				try {
					upsertPersistedReport({
						category: "completion",
						title,
						summary,
						sourcePath: join(cwd, ".context", "todo.md"),
						viewerPath: join(cwd, ".context", "todo.md"),
						viewerLabel: title,
						tags: ["completion", "git", "diff"],
						metadata: {
							baseRef: report.baseRef,
							fileCount: report.files.length,
							totalAdditions: report.totalAdditions,
							totalDeletions: report.totalDeletions,
							action: result.action,
							rolledBackFiles: result.rolledBackFiles,
						},
					});
				} catch {}

				const rolledBack = result.rolledBackFiles.length;
				const summary = rolledBack > 0
					? `Report closed. ${rolledBack} file${rolledBack > 1 ? "s" : ""} rolled back: ${result.rolledBackFiles.join(", ")}`
					: "Report closed. No files were rolled back.";

				return {
					content: [{ type: "text" as const, text: summary }],
					details: {
						action: result.action,
						rolledBackFiles: result.rolledBackFiles,
						totalFiles: report.files.length,
						totalAdditions: report.totalAdditions,
						totalDeletions: report.totalDeletions,
					},
				};
			} finally {
				cleanupServer();
			}
		},

		renderCall(args, theme) {
			const titleArg = (args as any).title || "Completion Report";
			const text =
				theme.fg("toolTitle", theme.bold("show_report ")) +
				theme.fg("success", titleArg);
			return new Text(outputLine(theme, "success", text), 0, 0);
		},

		renderResult(result, _options, theme) {
			const details = ((result as any).details || result) as any;
			if (!details || (details.totalFiles === undefined && !details.content)) {
				const text = result.content[0];
				return new Text(text?.type === "text" ? text.text : "", 0, 0);
			}

			const fileCount = details.totalFiles ?? 0;
			const totalAdditions = details.totalAdditions ?? 0;
			const totalDeletions = details.totalDeletions ?? 0;
			const rolledBack = (details.rolledBackFiles || []).length;

			let info = `${fileCount} files · +${totalAdditions} -${totalDeletions}`;
			if (rolledBack > 0) {
				info += ` · ${rolledBack} rolled back`;
				return new Text(
					outputLine(theme, "warning", `Report closed — ${info}`),
					0, 0,
				);
			}

			return new Text(
				outputLine(theme, "success", `Report closed — ${info}`),
				0, 0,
			);
		},
	});

	// ── /report command ──────────────────────────────────────────────

	pi.registerCommand("report", {
		description: "Open the completion report viewer for current git changes",
		handler: async (args, ctx) => {
			if (!ctx.hasUI) {
				ctx.ui.notify("/report requires interactive mode", "error");
				return;
			}

			const cwd = ctx.cwd || process.cwd();

			if (!isGitRepo(cwd)) {
				ctx.ui.notify("Not a git repository", "error");
				return;
			}

			// Parse optional base ref from args
			const baseRef = args.trim() || "";
			const report = gatherReportData(cwd, "Completion Report", "", baseRef);

			if (report.files.length === 0) {
				ctx.ui.notify("No file changes detected", "info");
				return;
			}

			cleanupServer();

			const { port, server, waitForResult } = await startReportServer(report, cwd);
			activeServer = server;

			const url = `http://127.0.0.1:${port}`;
			activeSession = {
				kind: "report",
				title: "Completion Report",
				url,
				server,
				onClose: () => {
					activeServer = null;
					activeSession = null;
				},
			};
			registerActiveViewer(activeSession);
			openBrowser(url);
			notifyViewerOpen(ctx, activeSession);

			const result = await waitForResult();
			cleanupServer();

			if (result.rolledBackFiles.length > 0) {
				ctx.ui.notify(
					`Report closed — ${result.rolledBackFiles.length} file(s) rolled back`,
					"info",
				);
			} else {
				ctx.ui.notify("Report closed", "info");
			}
		},
	});

	// ── Session lifecycle ────────────────────────────────────────────

	pi.on("session_start", async (_event, ctx) => {
		applyExtensionDefaults(import.meta.url, ctx);
	});

	pi.on("session_shutdown", async () => {
		cleanupServer();
	});
}