pi-copy-message 4.0.0 → 4.0.1

package/CHANGELOG.md

@@ -1,5 +1,15 @@
 # Changelog
 
+## Unreleased
+
+## 4.0.1 - 2026-10-31
+
+- Stop enumerating graphemes once a message preview is known to need truncation; preserve preview and copied text.
+
+- Wrap only enough of a long message to fill the `Tab` peek instead of the whole message; very long messages note that more lines remain instead of counting them. Copied text is unchanged.
+
+- Cancel pending copy pickers when the host disposes their UI, including during reload, without interrupting sibling dialog teardown.
+
 ## 4.0.0 - 2026-10-01
 
 - Require Pi 1.0.0 and pin the official 1.0.0 development cohort.

package/README.md

@@ -1,157 +1,74 @@
-# pi copy-message extension
+# pi-copy-message
 
-A [pi](https://github.com/earendil-works/pi-mono) extension that adds `/copy-message`: a keyboard-first picker for copying raw session message text without terminal wrapping, padding, or rendered TUI artifacts. It also adds `/copy-user` as a direct shortcut for the most recent user message.
+Copy message text from a [Pi](https://github.com/earendil-works/pi-mono) session with a searchable picker. The copied text keeps the original code, spaces, and line breaks.
 
-`pi-copy-message` supersedes [`pi-copy-user-message`](https://github.com/fitchmultz/pi-copy-user-message), which only copied the most recent user message.
+![Flow from the /copy-message command through message search and selection to the clipboard.](.github/readme/copy-flow.png)
 
-## What it does
+## Install and start
 
-- Adds `/copy-message`
-- Adds `/copy-user` for copying the most recent user message directly
-- Copies raw stored session message text, not rendered terminal lines
-- Shows messages in chat order: oldest at top, newest at bottom
-- Selects the newest visible message by default
-- Supports role filters for user, assistant, tool/bash, and custom messages
-- Hides tool/bash messages by default
-- Supports type-to-filter search across role and message text, with `time:<term>` for timestamp search
-- Supports Home/End jumps for oldest/newest visible messages
-- Uses Pi's configured selection bindings for navigation, copy, and cancel
-- Keeps Pi's platform clipboard writes and OSC 52 fallback, but reports success only after local text readback
-- Includes fast paths: `/copy-message latest`, `/copy-message last`, and `/copy-message newest`, using Pi's canonical active context
-- Supports direct numbered copies like `/copy-message 3`
-- Supports metadata copies with `--with-meta`, `--with-metadata`, or `--with-role`
+This extension requires Pi and Node.js. See the [development guide](https://github.com/fitchmultz/pi-copy-message/blob/main/docs/development.md) for version requirements.
 
-## Install
-
-Install it from npm with pi:
+Run this command:
 
 ```bash
 pi install npm:pi-copy-message
 ```
 
-Or install it directly from GitHub with pi:
-
-```bash
-pi install https://github.com/fitchmultz/pi-copy-message
-```
-
-Use `/reload` after updating extension code on current official Pi and the maintained fork. Restart Pi after changing dependencies or the host runtime.
-
-If you prefer to load it directly from a local checkout during development:
-
-```bash
-pi -e ./extensions/copy-message.ts
-```
-
-## Usage
+Start Pi. Run `/reload` if Pi is already open.
 
-Open the picker:
+Enter this command in Pi:
 
 ```text
 /copy-message
 ```
 
-Copy the most recent user message directly:
+Type words to search roles and message text. Use ↑ or ↓ to select a message. Press Enter to copy the message.
 
-```text
-/copy-user
-```
+## Use the picker
 
-Copy the latest visible default message directly:
+Messages appear from oldest to newest. The picker selects the newest visible message first.
 
-```text
-/copy-message latest
-```
+User, assistant, and visible custom messages appear by default. Tool and bash messages stay hidden.
 
-Aliases:
+| Key | Action |
+|---|---|
+| Ctrl+T | Show or hide tool and bash messages |
+| Ctrl+U / Ctrl+A / Alt+C | Show or hide user, assistant, or custom messages |
+| Tab | Preview the selected message |
+| Alt+M | Switch between raw text and text with role/time metadata |
+| Esc / Ctrl+C | Cancel |
 
-```text
-/copy-message last
-/copy-message newest
-```
+The picker uses your Pi selection bindings. Its key hints show those bindings.
 
-Copy the 3rd default-visible message directly, matching the picker's 1-based oldest-to-newest numbering:
+## Direct copies
 
-```text
-/copy-message 3
-```
+| Command | What it copies |
+|---|---|
+| `/copy-user` | Your most recent message with text |
+| `/copy-message latest` | The latest default-visible message in Pi's active context |
+| `/copy-message 3` | The third default-visible message, counted oldest to newest |
+| `/copy-message latest --with-meta` | The latest message with its role and local time, when available |
 
-Copy with role and timestamp metadata instead of raw text only:
+`latest` follows Pi's active context, including context edits, omissions, and compaction. The picker, numbered copies, and `/copy-user` use stored messages on the current branch, including history before compaction.
 
-```text
-/copy-message latest --with-meta
-/copy-message 3 --with-role
-/copy-user --with-meta
-```
+## Limits and clipboard
 
-## Keyboard controls
+The picker requires interactive TUI mode. Direct commands also work outside TUI mode, where Pi may hide notifications.
 
-| Key | Action |
-|---|---|
-| Configured `tui.select.up` (default: `↑`) | Move to older visible message |
-| Configured `tui.select.down` (default: `↓`) | Move to newer visible message |
-| Configured `tui.select.pageUp` / `tui.select.pageDown` (default: `PageUp` / `PageDown`) | Move by one visible page in regular mode |
-| `Ctrl+PageUp` / `Ctrl+PageDown` | Move by one visible page in fullscreen mode |
-| `Home` / `End` | Jump to oldest/newest visible message in regular mode |
-| `Ctrl+Home` / `Ctrl+End` | Jump to oldest/newest visible message in fullscreen mode |
-| Type or paste text | Filter visible messages; pasted line breaks and tabs become spaces |
-| `time:<term>` | Search timestamps |
-| `Backspace` | Delete one search character |
-| `Ctrl+U` | Toggle user messages |
-| `Ctrl+A` | Toggle assistant messages |
-| `Ctrl+T` | Toggle tool/bash messages |
-| `Alt+C` | Toggle custom messages |
-| `Tab` | Toggle a wrapped preview of the selected message |
-| `Alt+M` | Toggle raw vs metadata copy format |
-| Configured `tui.select.confirm` (default: `Enter`) | Copy selected message text |
-| Configured `tui.select.cancel` (default: `Esc`/`Ctrl+C`) | Cancel |
-
-## Behavior notes
-
-- Entry IDs are hidden from the picker.
-- Mouse wheel navigates; clicking a message copies it and returns keyboard focus to the unchanged draft.
-- Session replacement, tree navigation, shutdown and reload cancel pending pickers without stale-context notifications.
-- The picker caps visible rows and scrolls instead of filling the screen.
-- Search preserves your original selected message and restores it when the search is cleared.
-- General search does not match timestamps; use `time:<term>` when you want to search by displayed time.
-- Picker previews and copy notifications remove terminal control characters; the copied text stays raw.
-- Filter labels honor the active pi theme.
-- Selection key hints show the active Pi bindings. Configured selection actions take precedence if they collide with the picker's filter, preview, or format shortcuts.
-- Only custom messages with `display: true` can appear or be copied from current `custom_message` entries or legacy message-wrapped entries; visible ones can be toggled with `Alt+C`.
-- A verified local copy notification includes the role and a short preview. Unavailable, empty or mismatched readback warns instead of claiming success; rejected writes report an error. Remote terminal delivery via OSC 52 cannot be verified, even when the remote machine's local clipboard matches.
-- Native clipboard reads are preferred where available. Bounded platform readers preserve exact text, including newlines; the extension never replaces Pi's working write fallback chain.
-- `latest` uses Pi's canonical session projection, honoring context edits, omissions and compaction. Numbered selectors, the picker and `/copy-user` retain raw active-branch history, including older pre-compaction messages.
-- `/copy-message latest` and numbered selectors include user, assistant, and visible custom messages by default while hiding tool/bash messages. `Alt+C` only affects the picker. If only hidden messages exist, `latest` falls back to the newest message so the command still does something useful.
-- `/copy-message` with no direct selector requires interactive TUI mode because the picker is a custom TUI component.
-- Direct commands such as `/copy-user`, `/copy-message latest`, and `/copy-message 3` do not require TUI mode, though non-UI modes may not display notifications.
-
-## Compatibility
-
-- Requires Pi 1.0.0 or later. Development and standalone/packed qualification use the official 1.0.0 cohort; a maintained fork must be qualified at its actual supported revision, not by version-string equivalence.
-- The picker coalesces legacy in-progress assistant snapshots within their context-window boundaries so each response appears once. Current official Pi and the maintained fork no longer write those legacy boundaries.
-- Android/Termux clipboard writes use Pi's normal fallback chain, including OSC 52 when native clipboard commands are unavailable.
-- Pi includes `Ctrl+X` for copying the latest assistant response; this extension remains useful for searchable history, other roles, metadata, and direct selectors.
-- Node.js `>=24.15.0`. `.nvmrc` pins Node 24.21.0 for local development.
-
-This package keeps pi core packages as wildcard peers (`*`) per pi package guidance. Pi aliases these imports to its own bundled copies when loading the extension, so the package never bundles or shadows pi core. Local development uses `@earendil-works/pi-coding-agent` and `@earendil-works/pi-tui` as dev dependencies for typechecking and tests.
-
-## Development
+The extension uses Pi's clipboard support and OSC 52 fallback. It reports “Copied” only after local clipboard readback matches the copied text.
 
-```bash
-npm ci
-npm run check # node --test + tsc --noEmit + pack dry-run
-```
+Unverified delivery produces a warning. A failed write produces an error.
 
-No production build or `prepare` is required. Tests run TypeScript directly with Node's type stripping, so source must stay erasable (`erasableSyntaxOnly`). Qualification uses the selected host installed in this checkout's dependency graph. The tests cover formatting, native command execution, canonical context and branch selection, both TUI modes, pointer focus and pending-picker disposal. Native addon interception and an isolated command PATH keep test clipboard writes in temporary fixture files, never the operator clipboard. Actual Android/Termux, Windows/WSL and remote terminal delivery still require platform qualification.
+The extension cannot verify remote terminal delivery. Paste into a local editor to check delivery.
 
-`package-lock.json` must only reference `registry.npmjs.org`; a test fails on private-registry URLs. Behind a registry proxy that rewrites tarball URLs, refresh the lock from public metadata and install through the proxy:
+Restart Pi after you change dependencies or the Pi runtime.
 
-```bash
-npm install --package-lock-only --ignore-scripts --registry=https://registry.npmjs.org/
-npm ci
-```
+## Docs and support
+
+The [reference](https://github.com/fitchmultz/pi-copy-message/blob/main/docs/reference.md) lists all controls, command aliases, and clipboard details. The [development guide](https://github.com/fitchmultz/pi-copy-message/blob/main/docs/development.md) covers tests and releases.
+
+[Changelog](CHANGELOG.md) · [Report an issue](https://github.com/fitchmultz/pi-copy-message/issues)
 
-Key files:
+## License
 
-- `extensions/copy-message.ts` — publishable extension implementation
-- `tests/copy-message.test.ts` — regression tests for command wiring, filtering, search, navigation, and copy formatting
+[MIT](LICENSE) · Mitch Fultz

package/extensions/copy-message.ts

@@ -6,6 +6,7 @@ import { decodeKittyPrintable, getNativeClipboard, matchesKey, truncateToWidth,
 
 const MAX_VISIBLE_MESSAGES = 8;
 const MAX_PEEK_LINES = 16;
+const PEEK_WRAP_CHARS = 65_536;
 
 export type CopyFormat = "raw" | "metadata";
 
@@ -65,10 +66,14 @@ function splitGraphemes(text: string): string[] {
 
 function truncateGraphemes(text: string, max: number): string {
 	if (max <= 0) return "";
-	const graphemes = splitGraphemes(text);
-	if (graphemes.length <= max) return text;
-	if (max === 1) return "…";
-	return `${graphemes.slice(0, max - 1).join("")}…`;
+	let count = 0;
+	let prefixEnd = 0;
+	for (const { index } of graphemeSegmenter.segment(text)) {
+		if (count === max) return `${text.slice(0, prefixEnd)}…`;
+		if (count === max - 1) prefixEnd = index;
+		count++;
+	}
+	return text;
 }
 
 function safeDisplayText(text: string): string {
@@ -257,7 +262,7 @@ function messageMatchesSearch(message: CopyableMessage, search: string): boolean
 	});
 }
 
-export function filteredMessages(messages: CopyableMessage[], visibility: MessageVisibility, search = ""): CopyableMessage[] {
+function filteredMessages(messages: CopyableMessage[], visibility: MessageVisibility, search = ""): CopyableMessage[] {
 	return messages.filter((message) => isVisibleMessage(message, visibility) && messageMatchesSearch(message, search));
 }
 
@@ -274,7 +279,7 @@ export function messageByDefaultNumber(messages: CopyableMessage[], number: numb
 	return defaultVisibleMessages(messages)[number - 1];
 }
 
-export function formatMessageForCopy(message: CopyableMessage, format: CopyFormat): string {
+function formatMessageForCopy(message: CopyableMessage, format: CopyFormat): string {
 	if (format === "raw") return message.text;
 	const time = formatTime(message.timestamp);
 	const label = roleLabel(message.role);
@@ -339,13 +344,25 @@ function renderMessageLine(
 
 function renderPeekLines(message: CopyableMessage, width: number, theme: CopyMessageTheme, format: CopyFormat): string[] {
 	const contentWidth = Math.max(1, width - 2);
-	const text = safeDisplayText(formatMessageForCopy(message, format));
-	const wrapped = wrapTextWithAnsi(styleRoleText(theme, message.role, text, false), contentWidth);
+	let wrapped: string[];
+	let complete: boolean;
+	// Wrap a growing prefix instead of the whole message. A cut prefix ends on visible text: styling
+	// codes attached to a trailing whitespace run would make the wrapper break it into blank lines that
+	// a full wrap drops. Greedy wrapping then changes only the prefix's last two wrapped lines, so stop
+	// once those lines fall outside the visible window.
+	for (let chars = PEEK_WRAP_CHARS; ; chars *= 2) {
+		complete = chars >= message.text.length;
+		const prefix = complete ? message : { ...message, text: message.text.slice(0, chars) };
+		const text = safeDisplayText(formatMessageForCopy(prefix, format));
+		wrapped = wrapTextWithAnsi(styleRoleText(theme, message.role, complete ? text : text.trimEnd(), false), contentWidth);
+		if (complete || wrapped.length >= MAX_PEEK_LINES + 2) break;
+	}
 	const shown = wrapped.slice(0, MAX_PEEK_LINES);
 	const remaining = wrapped.length - shown.length;
 	const title = theme.fg("dim", `Peek ${format === "metadata" ? "metadata" : "raw"} ${roleLabel(message.role)} message`);
 	const lines = [title, ...shown.map((line) => `  ${line}`)];
-	if (remaining > 0) lines.push(theme.fg("dim", `  … ${remaining} more wrapped line${remaining === 1 ? "" : "s"}`));
+	if (!complete) lines.push(theme.fg("dim", "  … more wrapped lines"));
+	else if (remaining > 0) lines.push(theme.fg("dim", `  … ${remaining} more wrapped line${remaining === 1 ? "" : "s"}`));
 	return lines;
 }
 
@@ -679,7 +696,7 @@ function parseCopyArgs(args: string | undefined): ParsedCopyMessageArgs {
 	return result;
 }
 
-export function copyArgumentCompletions(prefix: string, includeSelectors: boolean): AutocompleteItem[] | null {
+function copyArgumentCompletions(prefix: string, includeSelectors: boolean): AutocompleteItem[] | null {
 	if (/^\d+$/.test(prefix)) return null;
 	const candidates = includeSelectors ? [...COPY_LATEST_SELECTORS, ...COPY_METADATA_FLAGS] : COPY_METADATA_FLAGS;
 	const normalized = prefix.toLowerCase();
@@ -693,15 +710,22 @@ async function pickMessage(ctx: ExtensionCommandContext, messages: CopyableMessa
 	if (signal.aborted) return null;
 	return ctx.ui.custom<{ message: CopyableMessage; text: string } | null>((tui, theme, keybindings, done) => {
 		const state = new CopyMessagePickerState(messages, initialFormat);
-		const abort = () => done(null);
+		let settled = false;
+		const finish = (result: Parameters<typeof done>[0], defer = false) => {
+			if (settled) return;
+			settled = true;
+			if (defer) queueMicrotask(() => done(result));
+			else done(result);
+		};
+		const abort = () => finish(null);
 		if (signal.aborted) abort();
 		else signal.addEventListener("abort", abort, { once: true });
 		const act = (result: PickerInputResult) => {
 			if (result === "copy") {
 				const selected = state.selectedMessage();
 				const text = state.selectedCopyText();
-				done(selected && text !== undefined ? { message: selected, text } : null);
-			} else if (result === "cancel") done(null);
+				finish(selected && text !== undefined ? { message: selected, text } : null);
+			} else if (result === "cancel") finish(null);
 			else if (result === "render") tui.requestRender();
 		};
 		return {
@@ -715,7 +739,12 @@ async function pickMessage(ctx: ExtensionCommandContext, messages: CopyableMessa
 				act(result);
 				if (result !== "none") return { handled: true, focus: result === "render" };
 			},
-			dispose: () => signal.removeEventListener("abort", abort),
+			dispose() {
+				signal.removeEventListener("abort", abort);
+				// Settle even if disposal precedes session_shutdown. Defer the host callback:
+				// it can remove this component from a live dialog-disposal iteration.
+				finish(null, true);
+			},
 		};
 	});
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-copy-message",
-  "version": "4.0.0",
+  "version": "4.0.1",
   "description": "Pi extension with a searchable TUI picker for copying raw session messages",
   "author": "Mitch Fultz (https://github.com/fitchmultz)",
   "license": "MIT",
@@ -47,7 +47,7 @@
   "devDependencies": {
     "@earendil-works/pi-coding-agent": "1.0.0",
     "@earendil-works/pi-tui": "1.0.0",
-    "@types/node": "24.19.0",
+    "@types/node": "24.19.1",
     "typescript": "7.0.2"
   },
   "pi": {