pi-anycopy 0.1.4 → 0.2.1

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, the ability to copy any node(s) to the clipboard, and optional display of the timestamps of labeled nodes' last labelings.
 
 <p align="center">
   <img width="450" alt="anycopy demo" src="https://raw.githubusercontent.com/w-winter/dot314/main/assets/anycopy-demo.gif" />
@@ -42,47 +40,53 @@ 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+T` | Toggle label timestamps for labeled nodes |
 | `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
+- `Shift+T` is configurable via `keys.toggleLabelTimestamps` in `config.json`
+- `Shift+T` shows timestamps for labeled nodes only, using the latest label-change time for each label
+- Same-day labels show `HH:MM`; older labels show `M/D HH:MM`; cross-year labels show `YY/M/D HH:MM`
 - 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, including the label timestamp toggle
 
 ```json
 {
-  "shortcut": "ctrl+`",
   "treeFilterMode": "default",
   "keys": {
     "toggleSelect": "space",
     "copy": "shift+c",
     "clear": "shift+x",
+    "toggleLabelTimestamps": "shift+t",
     "scrollUp": "shift+up",
     "scrollDown": "shift+down",
     "pageUp": "shift+left",

package/extensions/anycopy/config.json

@@ -1,10 +1,10 @@
 {
-  "shortcut": "ctrl+`",
   "treeFilterMode": "no-tools",
   "keys": {
     "toggleSelect": "space",
     "copy": "shift+c",
     "clear": "shift+x",
+    "toggleLabelTimestamps": "shift+t",
     "scrollUp": "shift+up",
     "scrollDown": "shift+down",
     "pageUp": "shift+left",

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

@@ -7,13 +7,14 @@
  *   Space     - select/unselect focused node for copy
  *   Shift+C   - copy selected nodes (or focused node if none selected)
  *   Shift+X   - clear selection
- *   Shift+L   - label node (native tree behavior)
+ *   Shift+L   - label node
+ *   Shift+T   - toggle label timestamps for labeled nodes
  *   Shift+↑/↓ - scroll preview
  *   Shift+←/→ - page preview
  *   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,19 +27,24 @@ 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[];
 	label?: string;
+	labelTimestamp?: string;
 };
 
 type anycopyKeyConfig = {
 	toggleSelect: string;
 	copy: string;
 	clear: string;
+	toggleLabelTimestamps: string;
 	scrollDown: string;
 	scrollUp: string;
 	pageDown: string;
@@ -48,21 +54,26 @@ 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",
 	clear: "shift+x",
+	toggleLabelTimestamps: "shift+t",
 	scrollDown: "shift+down",
 	scrollUp: "shift+up",
 	pageDown: "shift+right",
@@ -70,7 +81,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 +88,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 +123,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,11 +131,14 @@ 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,
 				clear: typeof keys.clear === "string" ? keys.clear : DEFAULT_KEYS.clear,
+				toggleLabelTimestamps:
+					typeof keys.toggleLabelTimestamps === "string"
+						? keys.toggleLabelTimestamps
+						: DEFAULT_KEYS.toggleLabelTimestamps,
 				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,
@@ -116,7 +148,6 @@ const loadConfig = (): anycopyRuntimeConfig => {
 		};
 	} catch {
 		return {
-			shortcut: DEFAULT_SHORTCUT,
 			keys: { ...DEFAULT_KEYS },
 			treeFilterMode: DEFAULT_TREE_FILTER_MODE,
 		};
@@ -345,10 +376,74 @@ const buildNodeOrder = (roots: SessionTreeNode[]): Map<string, number> => {
 	return order;
 };
 
-/** Clipboard text: role:\n\ncontent\n\n---\n\nrole:\n\ncontent
+const buildLatestLabelTimestamps = (entries: SessionEntry[]): Map<string, string> => {
+	const timestampsById = new Map<string, string>();
+	for (const entry of entries) {
+		if (entry.type !== "label") continue;
+		if (entry.label) {
+			timestampsById.set(entry.targetId, entry.timestamp);
+		} else {
+			timestampsById.delete(entry.targetId);
+		}
+	}
+	return timestampsById;
+};
+
+const applyLabelTimestampsToTree = (roots: SessionTreeNode[], timestampsById: Map<string, string>): void => {
+	const stack = [...roots];
+	while (stack.length > 0) {
+		const node = stack.pop()!;
+		node.labelTimestamp = timestampsById.get(node.entry.id);
+		for (const child of node.children) stack.push(child);
+	}
+};
+
+const formatLabelTimestamp = (timestamp: string): string => {
+	const date = new Date(timestamp);
+	if (Number.isNaN(date.getTime())) return timestamp;
+
+	const now = new Date();
+	const hours = date.getHours().toString().padStart(2, "0");
+	const minutes = date.getMinutes().toString().padStart(2, "0");
+	const time = `${hours}:${minutes}`;
+
+	if (
+		date.getFullYear() === now.getFullYear() &&
+		date.getMonth() === now.getMonth() &&
+		date.getDate() === now.getDate()
+	) {
+		return time;
+	}
+
+	const month = date.getMonth() + 1;
+	const day = date.getDate();
+	if (date.getFullYear() === now.getFullYear()) {
+		return `${month}/${day} ${time}`;
+	}
+
+	const year = date.getFullYear().toString().slice(-2);
+	return `${year}/${month}/${day} ${time}`;
+};
+
+const insertLabelTimestampIntoLine = (line: string, node: SessionTreeNode, theme: any): string => {
+	if (!node.label || !node.labelTimestamp) return line;
+
+	const labelToken = `[${node.label}] `;
+	const labelIndex = line.indexOf(labelToken);
+	if (labelIndex === -1) return line;
+
+	const insertAt = labelIndex + labelToken.length;
+	return `${line.slice(0, insertAt)}${theme.fg("muted", `${formatLabelTimestamp(node.labelTimestamp)} `)}${line.slice(insertAt)}`;
+};
+
+/** 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);
@@ -365,6 +460,7 @@ class anycopyOverlay implements Focusable {
 	private _focused = false;
 	private previewScrollOffset = 0;
 	private lastPreviewHeight = 0;
+	private showLabelTimestamps = false;
 	private previewCache: {
 		entryId: string;
 		width: number;
@@ -377,7 +473,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 +492,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;
@@ -413,6 +510,11 @@ class anycopyOverlay implements Focusable {
 			this.clearSelection();
 			return;
 		}
+		if (matchesKey(data, this.keys.toggleLabelTimestamps)) {
+			this.showLabelTimestamps = !this.showLabelTimestamps;
+			this.requestRender();
+			return;
+		}
 
 		if (matchesKey(data, this.keys.scrollDown)) {
 			this.previewScrollOffset += 1;
@@ -441,6 +543,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;
@@ -490,6 +596,10 @@ class anycopyOverlay implements Focusable {
 		return this.selectedNodeIds.has(id);
 	}
 
+	isShowingLabelTimestamps(): boolean {
+		return this.showLabelTimestamps;
+	}
+
 	copySelectedOrFocusedNode(): void {
 		const focused = this.getFocusedNode();
 		const ids =
@@ -551,12 +661,13 @@ 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}`;
+			` • ${formatKeyHint(this.keys.toggleLabelTimestamps)}: label time` +
+			` • Esc: close`;
 		return truncateToWidth(this.theme.fg("dim", hint), width);
 	}
 
@@ -636,7 +747,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,50 +781,77 @@ 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[];
+		const buildAnnotatedTree = (): SessionTreeNode[] => {
+			const tree = ctx.sessionManager.getTree() as SessionTreeNode[];
+			applyLabelTimestampsToTree(tree, buildLatestLabelTimestamps(ctx.sessionManager.getEntries() as SessionEntry[]));
+			return tree;
+		};
+
+		const initialTree = buildAnnotatedTree();
 		if (initialTree.length === 0) {
 			ctx.ui.notify("No entries in session", "warning");
 			return;
 		}
 
-		const getTree = () => ctx.sessionManager.getTree() as SessionTreeNode[];
+		const getTree = () => buildAnnotatedTree();
+		const refreshInitialTreeLabelTimestamps = () => {
+			applyLabelTimestampsToTree(initialTree, buildLatestLabelTimestamps(ctx.sessionManager.getEntries() as SessionEntry[]));
+		};
 		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);
+					refreshInitialTreeLabelTimestamps();
 				},
+				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 +859,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);
@@ -737,8 +866,15 @@ export default function anycopyExtension(pi: ExtensionAPI) {
 
 				const tl = treeList as any;
 				const filteredRaw = tl.filteredNodes;
+				const decorateStatusLine = (line: string): string => {
+					const status = overlay.isShowingLabelTimestamps() ? `${line}${theme.fg("muted", " [+label time]")}` : line;
+					return truncateToWidth(`  ${status}`, width);
+				};
+
 				if (!Array.isArray(filteredRaw) || filteredRaw.length === 0) {
-					return lines.map((line: string) => "  " + line);
+					return lines.map((line: string, i: number) =>
+						i === lines.length - 1 ? decorateStatusLine(line) : truncateToWidth(`  ${line}`, width),
+					);
 				}
 				const filtered = filteredRaw as { node: SessionTreeNode }[];
 
@@ -758,16 +894,18 @@ export default function anycopyExtension(pi: ExtensionAPI) {
 				const treeRowCount = Math.max(0, lines.length - 1);
 
 				return lines.map((line: string, i: number) => {
-					if (i >= treeRowCount) return "  " + line;
+					if (i >= treeRowCount) return decorateStatusLine(line);
 
 					const nodeIdx = startIdx + i;
 					const node = filtered[nodeIdx]?.node as SessionTreeNode | undefined;
 					const nodeId = node?.entry?.id;
-					if (typeof nodeId !== "string") return "  " + line;
+					if (typeof nodeId !== "string") return truncateToWidth(`  ${line}`, width);
 
 					const selected = overlay.isSelectedNode(nodeId);
-					const marker = selected ? theme.fg("success", "\u2713 ") : theme.fg("dim", "\u25CB ");
-					return marker + line;
+					const marker = selected ? theme.fg("success", "✓ ") : theme.fg("dim", "○ ");
+					const lineWithTimestamp =
+						overlay.isShowingLabelTimestamps() && node ? insertLabelTimestampIntoLine(line, node, theme) : line;
+					return truncateToWidth(marker + lineWithTimestamp, width);
 				});
 			};
 
@@ -782,13 +920,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.1",
+  "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" }
     ],