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

package/README.md

@@ -4,12 +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
 
@@ -27,7 +27,7 @@ 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,111 +1,88 @@
-import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
+import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
 
 type TodoStatus = "todo" | "partial" | "done";
 type TodoItem = { text: string; status: TodoStatus };
-type TodoState = { visible: boolean; items: TodoItem[]; offset: number; doneDismissHint: boolean };
+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 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 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;
+  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} 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. Use explicit statuses: [ ] not started, [-] partial/in progress, [x] complete. Update checklist markers immediately when work status changes, including partial progress.",
+        "\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, status: "todo" }));
-      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;
+    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 normalizedText = normalize(text);
 
     const checklistRegex = /^\s*(?:[-*]|\d+[.)])\s*\[( |x|X|-)\]\s+(.+)$/gm;
-    const checklist: Array<{ status: TodoStatus; text: string }> = [];
+    const checklist: Array<TodoItem> = [];
     for (const match of text.matchAll(checklistRegex)) {
       const mark = (match[1] || " ").toLowerCase();
-      const label = normalize(match[2] || "");
+      const label = (match[2] || "").trim().replace(/\s+/g, " ");
       const status: TodoStatus = mark === "x" ? "done" : mark === "-" ? "partial" : "todo";
-      checklist.push({ status, text: label });
+      if (label) checklist.push({ status, text: label });
     }
 
-    for (const item of state.items) {
-      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";
-      }
-    }
+    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.status === "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");
     },
   });
@@ -113,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.2",
+  "version": "0.1.4",
   "description": "Aggressive automatic todo progress widget for multi-goal prompts in Pi.",
   "license": "MIT",
   "keywords": [
@@ -16,8 +16,8 @@
     ]
   },
   "peerDependencies": {
-    "@mariozechner/pi-coding-agent": "*",
-    "@mariozechner/pi-tui": "*"
+    "@earendil-works/pi-coding-agent": "*",
+    "@earendil-works/pi-tui": "*"
   },
   "files": [
     "index.ts",