@docyrus/docyrus 0.0.20 → 0.0.21

package/resources/pi-agent/extensions/diff.ts

@@ -0,0 +1,218 @@
+/**
+ * Diff Extension
+ *
+ * /diff command shows modified/deleted/new files from git status and opens
+ * the selected file in VS Code's diff view.
+ */
+
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { DynamicBorder } from "@mariozechner/pi-coding-agent";
+import { Container, Key, matchesKey, type SelectItem, SelectList, Text } from "@mariozechner/pi-tui";
+
+interface FileInfo {
+	status: string;
+	statusLabel: string;
+	file: string;
+}
+
+export default function (pi: ExtensionAPI) {
+	pi.registerCommand("diff", {
+		description: "Show git changes and open in VS Code diff view",
+		handler: async (_args, ctx) => {
+			if (!ctx.hasUI) {
+				ctx.ui.notify("No UI available", "error");
+				return;
+			}
+
+			// Get changed files from git status
+			const result = await pi.exec("git", ["status", "--porcelain"], { cwd: ctx.cwd });
+
+			if (result.code !== 0) {
+				ctx.ui.notify(`git status failed: ${result.stderr}`, "error");
+				return;
+			}
+
+			if (!result.stdout || !result.stdout.trim()) {
+				ctx.ui.notify("No changes in working tree", "info");
+				return;
+			}
+
+			// Parse git status output
+			// Format: XY filename (where XY is two-letter status, then space, then filename)
+			const lines = result.stdout.split("\n");
+			const files: FileInfo[] = [];
+
+			for (const line of lines) {
+				if (line.length < 4) continue; // Need at least "XY f"
+
+				const status = line.slice(0, 2);
+				const file = line.slice(2).trimStart();
+
+				// Translate status codes to short labels
+				let statusLabel: string;
+				if (status.includes("M")) statusLabel = "M";
+				else if (status.includes("A")) statusLabel = "A";
+				else if (status.includes("D")) statusLabel = "D";
+				else if (status.includes("?")) statusLabel = "?";
+				else if (status.includes("R")) statusLabel = "R";
+				else if (status.includes("C")) statusLabel = "C";
+				else statusLabel = status.trim() || "~";
+
+				files.push({ status: statusLabel, statusLabel, file });
+			}
+
+			if (files.length === 0) {
+				ctx.ui.notify("No changes found", "info");
+				return;
+			}
+
+			const WINDOWS_UNSAFE_CMD_CHARS_RE = /[&|<>^%\r\n]/;
+			const quoteCmdArg = (value: string) => `"${value.replace(/"/g, '""')}"`;
+
+			const openWithCode = async (file: string) => {
+				if (process.platform === "win32") {
+					if (WINDOWS_UNSAFE_CMD_CHARS_RE.test(file)) {
+						ctx.ui.notify(
+							`Refusing to open ${file}: path contains Windows cmd metacharacters (& | < > ^ % or newline).`,
+							"error",
+						);
+						return null;
+					}
+					const commandLine = `code -g ${quoteCmdArg(file)}`;
+					return pi.exec("cmd", ["/d", "/s", "/c", commandLine], { cwd: ctx.cwd });
+				}
+				return pi.exec("code", ["-g", file], { cwd: ctx.cwd });
+			};
+
+			const openSelected = async (fileInfo: FileInfo): Promise<void> => {
+				try {
+					// Open in VS Code diff view.
+					// For untracked files, git difftool won't work, so fall back to just opening the file.
+					if (fileInfo.status === "?") {
+						const openResult = await openWithCode(fileInfo.file);
+						if (!openResult) return;
+						if (openResult.code !== 0) {
+							const openStderr = openResult.stderr.trim();
+							ctx.ui.notify(
+								`Failed to open ${fileInfo.file} (exit ${openResult.code})${openStderr ? `: ${openStderr}` : ""}`,
+								"error",
+							);
+						}
+						return;
+					}
+
+					const diffResult = await pi.exec("git", ["difftool", "-y", "--tool=vscode", fileInfo.file], {
+						cwd: ctx.cwd,
+					});
+					if (diffResult.code !== 0) {
+						const diffStderr = diffResult.stderr.trim();
+						ctx.ui.notify(
+							`Failed to show diff with vscode for ${fileInfo.file} (exit ${diffResult.code})${diffStderr ? `: ${diffStderr}` : ""}`,
+							"error",
+						);
+						ctx.ui.notify(
+							"Troubleshooting: check git difftool config (e.g. `git config --get difftool.vscode.cmd`).",
+							"info",
+						);
+
+						const openResult = await openWithCode(fileInfo.file);
+						if (!openResult) return;
+						if (openResult.code !== 0) {
+							const openStderr = openResult.stderr.trim();
+							ctx.ui.notify(
+								`Failed to open ${fileInfo.file} (exit ${openResult.code})${openStderr ? `: ${openStderr}` : ""}`,
+								"error",
+							);
+						}
+					}
+				} catch (error) {
+					const message = error instanceof Error ? error.message : String(error);
+					ctx.ui.notify(`Failed to open ${fileInfo.file}: ${message}`, "error");
+				}
+			};
+
+			// Show file picker with SelectList
+			await ctx.ui.custom<void>((tui, theme, _kb, done) => {
+				const container = new Container();
+
+				// Top border
+				container.addChild(new DynamicBorder((s: string) => theme.fg("accent", s)));
+
+				// Title
+				container.addChild(new Text(theme.fg("accent", theme.bold(" Select file to diff")), 0, 0));
+
+				// Build select items with colored status
+				const items: SelectItem[] = files.map((f) => {
+					let statusColor: string;
+					switch (f.status) {
+						case "M":
+							statusColor = theme.fg("warning", f.status);
+							break;
+						case "A":
+							statusColor = theme.fg("success", f.status);
+							break;
+						case "D":
+							statusColor = theme.fg("error", f.status);
+							break;
+						case "?":
+							statusColor = theme.fg("muted", f.status);
+							break;
+						default:
+							statusColor = theme.fg("dim", f.status);
+					}
+					return {
+						value: f,
+						label: `${statusColor} ${f.file}`,
+					};
+				});
+
+				const visibleRows = Math.min(files.length, 15);
+				let currentIndex = 0;
+
+				const selectList = new SelectList(items, visibleRows, {
+					selectedPrefix: (t) => theme.fg("accent", t),
+					selectedText: (t) => t, // Keep existing colors
+					description: (t) => theme.fg("muted", t),
+					scrollInfo: (t) => theme.fg("dim", t),
+					noMatch: (t) => theme.fg("warning", t),
+				});
+				selectList.onSelect = (item) => {
+					void openSelected(item.value as FileInfo);
+				};
+				selectList.onCancel = () => done();
+				selectList.onSelectionChange = (item) => {
+					currentIndex = items.indexOf(item);
+				};
+				container.addChild(selectList);
+
+				// Help text
+				container.addChild(
+					new Text(theme.fg("dim", " ↑↓ navigate • ←→ page • enter open • esc close"), 0, 0),
+				);
+
+				// Bottom border
+				container.addChild(new DynamicBorder((s: string) => theme.fg("accent", s)));
+
+				return {
+					render: (w) => container.render(w),
+					invalidate: () => container.invalidate(),
+					handleInput: (data) => {
+						// Add paging with left/right
+						if (matchesKey(data, Key.left)) {
+							// Page up - clamp to 0
+							currentIndex = Math.max(0, currentIndex - visibleRows);
+							selectList.setSelectedIndex(currentIndex);
+						} else if (matchesKey(data, Key.right)) {
+							// Page down - clamp to last
+							currentIndex = Math.min(items.length - 1, currentIndex + visibleRows);
+							selectList.setSelectedIndex(currentIndex);
+						} else {
+							selectList.handleInput(data);
+						}
+						tui.requestRender();
+					},
+				};
+			});
+		},
+	});
+}

