pi-anycopy 0.1.3 → 0.2.0

package/README.md

@@ -1,8 +1,6 @@
 # anycopy for Pi (`pi-anycopy`)
 
-Browse session tree nodes with a live preview and copy any of them to the clipboard.
-
-By comparison to Pi's native `/copy` (copies only the last assistant message) and `/md` (bulk-exports the entire branch as a Markdown transcript), `/anycopy` allows you to navigate the full session tree, preview each node's content with syntax highlighting, and copy to the clipboard any node(s) from the tree.
+This extension mirrors all the behaviors of Pi's native `/tree` while adding a live, syntax-highlighting preview of each node's content and the ability to copy any node(s) to the clipboard.
 
 <p align="center">
   <img width="450" alt="anycopy demo" src="https://raw.githubusercontent.com/w-winter/dot314/main/assets/anycopy-demo.gif" />
@@ -48,8 +46,9 @@ Defaults (customizable in `config.json`):
 
 | Key | Action |
 |-----|--------|
-| `Space` | Select/unselect focused node |
-| `Shift+C` | Copy selected nodes (or focused node if nothing is selected) |
+| `Enter` | Navigate to the focused node (same semantics as `/tree`) |
+| `Space` | Select/unselect focused node for copy |
+| `Shift+C` | Copy selected nodes, or the focused node if nothing is selected |
 | `Shift+X` | Clear selection |
 | `Shift+L` | Label node (native tree behavior) |
 | `Shift+Up` / `Shift+Down` | Scroll node preview by line |
@@ -57,18 +56,24 @@ Defaults (customizable in `config.json`):
 | `Esc` | Close |
 
 Notes:
+- `Enter` always navigates the focused node, not the marked set
+- After `Enter`, `/anycopy` offers the same summary choices as `/tree`: `No summary`, `Summarize`, and `Summarize with custom prompt`
+- If `branchSummary.skipPrompt` is `true` in Pi settings, `/anycopy` matches native `/tree` and skips the summary chooser, defaulting to no summary
+- Escaping the summary chooser reopens `/anycopy` with focus restored to the node you tried to select
+- Cancelling the custom summarization editor returns to the summary chooser
 - If no nodes are selected, `Shift+C` copies the focused node
-- When copying multiple selected nodes, they are auto-sorted chronologically (by position in the session tree), not by selection order
+- Single-node copies use just that node's content; role prefixes like `user:` or `assistant:` are only added when copying 2 or more nodes
+- When copying multiple selected nodes, they are auto-sorted chronologically by position in the session tree, not by selection order
+- Space/`Shift+C` multi-select copy behavior is unchanged by navigation support
 - Label edits are persisted via `pi.setLabel(...)`
-- Despite reoffering node labeling (`/anycopy` is arguably a better UI than `/tree` to also perform this action in), this extension doesn't offer a full reproduction of `/tree`'s other features (e.g., branch switching and summarization are not included)
 
 ## Configuration
 
 Edit `~/.pi/agent/extensions/anycopy/config.json`:
 
