feat(pi-notes): limit notes to two lines (#179)

Author: HenryQW

package-lock.json

@@ -1,11 +1,11 @@
 # `@henryqw/pi-notes`
 
-Persistent notes shown in a Pi widget, managed with slash commands.
+Persistent post-it reminders shown in a Pi widget, managed with slash commands.
 
 ## Why
 
-- **Created for**: Keeping persistent scratch notes visible per worktree without leaving the Pi session.
-- **Advantage**: Slash-command-managed notes render in a widget and survive across sessions without leaking between worktrees.
+- **Created for**: Keeping a few brief, post-it-style reminders visible per worktree without leaving the Pi session.
+- **Advantage**: Notes stay intentionally bounded and visible instead of becoming clipboard storage or history.
 
 ## Install
 
@@ -21,7 +21,7 @@ pi install npm:@henryqw/pi-notes
 | `/note-rm` | command | Pick a note from current worktree to remove. |
 | `/note-clear` | command | Clear current worktree's notes. |
 
-Notes are isolated per Git worktree, render as a numbered widget above editor, and persist across sessions under `~/.pi/agent/config/pi-notes/`. Empty worktrees show no widget. Stale files for removed repositories and worktrees are deleted silently when a session starts or notes change.
+Notes are isolated per Git worktree, render as a numbered widget above editor with at most two lines per note, and persist across sessions under `~/.pi/agent/config/pi-notes/`. Empty worktrees show no widget. Stale files for removed repositories and worktrees are deleted silently when a session starts or notes change.
 
 Each worktree file is validated as untrusted data. Malformed files are preserved and block mutation for affected worktree until fixed or reset with `/note-clear`.
 

packages/pi-notes/README.md

@@ -7,6 +7,7 @@ import {
 	type ExtensionAPI,
 	type ExtensionContext,
 } from "@earendil-works/pi-coding-agent";
+import { truncateToWidth, wrapTextWithAnsi } from "@earendil-works/pi-tui";
 
 const MAX_NOTES = 4;
 const WIDGET_KEY = "pi-notes";
@@ -70,8 +71,29 @@ export function renderNotes(notes: string[]): string[] {
 	return notes.map((note, i) => `${i + 1}. ${note}`);
 }
 
+export function renderNotesWidget(notes: string[], width: number): string[] {
+	const renderWidth = Math.max(1, width);
+	return renderNotes(notes).flatMap((note) => {
+		const lines = wrapTextWithAnsi(note, renderWidth);
+		return lines.length > 2
+			? [lines[0]!, truncateToWidth(`${lines[1]}…`, renderWidth, "…")]
+			: lines;
+	});
+}
+
 function setNotesWidget(ctx: ExtensionContext, notes: string[]): void {
-	ctx.ui.setWidget(WIDGET_KEY, notes.length ? renderNotes(notes) : undefined);
+	if (!notes.length) {
+		ctx.ui.setWidget(WIDGET_KEY, undefined);
+		return;
+	}
+	if (ctx.mode !== "tui") {
+		ctx.ui.setWidget(WIDGET_KEY, renderNotes(notes));
+		return;
+	}
+	ctx.ui.setWidget(WIDGET_KEY, () => ({
+		invalidate() {},
+		render: (width) => renderNotesWidget(notes, width),
+	}));
 }
 
 async function resolveWorktree(pi: ExtensionAPI, cwd: string): Promise<WorktreeIdentity> {

packages/pi-notes/extensions/notes.ts

@@ -1,6 +1,6 @@
 {
   "name": "@henryqw/pi-notes",
-  "version": "0.1.4",
+  "version": "0.2.0",
   "description": "Persistent notes in a Pi widget, managed with /note commands.",
   "keywords": [
     "pi-package",
@@ -24,7 +24,8 @@
     "pack:check": "npm pack --dry-run"
   },
   "peerDependencies": {
-    "@earendil-works/pi-coding-agent": "^0.84.2"
+    "@earendil-works/pi-coding-agent": "^0.84.2",
+    "@earendil-works/pi-tui": "^0.84.2"
   },
   "repository": {
     "type": "git",

packages/pi-notes/package.json

@@ -4,7 +4,7 @@ import { tmpdir } from "node:os";
 import { join } from "node:path";
 import test, { type TestContext } from "node:test";
 import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
-import notesExtension, { parseNotes, renderNotes } from "../extensions/notes.ts";
+import notesExtension, { parseNotes, renderNotes, renderNotesWidget } from "../extensions/notes.ts";
 
 type CommandHandler = (args: string, ctx: ExtensionContext) => Promise<void>;
 type Identity = { repository: string; worktree: string; gitDir: string };
@@ -52,8 +52,13 @@ async function harness(t: TestContext) {
 	let select: ((options: string[]) => Promise<string | undefined>) | undefined;
 	const context = (cwd: string) => ({
 		cwd,
+		mode: "tui",
 		ui: {
-			setWidget(_key: string, content: string[] | undefined) { widget = content; },
+			setWidget(_key: string, content: unknown) {
+				widget = typeof content === "function"
+					? (content as () => { render(width: number): string[] })().render(80)
+					: content as string[] | undefined;
+			},
 			notify(message: string) { notified = message; },
 			select(_title: string, options: string[]) { return select ? select(options) : Promise.resolve(undefined); },
 		},
@@ -97,6 +102,13 @@ test("renderNotes numbers entries", () => {
 	assert.deepEqual(renderNotes([]), []);
 });
 
+test("widget limits each note to two lines", () => {
+	assert.deepEqual(
+		renderNotesWidget(["short", "alpha beta gamma delta"], 12),
+		["1. short", "2. alpha", "beta gamma…"],
+	);
+});
+
 test("notes persist independently per worktree without creating config on startup", async (t) => {
 	const h = await harness(t);
 	const firstCtx = h.context(h.first);