@docyrus/docyrus 0.0.34 → 0.0.36

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

@@ -17,183 +17,183 @@ interface FileEntry {
 
 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;
-			}
+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();
+      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 });
-								}
-							}
-						}
-					}
-				}
-			}
+      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;
-			}
+      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");
-				}
-			};
+      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();
+      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)));
+        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));
+        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);
+        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),
-				);
+        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)));
+        container.addChild(new DynamicBorder((s: string) => theme.fg("accent", s)));
 
-				return {
-					render: (w) => container.render(w),
-					invalidate: () => container.invalidate(),
-					handleInput: (data) => {
+        return {
+          render: (w) => container.render(w),
+          invalidate: () => container.invalidate(),
+          handleInput: (data) => {
 						// Add paging with left/right
-						if (matchesKey(data, Key.left)) {
+            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)) {
+              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();
-					},
-				};
-			});
-		},
-	});
+              currentIndex = Math.min(items.length - 1, currentIndex + visibleRows);
+              selectList.setSelectedIndex(currentIndex);
+            } else {
+              selectList.handleInput(data);
+            }
+            tui.requestRender();
+          },
+        };
+      });
+    },
+  });
 }