-- `treeFilterMode`: initial tree filter mode when opening `/anycopy` (idea sourced from [lajarre](https://github.com/lajarre)'s [pi-mono/issues/1845](https://github.com/badlogic/pi-mono/issues/1845))
+- `treeFilterMode`: initial tree filter mode when opening `/anycopy`; defaults to `default` to match `/tree`
   - one of: `default` | `no-tools` | `user-only` | `labeled-only` | `all`
-- `keys`: keybindings (see above)
+- `keys`: keybindings used inside the `/anycopy` overlay for copy/preview actions
 
 ```json
 {

package/extensions/anycopy/enter-navigation.ts

@@ -0,0 +1,101 @@
+export type AnycopyEnterNavigationResult = "reopen" | "closed";
+
+export type AnycopySummaryChoice = "No summary" | "Summarize" | "Summarize with custom prompt";
+
+export type AnycopyEnterNavigationDeps = {
+	entryId: string;
+	effectiveLeafIdForNoop: string | null;
+	skipSummaryPrompt: boolean;
+	close: () => void;
+	reopen: (options: { initialSelectedId: string }) => void;
+	navigateTree: (
+		targetId: string,
+		options?: { summarize?: boolean; customInstructions?: string },
+	) => Promise<{ cancelled: boolean; aborted?: boolean }>;
+	ui: {
+		select: (title: string, options: AnycopySummaryChoice[]) => Promise<AnycopySummaryChoice | undefined>;
+		editor: (title: string) => Promise<string | undefined>;
+		setStatus: (source: string, message: string) => void;
+		setWorkingMessage: (message?: string) => void;
+		notify: (message: string, level: "error") => void;
+	};
+};
+
+export async function runAnycopyEnterNavigation(
+	deps: AnycopyEnterNavigationDeps,
+): Promise<AnycopyEnterNavigationResult> {
+	const { entryId, effectiveLeafIdForNoop, skipSummaryPrompt, close, reopen, navigateTree, ui } = deps;
+
+	if (effectiveLeafIdForNoop !== null && entryId === effectiveLeafIdForNoop) {
+		close();
+		ui.setStatus("anycopy", "Already at this point");
+		return "closed";
+	}
+
+	close();
+
+	let wantsSummary = false;
+	let customInstructions: string | undefined;
+
+	if (!skipSummaryPrompt) {
+		while (true) {
+			const choice = await ui.select("Summarize branch?", [
+				"No summary",
+				"Summarize",
+				"Summarize with custom prompt",
+			]);
+
+			if (choice === undefined) {
+				reopen({ initialSelectedId: entryId });
+				return "reopen";
+			}
+
+			if (choice === "No summary") {
+				wantsSummary = false;
+				customInstructions = undefined;
+				break;
+			}
+
+			if (choice === "Summarize") {
+				wantsSummary = true;
+				customInstructions = undefined;
+				break;
+			}
+
+			customInstructions = await ui.editor("Custom summarization instructions");
+			if (customInstructions === undefined) {
+				continue;
+			}
+
+			wantsSummary = true;
+			break;
+		}
+	}
+
+	ui.setWorkingMessage("Navigating tree…");
+
+	try {
+		const result = await navigateTree(entryId, {
+			summarize: wantsSummary,
+			customInstructions,
+		});
+
+		if (result.cancelled) {
+			if (wantsSummary) {
+				ui.setStatus("anycopy", "Branch summarization cancelled");
+				reopen({ initialSelectedId: entryId });
+				return "reopen";
+			}
+
+			ui.setStatus("anycopy", "Navigation cancelled");
+			return "closed";
+		}
+
+		return "closed";
+	} catch (error) {
+		ui.notify(error instanceof Error ? error.message : String(error), "error");
+		return "closed";
+	} finally {
+		ui.setWorkingMessage();
+	}
+}

package/extensions/anycopy/index.ts

@@ -26,9 +26,12 @@ import { Markdown, matchesKey, truncateToWidth } from "@mariozechner/pi-tui";
 import type { Focusable } from "@mariozechner/pi-tui";
 
 import { existsSync, readFileSync } from "fs";
+import { homedir } from "os";
 import { dirname, join } from "path";
 import { fileURLToPath } from "url";
 
+import { runAnycopyEnterNavigation } from "./enter-navigation.ts";
+
 type SessionTreeNode = {
 	entry: SessionEntry;
 	children: SessionTreeNode[];
@@ -57,6 +60,12 @@ type anycopyRuntimeConfig = {
 	treeFilterMode: TreeFilterMode;
 };
 
+type BranchSummarySettingsFile = {
+	branchSummary?: {
+		skipPrompt?: boolean;
+	};
+};
+
 const DEFAULT_KEYS: anycopyKeyConfig = {
 	toggleSelect: "space",
 	copy: "shift+c",
@@ -75,10 +84,35 @@ const getExtensionDir = (): string => {
 	return dirname(fileURLToPath(import.meta.url));
 };
 
+const getAgentDir = (): string => process.env.PI_CODING_AGENT_DIR || join(homedir(), ".pi", "agent");
+
+const readJsonFile = <T>(path: string): T | undefined => {
+	if (!existsSync(path)) return undefined;
+
+	try {
+		return JSON.parse(readFileSync(path, "utf8")) as T;
+	} catch {
+		return undefined;
+	}
+};
+
+const loadBranchSummarySkipPrompt = (cwd: string): boolean => {
+	const globalSettings = readJsonFile<BranchSummarySettingsFile>(join(getAgentDir(), "settings.json"));
+	const projectSettings = readJsonFile<BranchSummarySettingsFile>(join(cwd, ".pi", "settings.json"));
+	const projectSkipPrompt = projectSettings?.branchSummary?.skipPrompt;
+	if (typeof projectSkipPrompt === "boolean") return projectSkipPrompt;
+
+	const globalSkipPrompt = globalSettings?.branchSummary?.skipPrompt;
+	return typeof globalSkipPrompt === "boolean" ? globalSkipPrompt : false;
+};
+
 const loadConfig = (): anycopyRuntimeConfig => {
 	const configPath = join(getExtensionDir(), "config.json");
 	if (!existsSync(configPath)) {
-		return { keys: { ...DEFAULT_KEYS }, treeFilterMode: DEFAULT_TREE_FILTER_MODE };
+		return {
+			keys: { ...DEFAULT_KEYS },
+			treeFilterMode: DEFAULT_TREE_FILTER_MODE,
+		};
 	}
 
 	try {
@@ -105,7 +139,10 @@ const loadConfig = (): anycopyRuntimeConfig => {
 			treeFilterMode,
 		};
 	} catch {
-		return { keys: { ...DEFAULT_KEYS }, treeFilterMode: DEFAULT_TREE_FILTER_MODE };
+		return {
+			keys: { ...DEFAULT_KEYS },
+			treeFilterMode: DEFAULT_TREE_FILTER_MODE,
+		};
 	}
 };
 
@@ -331,10 +368,14 @@ const buildNodeOrder = (roots: SessionTreeNode[]): Map<string, number> => {
 	return order;
 };
 
-/** Clipboard text: role:\n\ncontent\n\n---\n\nrole:\n\ncontent
+/** Clipboard text omits role prefix for a single node and includes it for multi-node copies
  * The preview pane is truncated for performance, while the clipboard copy is not
  */
 const buildClipboardText = (nodes: SessionTreeNode[]): string => {
+	if (nodes.length === 1) {
+		return getEntryContent(nodes[0]!.entry);
+	}
+
 	return nodes
 		.map((node) => {
 			const label = getEntryRoleLabel(node.entry);
@@ -363,6 +404,7 @@ class anycopyOverlay implements Focusable {
 		private getTree: () => SessionTreeNode[],
 		private nodeById: Map<string, SessionTreeNode>,
 		private keys: anycopyKeyConfig,
+		private onClose: () => void,
 		private getTermHeight: () => number,
 		private requestRender: () => void,
 		private theme: any,
@@ -381,6 +423,12 @@ class anycopyOverlay implements Focusable {
 	}
 
 	handleInput(data: string): void {
+		if (this.isEditingNodeLabel()) {
+			this.selector.handleInput(data);
+			this.requestRender();
+			return;
+		}
+
 		if (matchesKey(data, this.keys.toggleSelect)) {
 			this.toggleSelectedFocusedNode();
 			return;
@@ -421,6 +469,10 @@ class anycopyOverlay implements Focusable {
 		this.requestRender();
 	}
 
+	private isEditingNodeLabel(): boolean {
+		return Boolean((this.selector as { labelInput?: unknown }).labelInput);
+	}
+
 	invalidate(): void {
 		// Preview is derived from focused entry + width; invalidate forces recompute
 		this.previewCache = null;
@@ -532,10 +584,11 @@ class anycopyOverlay implements Focusable {
 
 	private renderTreeHeaderHint(width: number): string {
 		const hint =
-			`   │ ${formatKeyHint(this.keys.toggleSelect)}: select` +
+			`   │ Enter: navigate` +
+			` • ${formatKeyHint(this.keys.toggleSelect)}: select` +
 			` • ${formatKeyHint(this.keys.copy)}: copy` +
 			` • ${formatKeyHint(this.keys.clear)}: clear` +
-			" • Esc: close";
+			` • Esc: close`;
 		return truncateToWidth(this.theme.fg("dim", hint), width);
 	}
 
@@ -615,7 +668,7 @@ class anycopyOverlay implements Focusable {
 		const selectorLines = this.selector.render(width);
 		const headerHint = this.renderTreeHeaderHint(width);
 
-		// Inject shortcut hint near the tree header (above the list)
+		// Inject action hints near the tree header (above the list)
 		const insertAfter = Math.max(0, selectorLines.findIndex((l) => l.includes("Type to search")));
 		if (selectorLines.length > 0) {
 			const idx = insertAfter >= 0 ? insertAfter + 1 : 1;
@@ -652,108 +705,121 @@ export default function anycopyExtension(pi: ExtensionAPI) {
 	const keys = config.keys;
 	const treeFilterMode = config.treeFilterMode;
 
-	pi.registerCommand("anycopy", {
-		description: "Browse session tree with preview and copy any node(s) to clipboard",
-		handler: async (args, ctx: ExtensionCommandContext) => {
-			if (!ctx.hasUI) return;
+	const openAnycopy = async (
+		ctx: ExtensionCommandContext,
+		opts?: { initialSelectedId?: string },
+	) => {
+		if (!ctx.hasUI) return;
 
-			const initialTree = ctx.sessionManager.getTree() as SessionTreeNode[];
-			if (initialTree.length === 0) {
-				ctx.ui.notify("No entries in session", "warning");
-				return;
-			}
+		const initialTree = ctx.sessionManager.getTree() as SessionTreeNode[];
+		if (initialTree.length === 0) {
+			ctx.ui.notify("No entries in session", "warning");
+			return;
+		}
 
-			const getTree = () => ctx.sessionManager.getTree() as SessionTreeNode[];
-			const currentLeafId = ctx.sessionManager.getLeafId();
-
-			await ctx.ui.custom<void>((tui, theme, _kb, done) => {
-				const termRows = tui.terminal?.rows ?? 40;
-				// Pass reduced height so tree takes ~35% of terminal (it uses floor(h/2) internally)
-				const treeTermHeight = Math.floor(termRows * 0.65);
-
-				const selector = new TreeSelectorComponent(
-					initialTree,
-					currentLeafId,
-					treeTermHeight,
-					() => {
-						// Intentionally ignore Enter: closing on Enter is counterintuitive here.
-						// Use Esc to close the overlay.
-					},
-					() => done(),
-					(entryId, label) => {
-						pi.setLabel(entryId, label);
-					},
-				);
+		const getTree = () => ctx.sessionManager.getTree() as SessionTreeNode[];
+		const currentLeafId = ctx.sessionManager.getLeafId();
+		const skipSummaryPrompt = loadBranchSummarySkipPrompt(ctx.cwd);
+
+		await ctx.ui.custom<void>((tui, theme, _kb, done) => {
+			const termRows = tui.terminal?.rows ?? 40;
+			const treeTermHeight = Math.floor(termRows * 0.65);
+
+			const selector = new TreeSelectorComponent(
+				initialTree,
+				currentLeafId,
+				treeTermHeight,
+				(entryId) => {
+					void runAnycopyEnterNavigation({
+						entryId,
+						effectiveLeafIdForNoop,
+						skipSummaryPrompt,
+						close: done,
+						reopen: (reopenOpts) => {
+							void openAnycopy(ctx, reopenOpts);
+						},
+						navigateTree: async (targetId, options) => ctx.navigateTree(targetId, options),
+						ui: {
+							select: (title, options) => ctx.ui.select(title, options),
+							editor: (title) => ctx.ui.editor(title),
+							setStatus: (source, message) => ctx.ui.setStatus(source, message),
+							setWorkingMessage: (message) => ctx.ui.setWorkingMessage(message),
+							notify: (message, level) => ctx.ui.notify(message, level),
+						},
+					});
+				},
+				() => done(),
+				(entryId, label) => {
+					pi.setLabel(entryId, label);
+				},
+				opts?.initialSelectedId,
+				treeFilterMode,
+			);
+			const effectiveLeafIdForNoop = selector.getTreeList().getSelectedNode()?.entry.id ?? currentLeafId;
+
+			const nodeById = buildNodeMap(initialTree);
+			const overlay = new anycopyOverlay(
+				selector,
+				getTree,
+				nodeById,
+				keys,
+				() => done(),
+				() => tui.terminal?.rows ?? 40,
+				() => tui.requestRender(),
+				theme,
+			);
+
+			const treeList = selector.getTreeList();
+			const originalRender = treeList.render.bind(treeList);
+			treeList.render = (width: number) => {
+				const innerWidth = Math.max(10, width - 2);
+				const lines = originalRender(innerWidth);
 
-				// Build a node map once for parent traversal (toolResult → parent assistant toolCall args)
-				const nodeById = buildNodeMap(initialTree);
-
-				const overlay = new anycopyOverlay(
-					selector,
-					getTree,
-					nodeById,
-					keys,
-					() => tui.terminal?.rows ?? 40,
-					() => tui.requestRender(),
-					theme,
+				const tl = treeList as any;
+				const filteredRaw = tl.filteredNodes;
+				if (!Array.isArray(filteredRaw) || filteredRaw.length === 0) {
+					return lines.map((line: string) => "  " + line);
+				}
+				const filtered = filteredRaw as { node: SessionTreeNode }[];
+
+				const selectedIdxRaw = tl.selectedIndex;
+				const maxVisibleRaw = tl.maxVisibleLines;
+				const selectedIdx =
+					typeof selectedIdxRaw === "number" && Number.isFinite(selectedIdxRaw) ? selectedIdxRaw : 0;
+				const maxVisible =
+					typeof maxVisibleRaw === "number" && Number.isFinite(maxVisibleRaw) && maxVisibleRaw > 0
+						? maxVisibleRaw
+						: filtered.length;
+
+				const startIdx = Math.max(
+					0,
+					Math.min(selectedIdx - Math.floor(maxVisible / 2), filtered.length - maxVisible),
 				);
+				const treeRowCount = Math.max(0, lines.length - 1);
 
-				const treeList = selector.getTreeList();
+				return lines.map((line: string, i: number) => {
+					if (i >= treeRowCount) return "  " + line;
 
-				// Set initial tree filter mode (same semantics as `/tree`)
-				const rawTreeList = treeList as any;
-				if (rawTreeList && typeof rawTreeList === "object") {
-					rawTreeList.filterMode = treeFilterMode;
-					if (typeof rawTreeList.applyFilter === "function") rawTreeList.applyFilter();
-				}
+					const nodeIdx = startIdx + i;
+					const node = filtered[nodeIdx]?.node as SessionTreeNode | undefined;
+					const nodeId = node?.entry?.id;
+					if (typeof nodeId !== "string") return "  " + line;
 
-				// Monkey-patch render to inject checkbox markers (✓/○) into tree rows
-				const originalRender = treeList.render.bind(treeList);
-				treeList.render = (width: number) => {
-					const innerWidth = Math.max(10, width - 2);
-					const lines = originalRender(innerWidth);
-
-					const tl = treeList as any;
-					const filteredRaw = tl.filteredNodes;
-					if (!Array.isArray(filteredRaw) || filteredRaw.length === 0) {
-						return lines.map((line: string) => "  " + line);
-					}
-					const filtered = filteredRaw as { node: SessionTreeNode }[];
-
-					const selectedIdxRaw = tl.selectedIndex;
-					const maxVisibleRaw = tl.maxVisibleLines;
-					const selectedIdx =
-						typeof selectedIdxRaw === "number" && Number.isFinite(selectedIdxRaw) ? selectedIdxRaw : 0;
-					const maxVisible =
-						typeof maxVisibleRaw === "number" && Number.isFinite(maxVisibleRaw) && maxVisibleRaw > 0
-							? maxVisibleRaw
-							: filtered.length;
-
-					const startIdx = Math.max(
-						0,
-						Math.min(selectedIdx - Math.floor(maxVisible / 2), filtered.length - maxVisible),
-					);
-					const treeRowCount = Math.max(0, lines.length - 1);
-
-					return lines.map((line: string, i: number) => {
-						if (i >= treeRowCount) return "  " + line;
-
-						const nodeIdx = startIdx + i;
-						const node = filtered[nodeIdx]?.node as SessionTreeNode | undefined;
-						const nodeId = node?.entry?.id;
-						if (typeof nodeId !== "string") return "  " + line;
-
-						const selected = overlay.isSelectedNode(nodeId);
-						const marker = selected
-							? theme.fg("success", "\u2713 ")
-							: theme.fg("dim", "\u25CB ");
-						return marker + line;
-					});
-				};
+					const selected = overlay.isSelectedNode(nodeId);
+					const marker = selected ? theme.fg("success", "✓ ") : theme.fg("dim", "○ ");
+					return marker + line;
+				});
+			};
 
-				tui.setFocus?.(overlay);
-				return overlay;
-			});
+			tui.setFocus?.(overlay);
+			return overlay;
+		});
+	};
+
+	pi.registerCommand("anycopy", {
+		description: "Browse session tree with preview and copy any node(s) to clipboard",
+		handler: async (_args, ctx: ExtensionCommandContext) => {
+			await openAnycopy(ctx);
 		},
 	});
 }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "pi-anycopy",
-  "version": "0.1.3",
-  "description": "Copy any single message, or multiple selected messages, from the session tree, with scrollable message preview",
+  "version": "0.2.0",
+  "description": "Pi's /tree with a live syntax-highlighted preview and ability to copy any node(s) to the clipboard",
   "keywords": ["pi-package", "pi", "pi-coding-agent"],
   "license": "MIT",
   "repository": {
@@ -28,6 +28,7 @@
   "dot314Prepack": {
     "copy": [
       { "from": "../../extensions/anycopy/index.ts", "to": "extensions/anycopy/index.ts" },
+      { "from": "../../extensions/anycopy/enter-navigation.ts", "to": "extensions/anycopy/enter-navigation.ts" },
       { "from": "../../extensions/anycopy/config.json", "to": "extensions/anycopy/config.json" },
       { "from": "../../LICENSE", "to": "LICENSE" }
     ],