pi-copy-message 1.0.1 → 1.0.2

package/CHANGELOG.md

@@ -1,5 +1,9 @@
 # Changelog
 
+## 1.0.2 - 2026-06-07
+
+- Add `/copy-user` shortcut for copying the most recent user message directly.
+
 ## 1.0.1 - 2026-06-07
 
 - Improve npm package description and keywords.

package/README.md

@@ -1,12 +1,13 @@
 # pi copy-message extension
 
-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.
+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.
 
 `pi-copy-message` supersedes [`pi-copy-user-message`](https://github.com/fitchmultz/pi-copy-user-message), which only copied the most recent user message.
 
 ## What it does
 
 - 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
@@ -50,6 +51,12 @@ Open the picker:
 /copy-message
 ```
 
+Copy the most recent user message directly:
+
+```text
+/copy-user
+```
+
 Copy the latest visible default message directly:
 
 ```text

package/extensions/copy-message.ts

@@ -86,29 +86,65 @@ function formatTime(timestamp: unknown): string {
 	return date.toLocaleTimeString(undefined, { hour: "2-digit", minute: "2-digit" });
 }
 
+function entryToCopyableMessage(entry: unknown): CopyableMessage | undefined {
+	if (entry === null || typeof entry !== "object") return undefined;
+	const record = entry as Record<string, unknown>;
+	if (record.type !== "message") return undefined;
+	if (record.message === null || typeof record.message !== "object") return undefined;
+
+	const message = record.message as Record<string, unknown>;
+	const role = typeof message.role === "string" ? message.role : "message";
+	const text = textFromMessage(message);
+	if (!text.trim()) return undefined;
+
+	return {
+		id: typeof record.id === "string" ? record.id : "unknown",
+		role,
+		timestamp: typeof record.timestamp === "string" ? record.timestamp : undefined,
+		text,
+	};
+}
+
 export function collectCopyableMessages(ctx: { sessionManager: { getBranch(): unknown[] } }): CopyableMessage[] {
-	const messages: CopyableMessage[] = [];
+	return ctx.sessionManager.getBranch().flatMap((entry) => {
+		const message = entryToCopyableMessage(entry);
+		return message ? [message] : [];
+	});
+}
+
+export type MostRecentUserMessageResult =
+	| { kind: "message"; message: CopyableMessage }
+	| { kind: "no-user-message" }
+	| { kind: "no-text" };
 
-	for (const entry of ctx.sessionManager.getBranch()) {
+export function getMostRecentUserMessage(ctx: { sessionManager: { getBranch(): unknown[] } }): MostRecentUserMessageResult {
+	const branch = ctx.sessionManager.getBranch();
+
+	for (let i = branch.length - 1; i >= 0; i--) {
+		const entry = branch[i];
 		if (entry === null || typeof entry !== "object") continue;
 		const record = entry as Record<string, unknown>;
 		if (record.type !== "message") continue;
 		if (record.message === null || typeof record.message !== "object") continue;
 
 		const message = record.message as Record<string, unknown>;
-		const role = typeof message.role === "string" ? message.role : "message";
+		if (message.role !== "user") continue;
+
 		const text = textFromMessage(message);
-		if (!text.trim()) continue;
-
-		messages.push({
-			id: typeof record.id === "string" ? record.id : "unknown",
-			role,
-			timestamp: typeof record.timestamp === "string" ? record.timestamp : undefined,
-			text,
-		});
+		if (!text.trim()) return { kind: "no-text" };
+
+		return {
+			kind: "message",
+			message: {
+				id: typeof record.id === "string" ? record.id : "unknown",
+				role: "user",
+				timestamp: typeof record.timestamp === "string" ? record.timestamp : undefined,
+				text,
+			},
+		};
 	}
 
-	return messages;
+	return { kind: "no-user-message" };
 }
 
 function commandExists(command: string): boolean {
@@ -419,7 +455,28 @@ function copySelectedMessage(ctx: Pick<ExtensionCommandContext, "ui">, selected:
 	ctx.ui.notify(`Copied ${roleLabel(selected.role)} message`, "info");
 }
 
+function copyMostRecentUserMessage(ctx: Pick<ExtensionCommandContext, "sessionManager" | "ui">) {
+	const result = getMostRecentUserMessage(ctx);
+	if (result.kind === "no-user-message") {
+		ctx.ui.notify("No user messages found", "warning");
+		return;
+	}
+	if (result.kind === "no-text") {
+		ctx.ui.notify("The most recent user message has no text to copy", "warning");
+		return;
+	}
+
+	copySelectedMessage(ctx, result.message);
+}
+
 export default function copyMessageExtension(pi: Pick<ExtensionAPI, "registerCommand">) {
+	pi.registerCommand("copy-user", {
+		description: "Copy the most recent user message to the clipboard",
+		handler: async (_args, ctx) => {
+			copyMostRecentUserMessage(ctx);
+		},
+	});
+
 	pi.registerCommand("copy-message", {
 		description: "Select a session message and copy its raw text to the clipboard",
 		handler: async (args, ctx) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-copy-message",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "Pi extension with a searchable TUI picker for copying raw session messages",
   "author": "Mitch Fultz (https://github.com/fitchmultz)",
   "license": "MIT",
@@ -15,6 +15,7 @@
     "extension",
     "clipboard",
     "copy-message",
+    "copy-user",
     "copy",
     "session-messages",
     "slash-command",