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

package/README.md

@@ -6,6 +6,7 @@ Auto todo/progress tracking for multi-goal prompts.
 
 - Auto-creates todos when prompts appear multi-step.
 - Keeps a progress widget visible until completion.
+- Uses explicit status markers: `[ ]` not started, `[-]` partial, `[x]` done.
 - Shows up to 5 rows.
 - Supports mouse wheel scrolling in custom UI.
 - Supports hiding completed list.
@@ -27,6 +28,7 @@ None.
 ## Shortcuts
 
 - `Ctrl+Alt+X` — hide completed list.
+- `Ctrl+Alt+J` / `Ctrl+Alt+K` — scroll todo list down/up.
 
 ## Tools
 

package/index.ts

@@ -1,6 +1,7 @@
 import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
 
-type TodoItem = { text: string; done: boolean };
+type TodoStatus = "todo" | "partial" | "done";
+type TodoItem = { text: string; status: TodoStatus };
 type TodoState = { visible: boolean; items: TodoItem[]; offset: number; doneDismissHint: boolean };
 
 const KEY = "todo-progress";
@@ -14,12 +15,23 @@ function splitGoals(text: string): string[] {
   return parts.slice(0, 12);
 }
 
+function normalize(text: string): string {
+  return text.toLowerCase().replace(/[`*_~]/g, "").replace(/\s+/g, " ").trim();
+}
+
+function statusLabel(status: TodoStatus): string {
+  if (status === "done") return "[x]";
+  if (status === "partial") return "[-]";
+  return "[ ]";
+}
+
 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;
+  const done = s.items.filter((i) => i.status === "done").length;
+  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);
@@ -32,7 +44,7 @@ export default function todoProgress(pi: ExtensionAPI) {
     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] Aggressively use a todo list for multi-goal requests. Prefer creating/updating todos once too much rather than once too little. Use explicit statuses: [ ] not started, [-] partial/in progress, [x] complete. Update checklist markers immediately when work status changes, including partial progress.",
     };
   });
 
@@ -40,7 +52,7 @@ export default function todoProgress(pi: ExtensionAPI) {
     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.items = goals.map((g) => ({ text: g, status: "todo" }));
       state.offset = 0;
       state.visible = true;
       render(ctx, state);
@@ -50,17 +62,44 @@ export default function todoProgress(pi: ExtensionAPI) {
 
   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();
+
+    const text = event.message.content.filter((c: any) => c.type === "text").map((c: any) => c.text).join("\n");
+    const normalizedText = normalize(text);
+
+    const checklistRegex = /^\s*(?:[-*]|\d+[.)])\s*\[( |x|X|-)\]\s+(.+)$/gm;
+    const checklist: Array<{ status: TodoStatus; text: string }> = [];
+    for (const match of text.matchAll(checklistRegex)) {
+      const mark = (match[1] || " ").toLowerCase();
+      const label = normalize(match[2] || "");
+      const status: TodoStatus = mark === "x" ? "done" : mark === "-" ? "partial" : "todo";
+      checklist.push({ status, text: label });
+    }
+
     for (const item of state.items) {
-      if (!item.done && text.includes(item.text.toLowerCase().slice(0, 24))) item.done = true;
+      const itemNorm = normalize(item.text);
+      const exact = checklist.find((c) => c.text === itemNorm);
+      const fuzzy = checklist.find((c) => c.text.includes(itemNorm) || itemNorm.includes(c.text));
+      const candidate = exact ?? fuzzy;
+
+      if (candidate) {
+        item.status = candidate.status;
+        continue;
+      }
+
+      // Fallback heuristic for free-form summaries without explicit checklist markers.
+      if (item.status !== "done" && normalizedText.includes(itemNorm.slice(0, 24))) {
+        if (/\b(done|completed|finished)\b/.test(normalizedText)) item.status = "done";
+        else if (/\b(partial|partly|in progress|started|wip)\b/.test(normalizedText)) item.status = "partial";
+      }
     }
+
     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);
+      const done = state.items.length > 0 && state.items.every((i) => i.status === "done");
       if (!done) {
         ctx.ui.notify("Todo not complete yet", "warning");
         return;

package/package.json

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