@firstpick/pi-extension-todo-progress 0.1.5 → 0.1.7

package/README.md

@@ -6,7 +6,9 @@ Auto todo/progress tracking for multi-goal prompts.
 
 - Instructs the agent to create concise, agent-authored todos for multi-step work.
 - Tracks checklist markers from assistant messages instead of copying raw user prompt lines.
-- Uses explicit status markers: `[ ]` not started, `[-]` partial, `[x]` done.
+- Instructs the agent to emit markdown checklist lines exactly like `- [ ] item`, `- [-] item`, or `- [x] item`.
+- Also accepts bare markers like `[ ] item` as a fallback for robustness.
+- Strips matched checklist lines from assistant messages after mirroring them into the widget, keeping the widget as the canonical todo view.
 - Clears the widget automatically when all items are complete.
 - Shows up to 5 rows.
 - Supports hiding the current list manually.
@@ -37,10 +39,10 @@ None.
 ## Example view
 
 ```text
-Todo
-- [x] Inspect package structure
-- [-] Update README examples
-- [ ] Run readiness checks
+Todo 1/3 done, 1 partial
+[x] Inspect package structure
+[-] Update README examples
+[ ] Run readiness checks
 ```
 
 For multi-step requests, Pi keeps a compact progress widget visible and updates it as work moves from planned to in-progress to done.

package/index.ts

@@ -6,6 +6,8 @@ type TodoState = { visible: boolean; items: TodoItem[]; offset: number };
 
 const KEY = "todo-progress";
 const MAX_ROWS = 5;
