pi-anycopy 0.1.4 → 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" />
@@ -42,42 +40,43 @@ Restart Pi after installation.
 /anycopy
 ```
 
-You can also open the overlay via the configurable shortcut in `config.json` without clearing the current editor draft. The default is **ctrl+`**.
-
 ## Keys
 
 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 |
 | `Shift+Left` / `Shift+Right` | Page through node preview |
-| `Esc`, or configured global `shortcut` | Close |
+| `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`:
 
-- `shortcut`: global shortcut that opens the `/anycopy` overlay while preserving whatever is currently in the editor
-  - default: **ctrl+`**
-  - set to `null` to disable it, or change it to another Pi key id such as `ctrl+a`
-- `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 used inside the `/anycopy` overlay (see above)
+- `keys`: keybindings used inside the `/anycopy` overlay for copy/preview actions
 
 ```json
 {
-  "shortcut": "ctrl+`",
   "treeFilterMode": "default",
   "keys": {
     "toggleSelect": "space",

package/extensions/anycopy/config.json

@@ -1,5 +1,4 @@
 {
-  "shortcut": "ctrl+`",
   "treeFilterMode": "no-tools",
   "keys": {
     "toggleSelect": "space",

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

@@ -13,7 +13,7 @@
  *   Esc       - close
  */
 
-import type { ExtensionAPI, ExtensionCommandContext, ExtensionContext, SessionEntry } from "@mariozechner/pi-coding-agent";
+import type { ExtensionAPI, ExtensionCommandContext, SessionEntry } from "@mariozechner/pi-coding-agent";
 import {
 	copyToClipboard,
 	getLanguageFromPath,
@@ -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[];
@@ -48,17 +51,21 @@ type anycopyKeyConfig = {
 type TreeFilterMode = "default" | "no-tools" | "user-only" | "labeled-only" | "all";
 
 type anycopyConfig = {
-	shortcut?: string | null;
 	keys?: Partial<anycopyKeyConfig>;
 	treeFilterMode?: TreeFilterMode;
 };
 
 type anycopyRuntimeConfig = {
-	shortcut: string | null;
 	keys: anycopyKeyConfig;
 	treeFilterMode: TreeFilterMode;
 };
 
+type BranchSummarySettingsFile = {
+	branchSummary?: {
+		skipPrompt?: boolean;
+	};
+};
+
 const DEFAULT_KEYS: anycopyKeyConfig = {
 	toggleSelect: "space",
 	copy: "shift+c",
@@ -70,7 +77,6 @@ const DEFAULT_KEYS: anycopyKeyConfig = {
 };
 
 const DEFAULT_TREE_FILTER_MODE: TreeFilterMode = "default";
-const DEFAULT_SHORTCUT = "ctrl+`";
 
 const getExtensionDir = (): string => {
 	// eslint-disable-next-line @typescript-eslint/no-unnecessary-condition
@@ -78,11 +84,32 @@ 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 {
-			shortcut: DEFAULT_SHORTCUT,
 			keys: { ...DEFAULT_KEYS },
 			treeFilterMode: DEFAULT_TREE_FILTER_MODE,
 		};
@@ -92,8 +119,6 @@ const loadConfig = (): anycopyRuntimeConfig => {
 		const raw = readFileSync(configPath, "utf8");
 		const parsed = JSON.parse(raw) as anycopyConfig;
 		const keys = parsed.keys ?? {};
-		const shortcut =
-			parsed.shortcut === null ? null : typeof parsed.shortcut === "string" ? parsed.shortcut : DEFAULT_SHORTCUT;
 		const treeFilterModeRaw = parsed.treeFilterMode;
 		const validTreeFilterModes: TreeFilterMode[] = ["default", "no-tools", "user-only", "labeled-only", "all"];
 		const treeFilterMode =
@@ -102,7 +127,6 @@ const loadConfig = (): anycopyRuntimeConfig => {
 				: DEFAULT_TREE_FILTER_MODE;
 
 		return {
-			shortcut,
 			keys: {
 				toggleSelect: typeof keys.toggleSelect === "string" ? keys.toggleSelect : DEFAULT_KEYS.toggleSelect,
 				copy: typeof keys.copy === "string" ? keys.copy : DEFAULT_KEYS.copy,
@@ -116,7 +140,6 @@ const loadConfig = (): anycopyRuntimeConfig => {
 		};
 	} catch {
 		return {
-			shortcut: DEFAULT_SHORTCUT,
 			keys: { ...DEFAULT_KEYS },
 			treeFilterMode: DEFAULT_TREE_FILTER_MODE,
 		};
@@ -345,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);
@@ -377,7 +404,6 @@ class anycopyOverlay implements Focusable {
 		private getTree: () => SessionTreeNode[],
 		private nodeById: Map<string, SessionTreeNode>,
 		private keys: anycopyKeyConfig,
-		private closeShortcut: string | null,
 		private onClose: () => void,
 		private getTermHeight: () => number,
 		private requestRender: () => void,
@@ -397,10 +423,12 @@ class anycopyOverlay implements Focusable {
 	}
 
 	handleInput(data: string): void {
-		if (this.closeShortcut && matchesKey(data, this.closeShortcut)) {
-			this.onClose();
+		if (this.isEditingNodeLabel()) {
+			this.selector.handleInput(data);
+			this.requestRender();
 			return;
 		}
+
 		if (matchesKey(data, this.keys.toggleSelect)) {
 			this.toggleSelectedFocusedNode();
 			return;
@@ -441,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;
@@ -551,12 +583,12 @@ class anycopyOverlay implements Focusable {
 	}
 
 	private renderTreeHeaderHint(width: number): string {
-		const closeHint = this.closeShortcut ? `${formatKeyHint(this.closeShortcut)}/Esc: close` : "Esc: close";
 		const hint =
-			`   │ ${formatKeyHint(this.keys.toggleSelect)}: select` +
+			`   │ Enter: navigate` +
+			` • ${formatKeyHint(this.keys.toggleSelect)}: select` +
 			` • ${formatKeyHint(this.keys.copy)}: copy` +
 			` • ${formatKeyHint(this.keys.clear)}: clear` +
-			` • ${closeHint}`;
+			` • Esc: close`;
 		return truncateToWidth(this.theme.fg("dim", hint), width);
 	}
 
@@ -636,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;
@@ -670,11 +702,13 @@ class anycopyOverlay implements Focusable {
 
 export default function anycopyExtension(pi: ExtensionAPI) {
 	const config = loadConfig();
-	const shortcut = config.shortcut;
 	const keys = config.keys;
 	const treeFilterMode = config.treeFilterMode;
 
-	const openAnycopy = async (ctx: ExtensionContext) => {
+	const openAnycopy = async (
+		ctx: ExtensionCommandContext,
+		opts?: { initialSelectedId?: string },
+	) => {
 		if (!ctx.hasUI) return;
 
 		const initialTree = ctx.sessionManager.getTree() as SessionTreeNode[];
@@ -685,35 +719,50 @@ export default function anycopyExtension(pi: ExtensionAPI) {
 
 		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;
-			// 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.
+				(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;
 
-			// 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,
-				shortcut,
 				() => done(),
 				() => tui.terminal?.rows ?? 40,
 				() => tui.requestRender(),
@@ -721,15 +770,6 @@ export default function anycopyExtension(pi: ExtensionAPI) {
 			);
 
 			const treeList = selector.getTreeList();
-
-			// 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();
-			}
-
-			// 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);
@@ -766,7 +806,7 @@ export default function anycopyExtension(pi: ExtensionAPI) {
 					if (typeof nodeId !== "string") return "  " + line;
 
 					const selected = overlay.isSelectedNode(nodeId);
-					const marker = selected ? theme.fg("success", "\u2713 ") : theme.fg("dim", "\u25CB ");
+					const marker = selected ? theme.fg("success", "✓ ") : theme.fg("dim", "○ ");
 					return marker + line;
 				});
 			};
@@ -782,13 +822,4 @@ export default function anycopyExtension(pi: ExtensionAPI) {
 			await openAnycopy(ctx);
 		},
 	});
-
-	if (shortcut) {
-		pi.registerShortcut(shortcut as any, {
-			description: "Open anycopy session tree overlay",
-			handler: async (ctx) => {
-				await openAnycopy(ctx);
-			},
-		});
-	}
 }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "pi-anycopy",
-  "version": "0.1.4",
-  "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" }
     ],