package/resources/pi-agent/extensions/files.ts

@@ -0,0 +1,199 @@
+/**
+ * Files Extension
+ *
+ * /files command lists all files the model has read/written/edited in the active session branch,
+ * coalesced by path and sorted newest first. Selecting a file opens it in VS Code.
+ */
+
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { DynamicBorder } from "@mariozechner/pi-coding-agent";
+import { Container, Key, matchesKey, type SelectItem, SelectList, Text } from "@mariozechner/pi-tui";
+
+interface FileEntry {
+	path: string;
+	operations: Set<"read" | "write" | "edit">;
+	lastTimestamp: number;
+}
+
+type FileToolName = "read" | "write" | "edit";
+
+export default function (pi: ExtensionAPI) {
+	pi.registerCommand("files", {
+		description: "Show files read/written/edited in this session",
+		handler: async (_args, ctx) => {
+			if (!ctx.hasUI) {
+				ctx.ui.notify("No UI available", "error");
+				return;
+			}
+
+			// Get the current branch (path from leaf to root)
+			const branch = ctx.sessionManager.getBranch();
+
+			// First pass: collect tool calls (id -> {path, name}) from assistant messages
+			const toolCalls = new Map<string, { path: string; name: FileToolName; timestamp: number }>();
+
+			for (const entry of branch) {
+				if (entry.type !== "message") continue;
+				const msg = entry.message;
+
+				if (msg.role === "assistant" && Array.isArray(msg.content)) {
+					for (const block of msg.content) {
+						if (block.type === "toolCall") {
+							const name = block.name;
+							if (name === "read" || name === "write" || name === "edit") {
+								const path = block.arguments?.path;
+								if (path && typeof path === "string") {
+									toolCalls.set(block.id, { path, name, timestamp: msg.timestamp });
+								}
+							}
+						}
+					}
+				}
+			}
+
+			// Second pass: match tool results to get the actual execution timestamp
+			const fileMap = new Map<string, FileEntry>();
+
+			for (const entry of branch) {
+				if (entry.type !== "message") continue;
+				const msg = entry.message;
+
+				if (msg.role === "toolResult") {
+					const toolCall = toolCalls.get(msg.toolCallId);
+					if (!toolCall) continue;
+
+					const { path, name } = toolCall;
+					const timestamp = msg.timestamp;
+
+					const existing = fileMap.get(path);
+					if (existing) {
+						existing.operations.add(name);
+						if (timestamp > existing.lastTimestamp) {
+							existing.lastTimestamp = timestamp;
+						}
+					} else {
+						fileMap.set(path, {
+							path,
+							operations: new Set([name]),
+							lastTimestamp: timestamp,
+						});
+					}
+				}
+			}
+
+			if (fileMap.size === 0) {
+				ctx.ui.notify("No files read/written/edited in this session", "info");
+				return;
+			}
+
+			// Sort by most recent first
+			const files = Array.from(fileMap.values()).sort((a, b) => b.lastTimestamp - a.lastTimestamp);
+
+			const WINDOWS_UNSAFE_CMD_CHARS_RE = /[&|<>^%\r\n]/;
+			const quoteCmdArg = (value: string) => `"${value.replace(/"/g, '""')}"`;
+
+			const openWithCode = async (path: string) => {
+				if (process.platform === "win32") {
+					if (WINDOWS_UNSAFE_CMD_CHARS_RE.test(path)) {
+						ctx.ui.notify(
+							`Refusing to open ${path}: path contains Windows cmd metacharacters (& | < > ^ % or newline).`,
+							"error",
+						);
+						return null;
+					}
+					const commandLine = `code -g ${quoteCmdArg(path)}`;
+					return pi.exec("cmd", ["/d", "/s", "/c", commandLine], { cwd: ctx.cwd });
+				}
+				return pi.exec("code", ["-g", path], { cwd: ctx.cwd });
+			};
+
+			const openSelected = async (file: FileEntry): Promise<void> => {
+				try {
+					const openResult = await openWithCode(file.path);
+					if (!openResult) return;
+					if (openResult.code !== 0) {
+						const openStderr = openResult.stderr.trim();
+						ctx.ui.notify(
+							`Failed to open ${file.path} (exit ${openResult.code})${openStderr ? `: ${openStderr}` : ""}`,
+							"error",
+						);
+					}
+				} catch (error) {
+					const message = error instanceof Error ? error.message : String(error);
+					ctx.ui.notify(`Failed to open ${file.path}: ${message}`, "error");
+				}
+			};
+
+			// Show file picker with SelectList
+			await ctx.ui.custom<void>((tui, theme, _kb, done) => {
+				const container = new Container();
+
+				// Top border
+				container.addChild(new DynamicBorder((s: string) => theme.fg("accent", s)));
+
+				// Title
+				container.addChild(new Text(theme.fg("accent", theme.bold(" Select file to open")), 0, 0));
+
+				// Build select items with colored operations
+				const items: SelectItem[] = files.map((f) => {
+					const ops: string[] = [];
+					if (f.operations.has("read")) ops.push(theme.fg("muted", "R"));
+					if (f.operations.has("write")) ops.push(theme.fg("success", "W"));
+					if (f.operations.has("edit")) ops.push(theme.fg("warning", "E"));
+					const opsLabel = ops.join("");
+					return {
+						value: f,
+						label: `${opsLabel} ${f.path}`,
+					};
+				});
+
+				const visibleRows = Math.min(files.length, 15);
+				let currentIndex = 0;
+
+				const selectList = new SelectList(items, visibleRows, {
+					selectedPrefix: (t) => theme.fg("accent", t),
+					selectedText: (t) => t, // Keep existing colors
+					description: (t) => theme.fg("muted", t),
+					scrollInfo: (t) => theme.fg("dim", t),
+					noMatch: (t) => theme.fg("warning", t),
+				});
+				selectList.onSelect = (item) => {
+					void openSelected(item.value as FileEntry);
+				};
+				selectList.onCancel = () => done();
+				selectList.onSelectionChange = (item) => {
+					currentIndex = items.indexOf(item);
+				};
+				container.addChild(selectList);
+
+				// Help text
+				container.addChild(
+					new Text(theme.fg("dim", " ↑↓ navigate • ←→ page • enter open • esc close"), 0, 0),
+				);
+
+				// Bottom border
+				container.addChild(new DynamicBorder((s: string) => theme.fg("accent", s)));
+
+				return {
+					render: (w) => container.render(w),
+					invalidate: () => container.invalidate(),
+					handleInput: (data) => {
+						// Add paging with left/right
+						if (matchesKey(data, Key.left)) {
+							// Page up - clamp to 0
+							currentIndex = Math.max(0, currentIndex - visibleRows);
+							selectList.setSelectedIndex(currentIndex);
+						} else if (matchesKey(data, Key.right)) {
+							// Page down - clamp to last
+							currentIndex = Math.min(items.length - 1, currentIndex + visibleRows);
+							selectList.setSelectedIndex(currentIndex);
+						} else {
+							selectList.handleInput(data);
+						}
+						tui.requestRender();
+					},
+				};
+			});
+		},
+	});
+}