+const MAX_ITEMS = 12;
+const TODO_LINE_REGEX = /^\s*(?:(?:[-*]|\d+[.)])\s*)?\[( |x|X|-)\]\s+(.+)$/;
 
 function statusLabel(status: TodoStatus): string {
   if (status === "done") return "[x]";
@@ -17,13 +19,13 @@ function clear(ctx: ExtensionContext, s: TodoState) {
   s.visible = false;
   s.items = [];
   s.offset = 0;
-  if (ctx.hasUI) ctx.ui.setWidget(KEY, []);
+  if (ctx.hasUI) ctx.ui.setWidget(KEY, undefined);
 }
 
 function render(ctx: ExtensionContext, s: TodoState) {
   if (!ctx.hasUI) return;
   if (!s.visible || s.items.length === 0) {
-    ctx.ui.setWidget(KEY, []);
+    ctx.ui.setWidget(KEY, undefined);
     return;
   }
 
@@ -41,14 +43,65 @@ function render(ctx: ExtensionContext, s: TodoState) {
   ctx.ui.setWidget(KEY, lines);
 }
 
+function parseTodoLine(line: string): TodoItem | undefined {
+  const match = TODO_LINE_REGEX.exec(line);
+  if (!match) return undefined;
+
+  const mark = (match[1] || " ").toLowerCase();
+  const label = (match[2] || "").trim().replace(/\s+/g, " ");
+  if (!label) return undefined;
+
+  return {
+    status: mark === "x" ? "done" : mark === "-" ? "partial" : "todo",
+    text: label,
+  };
+}
+
+function extractChecklist(text: string): TodoItem[] {
+  const checklist: TodoItem[] = [];
+  let inFence = false;
+
+  for (const line of text.split(/\r?\n/)) {
+    if (/^\s*```/.test(line)) {
+      inFence = !inFence;
+      continue;
+    }
+    if (inFence) continue;
+
+    const item = parseTodoLine(line);
+    if (item) checklist.push(item);
+  }
+
+  return checklist;
+}
+
+function stripChecklistLines(text: string): string {
+  let inFence = false;
+  const kept: string[] = [];
+
+  for (const line of text.split(/\r?\n/)) {
+    if (/^\s*```/.test(line)) {
+      inFence = !inFence;
+      kept.push(line);
+      continue;
+    }
+
+    if (!inFence && parseTodoLine(line)) continue;
+    kept.push(line);
+  }
+
+  return kept.join("\n").replace(/\n{3,}/g, "\n\n").trim();
+}
+
 export default function todoProgress(pi: ExtensionAPI) {
   const state: TodoState = { visible: false, items: [], offset: 0 };
 
-  pi.on("before_agent_start", async (event) => {
+  pi.on("before_agent_start", async (event, ctx) => {
+    clear(ctx, state);
     return {
       systemPrompt:
         event.systemPrompt +
-        "\n\n[TODO PROGRESS POLICY] For multi-step work, create a concise agent-authored checklist with 2-6 short items. Do not copy raw user-prompt lines as todos; rewrite them into clear action items. Use explicit markers: [ ] not started, [-] partial/in progress, [x] complete. Update checklist markers as work changes. Before your final answer, close the todo list by marking every remaining item [x] or by explicitly stating no todo list is needed for the completed single-step task.",
+        "\n\n[TODO PROGRESS POLICY] For multi-step work, create a concise agent-authored checklist with 2-6 short items. Do not copy raw user-prompt lines as todos; rewrite them into clear action items. Emit todo updates as markdown checklist lines exactly like `- [ ] item`, `- [-] item`, or `- [x] item`; do not use raw user prompt lines as todos. Update checklist markers as work changes. Todo checklists are live-turn progress only: still emit `[x]` updates when possible, but the extension will close the widget deterministically when the agent turn ends.",
     };
   });
 
@@ -60,24 +113,40 @@ export default function todoProgress(pi: ExtensionAPI) {
   pi.on("message_end", async (event, ctx) => {
     if (event.message.role !== "assistant") return;
 
-    const text = event.message.content.filter((c: any) => c.type === "text").map((c: any) => c.text).join("\n");
-
-    const checklistRegex = /^\s*(?:[-*]|\d+[.)])\s*\[( |x|X|-)\]\s+(.+)$/gm;
-    const checklist: Array<TodoItem> = [];
-    for (const match of text.matchAll(checklistRegex)) {
-      const mark = (match[1] || " ").toLowerCase();
-      const label = (match[2] || "").trim().replace(/\s+/g, " ");
-      const status: TodoStatus = mark === "x" ? "done" : mark === "-" ? "partial" : "todo";
-      if (label) checklist.push({ status, text: label });
-    }
+    const textParts = event.message.content.filter((c: any) => c.type === "text");
+    const checklist = textParts.flatMap((c: any) => extractChecklist(c.text));
 
     if (checklist.length === 0) return;
 
-    state.items = checklist.slice(0, 12);
+    state.items = checklist.slice(0, MAX_ITEMS);
     state.offset = Math.min(state.offset, Math.max(0, state.items.length - MAX_ROWS));
     state.visible = true;
     render(ctx, state);
+
+    return {
+      message: {
+        ...event.message,
+        content: event.message.content.map((c: any) => (c.type === "text" ? { ...c, text: stripChecklistLines(c.text) } : c)),
+      },
+    };
+  });
+
+  pi.on("agent_end", async (_event, ctx) => {
+    // The widget represents live progress for the active turn, not persistent task state.
+    // Do not leave stale partial/todo items visible if the model forgets a final update.
+    clear(ctx, state);
+  });
+
+  pi.on("session_shutdown", async (_event, ctx) => clear(ctx, state));
+  pi.on("session_before_switch", async (_event, ctx) => {
+    clear(ctx, state);
+    return undefined;
+  });
+  pi.on("session_before_fork", async (_event, ctx) => {
+    clear(ctx, state);
+    return undefined;
   });
+  pi.on("session_tree", async (_event, ctx) => clear(ctx, state));
 
   pi.registerShortcut("ctrl+alt+x", {
     description: "Dismiss completed todo widget",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@firstpick/pi-extension-todo-progress",
-  "version": "0.1.5",
+  "version": "0.1.7",
   "description": "Aggressive automatic todo progress widget for multi-goal prompts in Pi.",
   "license": "MIT",
   "keywords": [