@firstpick/pi-extension-todo-progress 0.1.1 → 0.1.3

package/README.md

@@ -4,11 +4,12 @@ Auto todo/progress tracking for multi-goal prompts.
 
 ## What it does
 
-- Auto-creates todos when prompts appear multi-step.
-- Keeps a progress widget visible until completion.
+- 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.
+- Clears the widget automatically when all items are complete.
 - Shows up to 5 rows.
-- Supports mouse wheel scrolling in custom UI.
-- Supports hiding completed list.
+- Supports hiding the current list manually.
 
 ## Install
 
@@ -26,7 +27,8 @@ None.
 
 ## Shortcuts
 
-- `Ctrl+Alt+X` — hide completed list.
+- `Ctrl+Alt+X` — hide current list.
+- `Ctrl+Alt+J` / `Ctrl+Alt+K` — scroll todo list down/up.
 
 ## Tools
 

package/index.ts

@@ -1,72 +1,88 @@
 import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
 
-type TodoItem = { text: string; done: boolean };
-type TodoState = { visible: boolean; items: TodoItem[]; offset: number; doneDismissHint: boolean };
+type TodoStatus = "todo" | "partial" | "done";
+type TodoItem = { text: string; status: TodoStatus };
+type TodoState = { visible: boolean; items: TodoItem[]; offset: number };
 
 const KEY = "todo-progress";
 const MAX_ROWS = 5;
 
-function splitGoals(text: string): string[] {
-  const lines = text.split(/\r?\n/).map((l) => l.trim()).filter(Boolean);
-  const bullets = lines.filter((l) => /^(-|\*|\d+[.)])\s+/.test(l)).map((l) => l.replace(/^(-|\*|\d+[.)])\s+/, ""));
-  if (bullets.length >= 2) return bullets;
-  const parts = text.split(/\band then\b|\bthen\b|,\s*and\s+/gi).map((s) => s.trim()).filter((s) => s.length > 8);
-  return parts.slice(0, 12);
+function statusLabel(status: TodoStatus): string {
+  if (status === "done") return "[x]";
+  if (status === "partial") return "[-]";
+  return "[ ]";
+}
+
+function clear(ctx: ExtensionContext, s: TodoState) {
+  s.visible = false;
+  s.items = [];
+  s.offset = 0;
+  if (ctx.hasUI) ctx.ui.setWidget(KEY, []);
 }
 
 function render(ctx: ExtensionContext, s: TodoState) {
-  if (!ctx.hasUI || !s.visible || s.items.length === 0) return;
-  const done = s.items.filter((i) => i.done).length;
+  if (!ctx.hasUI) return;
+  if (!s.visible || s.items.length === 0) {
+    ctx.ui.setWidget(KEY, []);
+    return;
+  }
+
+  const done = s.items.filter((i) => i.status === "done").length;
+  if (done === s.items.length) {
+    clear(ctx, s);
+    return;
+  }
+
+  const partial = s.items.filter((i) => i.status === "partial").length;
   const top = s.items.slice(s.offset, s.offset + MAX_ROWS);
-  const lines = [ctx.ui.theme.fg("accent", `Todo ${done}/${s.items.length}`)];
-  for (const item of top) lines.push(`${item.done ? "[x]" : "[ ]"} ${item.text}`);
+  const lines = [ctx.ui.theme.fg("accent", `Todo ${done}/${s.items.length} done${partial ? `, ${partial} partial` : ""}`)];
+  for (const item of top) lines.push(`${statusLabel(item.status)} ${item.text}`);
   if (s.items.length > MAX_ROWS) lines.push(ctx.ui.theme.fg("dim", `Scroll ${s.offset + 1}-${Math.min(s.offset + MAX_ROWS, s.items.length)} of ${s.items.length}`));
-  if (done === s.items.length && s.doneDismissHint) lines.push(ctx.ui.theme.fg("success", "Done — Ctrl+Alt+X to dismiss"));
   ctx.ui.setWidget(KEY, lines);
 }
 
 export default function todoProgress(pi: ExtensionAPI) {
-  const state: TodoState = { visible: true, items: [], offset: 0, doneDismissHint: true };
+  const state: TodoState = { visible: false, items: [], offset: 0 };
 
   pi.on("before_agent_start", async (event) => {
     return {
       systemPrompt:
         event.systemPrompt +
-        "\n\n[TODO PROGRESS POLICY] Aggressively use a todo list for multi-goal requests. Prefer creating/updating todos once too much rather than once too little. Mark items done as soon as completed.",
+        "\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.",
     };
   });
 
   pi.on("input", async (event, ctx) => {
-    if (event.text.startsWith("/")) return { action: "continue" as const };
-    const goals = splitGoals(event.text);
-    if (goals.length >= 2) {
-      state.items = goals.map((g) => ({ text: g, done: false }));
-      state.offset = 0;
-      state.visible = true;
-      render(ctx, state);
-    }
+    if (!event.text.startsWith("/")) clear(ctx, state);
     return { action: "continue" as const };
   });
 
   pi.on("message_end", async (event, ctx) => {
-    if (!state.items.length || event.message.role !== "assistant") return;
-    const text = event.message.content.filter((c: any) => c.type === "text").map((c: any) => c.text).join("\n").toLowerCase();
-    for (const item of state.items) {
-      if (!item.done && text.includes(item.text.toLowerCase().slice(0, 24))) item.done = true;
+    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 });
     }
+
+    if (checklist.length === 0) return;
+
+    state.items = checklist.slice(0, 12);
+    state.offset = Math.min(state.offset, Math.max(0, state.items.length - MAX_ROWS));
+    state.visible = true;
     render(ctx, state);
   });
 
   pi.registerShortcut("ctrl+alt+x", {
     description: "Dismiss completed todo widget",
     handler: async (ctx) => {
-      const done = state.items.length > 0 && state.items.every((i) => i.done);
-      if (!done) {
-        ctx.ui.notify("Todo not complete yet", "warning");
-        return;
-      }
-      state.visible = false;
-      ctx.ui.setWidget(KEY, []);
+      clear(ctx, state);
       ctx.ui.notify("Todo widget dismissed", "info");
     },
   });
@@ -74,5 +90,5 @@ export default function todoProgress(pi: ExtensionAPI) {
   pi.registerShortcut("ctrl+alt+j", { description: "Todo scroll down", handler: async (ctx) => { state.offset = Math.min(Math.max(0, state.items.length - MAX_ROWS), state.offset + 1); render(ctx, state); } });
   pi.registerShortcut("ctrl+alt+k", { description: "Todo scroll up", handler: async (ctx) => { state.offset = Math.max(0, state.offset - 1); render(ctx, state); } });
 
-  pi.on("session_start", async (_event, ctx) => render(ctx, state));
+  pi.on("session_start", async (_event, ctx) => clear(ctx, state));
 }

package/package.json

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