pi-anycopy 0.2.0 → 0.2.2

package/README.md

@@ -1,6 +1,6 @@
 # anycopy for Pi (`pi-anycopy`)
 
-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.
+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" />
@@ -47,10 +47,11 @@ Defaults (customizable in `config.json`):
 | Key | Action |
 |-----|--------|
 | `Enter` | Navigate to the focused node (same semantics as `/tree`) |
-| `Space` | Select/unselect focused node for copy |
+| `Shift+A` | 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` | Close |
@@ -64,8 +65,13 @@ Notes:
 - If no nodes are selected, `Shift+C` copies the focused node
 - 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+A`/`Shift+C` multi-select copy behavior is unchanged by navigation support, while plain space remains available for search queries
+- `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(...)`
+- [Folded](https://github.com/badlogic/pi-mono/blob/09e9de5749193beab234f30ed220a77f3d91cfad/packages/coding-agent/docs/tree.md#controls) branches are persisted by default in hidden `/anycopy` session entries, so closing/reopening `/anycopy`, switching to a sibling branch, or revisiting the session later restores the same folded view until you explicitly unfold it again
+- Search and filter changes still reset the live overlay's fold state temporarily; reopening `/anycopy` restores the persisted folded branches
 
 ## Configuration
 
@@ -73,15 +79,18 @@ Edit `~/.pi/agent/extensions/anycopy/config.json`:
 
 - `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 for copy/preview actions
+- `persistFoldState`: whether `/anycopy` persists folded branches across reopenings and later sessions; defaults to `true`; when disabled, `/anycopy` does not read or write hidden fold-state session entries
+- `keys`: keybindings used inside the `/anycopy` overlay for copy/preview actions, including the label timestamp toggle
 
 ```json
 {
   "treeFilterMode": "default",
+  "persistFoldState": true,
   "keys": {
-    "toggleSelect": "space",
+    "toggleSelect": "shift+a",
     "copy": "shift+c",
     "clear": "shift+x",
+    "toggleLabelTimestamps": "shift+t",
     "scrollUp": "shift+up",
     "scrollDown": "shift+down",
     "pageUp": "shift+left",

package/extensions/anycopy/config.json

@@ -1,9 +1,11 @@
 {
   "treeFilterMode": "no-tools",
+  "persistFoldState": true,
   "keys": {
-    "toggleSelect": "space",
+    "toggleSelect": "shift+a",
     "copy": "shift+c",
     "clear": "shift+x",
+    "toggleLabelTimestamps": "shift+t",
     "scrollUp": "shift+up",
     "scrollDown": "shift+down",
     "pageUp": "shift+left",

package/extensions/anycopy/index.ts

@@ -4,10 +4,11 @@
  * Layout: native TreeSelectorComponent at top, status bar, preview below
  *
  * Default keys (customizable via ./config.json):
- *   Space     - select/unselect focused node for copy
+ *   Shift+A   - 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
@@ -22,7 +23,7 @@ import {
 	TreeSelectorComponent,
 } from "@mariozechner/pi-coding-agent";
 
-import { Markdown, matchesKey, truncateToWidth } from "@mariozechner/pi-tui";
+import { getKeybindings, Markdown, matchesKey, truncateToWidth } from "@mariozechner/pi-tui";
 import type { Focusable } from "@mariozechner/pi-tui";
 
 import { existsSync, readFileSync } from "fs";
@@ -31,17 +32,29 @@ import { dirname, join } from "path";
 import { fileURLToPath } from "url";
 
 import { runAnycopyEnterNavigation } from "./enter-navigation.ts";
+import {
+	ANYCOPY_FOLD_STATE_CUSTOM_TYPE,
+	createFoldStateEntryData,
+	foldStateNodeIdListsEqual,
+	getSelectorFoldedNodeIds,
+	loadLatestFoldStateFromEntries,
+	mergeExplicitFoldMutation,
+	normalizeFoldedNodeIds,
+	setSelectorFoldedNodeIds,
+} from "./fold-state.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;
@@ -53,11 +66,13 @@ type TreeFilterMode = "default" | "no-tools" | "user-only" | "labeled-only" | "a
 type anycopyConfig = {
 	keys?: Partial<anycopyKeyConfig>;
 	treeFilterMode?: TreeFilterMode;
+	persistFoldState?: boolean;
 };
 
 type anycopyRuntimeConfig = {
 	keys: anycopyKeyConfig;
 	treeFilterMode: TreeFilterMode;
+	persistFoldState: boolean;
 };
 
 type BranchSummarySettingsFile = {
@@ -67,9 +82,10 @@ type BranchSummarySettingsFile = {
 };
 
 const DEFAULT_KEYS: anycopyKeyConfig = {
-	toggleSelect: "space",
+	toggleSelect: "shift+a",
 	copy: "shift+c",
 	clear: "shift+x",
+	toggleLabelTimestamps: "shift+t",
 	scrollDown: "shift+down",
 	scrollUp: "shift+up",
 	pageDown: "shift+right",
@@ -77,6 +93,7 @@ const DEFAULT_KEYS: anycopyKeyConfig = {
 };
 
 const DEFAULT_TREE_FILTER_MODE: TreeFilterMode = "default";
+const DEFAULT_PERSIST_FOLD_STATE = true;
 
 const getExtensionDir = (): string => {
 	// eslint-disable-next-line @typescript-eslint/no-unnecessary-condition
@@ -112,6 +129,7 @@ const loadConfig = (): anycopyRuntimeConfig => {
 		return {
 			keys: { ...DEFAULT_KEYS },
 			treeFilterMode: DEFAULT_TREE_FILTER_MODE,
+			persistFoldState: DEFAULT_PERSIST_FOLD_STATE,
 		};
 	}
 
@@ -125,23 +143,31 @@ const loadConfig = (): anycopyRuntimeConfig => {
 			typeof treeFilterModeRaw === "string" && validTreeFilterModes.includes(treeFilterModeRaw as TreeFilterMode)
 				? (treeFilterModeRaw as TreeFilterMode)
 				: DEFAULT_TREE_FILTER_MODE;
+		const persistFoldState =
+			typeof parsed.persistFoldState === "boolean" ? parsed.persistFoldState : DEFAULT_PERSIST_FOLD_STATE;
 
 		return {
 			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,
 				pageUp: typeof keys.pageUp === "string" ? keys.pageUp : DEFAULT_KEYS.pageUp,
 			},
 			treeFilterMode,
+			persistFoldState,
 		};
 	} catch {
 		return {
 			keys: { ...DEFAULT_KEYS },
 			treeFilterMode: DEFAULT_TREE_FILTER_MODE,
+			persistFoldState: DEFAULT_PERSIST_FOLD_STATE,
 		};
 	}
 };
@@ -368,6 +394,66 @@ const buildNodeOrder = (roots: SessionTreeNode[]): Map<string, number> => {
 	return order;
 };
 
+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
  */
@@ -392,6 +478,7 @@ class anycopyOverlay implements Focusable {
 	private _focused = false;
 	private previewScrollOffset = 0;
 	private lastPreviewHeight = 0;
+	private showLabelTimestamps = false;
 	private previewCache: {
 		entryId: string;
 		width: number;
@@ -404,6 +491,10 @@ class anycopyOverlay implements Focusable {
 		private getTree: () => SessionTreeNode[],
 		private nodeById: Map<string, SessionTreeNode>,
 		private keys: anycopyKeyConfig,
+		private onExplicitFoldMutation: ((
+			beforeTransientFoldedNodeIds: string[],
+			afterTransientFoldedNodeIds: string[],
+		) => void) | null,
 		private onClose: () => void,
 		private getTermHeight: () => number,
 		private requestRender: () => void,
@@ -441,6 +532,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;
@@ -465,7 +561,18 @@ class anycopyOverlay implements Focusable {
 			return;
 		}
 
+		const keybindings = getKeybindings();
+		const shouldTrackExplicitFoldMutation =
+			this.onExplicitFoldMutation !== null &&
+			(keybindings.matches(data, "app.tree.foldOrUp") || keybindings.matches(data, "app.tree.unfoldOrDown"));
+		const beforeTransientFoldedNodeIds = shouldTrackExplicitFoldMutation ? getSelectorFoldedNodeIds(this.selector) : null;
+
 		this.selector.handleInput(data);
+
+		if (beforeTransientFoldedNodeIds) {
+			this.onExplicitFoldMutation?.(beforeTransientFoldedNodeIds, getSelectorFoldedNodeIds(this.selector));
+		}
+
 		this.requestRender();
 	}
 
@@ -522,6 +629,10 @@ class anycopyOverlay implements Focusable {
 		return this.selectedNodeIds.has(id);
 	}
 
+	isShowingLabelTimestamps(): boolean {
+		return this.showLabelTimestamps;
+	}
+
 	copySelectedOrFocusedNode(): void {
 		const focused = this.getFocusedNode();
 		const ids =
@@ -588,6 +699,7 @@ class anycopyOverlay implements Focusable {
 			` • ${formatKeyHint(this.keys.toggleSelect)}: select` +
 			` • ${formatKeyHint(this.keys.copy)}: copy` +
 			` • ${formatKeyHint(this.keys.clear)}: clear` +
+			` • ${formatKeyHint(this.keys.toggleLabelTimestamps)}: label time` +
 			` • Esc: close`;
 		return truncateToWidth(this.theme.fg("dim", hint), width);
 	}
@@ -704,6 +816,7 @@ export default function anycopyExtension(pi: ExtensionAPI) {
 	const config = loadConfig();
 	const keys = config.keys;
 	const treeFilterMode = config.treeFilterMode;
+	const persistFoldState = config.persistFoldState;
 
 	const openAnycopy = async (
 		ctx: ExtensionCommandContext,
@@ -711,19 +824,35 @@ export default function anycopyExtension(pi: ExtensionAPI) {
 	) => {
 		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;
 			const treeTermHeight = Math.floor(termRows * 0.65);
+			const nodeById = buildNodeMap(initialTree);
+			const validNodeIds = new Set(nodeById.keys());
+			const restoredFoldState = persistFoldState
+				? loadLatestFoldStateFromEntries(ctx.sessionManager.getEntries() as SessionEntry[], validNodeIds)
+				: null;
+			let durableFoldedNodeIds = restoredFoldState?.foldedNodeIds ?? [];
+			let lastPersistedFoldedNodeIds = durableFoldedNodeIds;
 
 			const selector = new TreeSelectorComponent(
 				initialTree,
@@ -751,18 +880,65 @@ export default function anycopyExtension(pi: ExtensionAPI) {
 				() => done(),
 				(entryId, label) => {
 					pi.setLabel(entryId, label);
+					refreshInitialTreeLabelTimestamps();
 				},
 				opts?.initialSelectedId,
 				treeFilterMode,
 			);
-			const effectiveLeafIdForNoop = selector.getTreeList().getSelectedNode()?.entry.id ?? currentLeafId;
 
-			const nodeById = buildNodeMap(initialTree);
+			if (persistFoldState) {
+				const restoredFoldedNodeIds = normalizeFoldedNodeIds(
+					setSelectorFoldedNodeIds(selector, durableFoldedNodeIds),
+					validNodeIds,
+				);
+				durableFoldedNodeIds = restoredFoldedNodeIds;
+				lastPersistedFoldedNodeIds = restoredFoldedNodeIds;
+			}
+
+			const persistDurableFoldState = (nextDurableFoldedNodeIds: string[]): void => {
+				if (!persistFoldState || foldStateNodeIdListsEqual(nextDurableFoldedNodeIds, lastPersistedFoldedNodeIds)) {
+					return;
+				}
+
+				try {
+					pi.appendEntry(
+						ANYCOPY_FOLD_STATE_CUSTOM_TYPE,
+						createFoldStateEntryData(nextDurableFoldedNodeIds, validNodeIds),
+					);
+					lastPersistedFoldedNodeIds = nextDurableFoldedNodeIds;
+				} catch (error) {
+					ctx.ui.notify(
+						error instanceof Error ? error.message : "Failed to persist /anycopy fold state",
+						"error",
+					);
+				}
+			};
+
+			const handleExplicitFoldMutation = (
+				beforeTransientFoldedNodeIds: string[],
+				afterTransientFoldedNodeIds: string[],
+			): void => {
+				const nextDurableFoldedNodeIds = mergeExplicitFoldMutation({
+					durableFoldedNodeIds,
+					beforeTransientFoldedNodeIds,
+					afterTransientFoldedNodeIds,
+					validNodeIds,
+				});
+				if (foldStateNodeIdListsEqual(nextDurableFoldedNodeIds, durableFoldedNodeIds)) {
+					return;
+				}
+
+				durableFoldedNodeIds = nextDurableFoldedNodeIds;
+				persistDurableFoldState(nextDurableFoldedNodeIds);
+			};
+
+			const effectiveLeafIdForNoop = selector.getTreeList().getSelectedNode()?.entry.id ?? currentLeafId;
 			const overlay = new anycopyOverlay(
 				selector,
 				getTree,
 				nodeById,
 				keys,
+				persistFoldState ? handleExplicitFoldMutation : null,
 				() => done(),
 				() => tui.terminal?.rows ?? 40,
 				() => tui.requestRender(),
@@ -777,8 +953,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 }[];
 
@@ -798,16 +981,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", "✓ ") : theme.fg("dim", "○ ");
-					return marker + line;
+					const lineWithTimestamp =
+						overlay.isShowingLabelTimestamps() && node ? insertLabelTimestampIntoLine(line, node, theme) : line;
+					return truncateToWidth(marker + lineWithTimestamp, width);
 				});
 			};
 

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "pi-anycopy",
-  "version": "0.2.0",
-  "description": "Pi's /tree with a live syntax-highlighted preview and ability to copy any node(s) to the clipboard",
+  "version": "0.2.2",
+  "description": "Pi's /tree with a live syntax-highlighted preview, ability to copy any node(s) to the clipboard, and persistence of folded branches",
   "keywords": ["pi-package", "pi", "pi-coding-agent"],
   "license": "MIT",
   "repository": {