pi-anycopy 0.1.2 → 0.1.4

package/README.md

@@ -42,6 +42,8 @@ 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`):
@@ -54,7 +56,7 @@ Defaults (customizable in `config.json`):
 | `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` | Close |
+| `Esc`, or configured global `shortcut` | Close |
 
 Notes:
 - If no nodes are selected, `Shift+C` copies the focused node
@@ -66,8 +68,17 @@ Notes:
 
 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))
+  - one of: `default` | `no-tools` | `user-only` | `labeled-only` | `all`
+- `keys`: keybindings used inside the `/anycopy` overlay (see above)
+
 ```json
 {
+  "shortcut": "ctrl+`",
+  "treeFilterMode": "default",
   "keys": {
     "toggleSelect": "space",
     "copy": "shift+c",

package/extensions/anycopy/config.json

@@ -1,4 +1,6 @@
 {
+  "shortcut": "ctrl+`",
+  "treeFilterMode": "no-tools",
   "keys": {
     "toggleSelect": "space",
     "copy": "shift+c",

package/extensions/anycopy/index.ts

@@ -13,7 +13,7 @@
  *   Esc       - close
  */
 
-import type { ExtensionAPI, ExtensionCommandContext, SessionEntry } from "@mariozechner/pi-coding-agent";
+import type { ExtensionAPI, ExtensionCommandContext, ExtensionContext, SessionEntry } from "@mariozechner/pi-coding-agent";
 import {
 	copyToClipboard,
 	getLanguageFromPath,
@@ -45,8 +45,18 @@ type anycopyKeyConfig = {
 	pageUp: string;
 };
 
+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;
 };
 
 const DEFAULT_KEYS: anycopyKeyConfig = {
@@ -59,31 +69,57 @@ const DEFAULT_KEYS: anycopyKeyConfig = {
 	pageUp: "shift+left",
 };
 
+const DEFAULT_TREE_FILTER_MODE: TreeFilterMode = "default";
+const DEFAULT_SHORTCUT = "ctrl+`";
+
 const getExtensionDir = (): string => {
 	// eslint-disable-next-line @typescript-eslint/no-unnecessary-condition
 	if (typeof __dirname !== "undefined") return __dirname;
 	return dirname(fileURLToPath(import.meta.url));
 };
 
-const loadConfig = (): anycopyKeyConfig => {
+const loadConfig = (): anycopyRuntimeConfig => {
 	const configPath = join(getExtensionDir(), "config.json");
-	if (!existsSync(configPath)) return { ...DEFAULT_KEYS };
+	if (!existsSync(configPath)) {
+		return {
+			shortcut: DEFAULT_SHORTCUT,
+			keys: { ...DEFAULT_KEYS },
+			treeFilterMode: DEFAULT_TREE_FILTER_MODE,
+		};
+	}
 
 	try {
 		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 =
+			typeof treeFilterModeRaw === "string" && validTreeFilterModes.includes(treeFilterModeRaw as TreeFilterMode)
+				? (treeFilterModeRaw as TreeFilterMode)
+				: DEFAULT_TREE_FILTER_MODE;
+
 		return {
-			toggleSelect: typeof keys.toggleSelect === "string" ? keys.toggleSelect : DEFAULT_KEYS.toggleSelect,
-			copy: typeof keys.copy === "string" ? keys.copy : DEFAULT_KEYS.copy,
-			clear: typeof keys.clear === "string" ? keys.clear : DEFAULT_KEYS.clear,
-			scrollDown: typeof keys.scrollDown === "string" ? keys.scrollDown : DEFAULT_KEYS.scrollDown,
-			scrollUp: typeof keys.scrollUp === "string" ? keys.scrollUp : DEFAULT_KEYS.scrollUp,
-			pageDown: typeof keys.pageDown === "string" ? keys.pageDown : DEFAULT_KEYS.pageDown,
-			pageUp: typeof keys.pageUp === "string" ? keys.pageUp : DEFAULT_KEYS.pageUp,
+			shortcut,
+			keys: {
+				toggleSelect: typeof keys.toggleSelect === "string" ? keys.toggleSelect : DEFAULT_KEYS.toggleSelect,
+				copy: typeof keys.copy === "string" ? keys.copy : DEFAULT_KEYS.copy,
+				clear: typeof keys.clear === "string" ? keys.clear : DEFAULT_KEYS.clear,
+				scrollDown: typeof keys.scrollDown === "string" ? keys.scrollDown : DEFAULT_KEYS.scrollDown,
+				scrollUp: typeof keys.scrollUp === "string" ? keys.scrollUp : DEFAULT_KEYS.scrollUp,
+				pageDown: typeof keys.pageDown === "string" ? keys.pageDown : DEFAULT_KEYS.pageDown,
+				pageUp: typeof keys.pageUp === "string" ? keys.pageUp : DEFAULT_KEYS.pageUp,
+			},
+			treeFilterMode,
 		};
 	} catch {
-		return { ...DEFAULT_KEYS };
+		return {
+			shortcut: DEFAULT_SHORTCUT,
+			keys: { ...DEFAULT_KEYS },
+			treeFilterMode: DEFAULT_TREE_FILTER_MODE,
+		};
 	}
 };
 
@@ -341,6 +377,8 @@ 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,
 		private theme: any,
@@ -359,6 +397,10 @@ class anycopyOverlay implements Focusable {
 	}
 
 	handleInput(data: string): void {
+		if (this.closeShortcut && matchesKey(data, this.closeShortcut)) {
+			this.onClose();
+			return;
+		}
 		if (matchesKey(data, this.keys.toggleSelect)) {
 			this.toggleSelectedFocusedNode();
 			return;
@@ -509,11 +551,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` +
 			` • ${formatKeyHint(this.keys.copy)}: copy` +
 			` • ${formatKeyHint(this.keys.clear)}: clear` +
-			" • Esc: close";
+			` • ${closeHint}`;
 		return truncateToWidth(this.theme.fg("dim", hint), width);
 	}
 
@@ -626,103 +669,126 @@ class anycopyOverlay implements Focusable {
 }
 
 export default function anycopyExtension(pi: ExtensionAPI) {
-	const keys = loadConfig();
+	const config = loadConfig();
+	const shortcut = config.shortcut;
+	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: ExtensionContext) => {
+		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);
+				},
+			);
+
+			// 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(),
+				theme,
+			);
+
+			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();
 			}
 
-			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);
-					},
-				);
+			// 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);
 
-				// 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();
-
-				// 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;
-					});
-				};
-
-				tui.setFocus?.(overlay);
-				return overlay;
-			});
+				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;
+				});
+			};
+
+			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);
 		},
 	});
+
+	if (shortcut) {
+		pi.registerShortcut(shortcut as any, {
+			description: "Open anycopy session tree overlay",
+			handler: async (ctx) => {
+				await openAnycopy(ctx);
+			},
+		});
+	}
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-anycopy",
-  "version": "0.1.2",
+  "version": "0.1.4",
   "description": "Copy any single message, or multiple selected messages, from the session tree, with scrollable message preview",
   "keywords": ["pi-package", "pi", "pi-coding-agent"],
   "license": "MIT",