@comfanion/usethis_todo 0.1.7-dev.2 → 0.1.7-dev.3

package/index.ts

@@ -37,48 +37,56 @@ const UsethisTodoPlugin: Plugin = async ({ directory, client }) => {
         "todoread",
       ])
 
-      // 1. Collect all "## TODO" user messages and all tool outputs
-      const todoTextParts: { part: any }[] = []
-      const todoToolParts: { part: any; isLast: boolean }[] = []
+      // 1. Collect all TODO-related parts
+      const toolParts: { part: any; isLast: boolean }[] = []
+      const snapshotParts = new Set<any>()
+      let lastSnapshotPart: any = null
 
       for (const msg of messages) {
-        const parts = Array.isArray(msg.parts) ? msg.parts : []
-        for (const part of parts) {
-          // Tool outputs
+        for (const part of (msg.parts || [])) {
+          // Tool parts (contain both state.input and state.output)
           if (part.type === "tool" && prunedToolNames.has(part.tool) && part.state?.status === "completed") {
-            todoToolParts.push({ part, isLast: part.callID === state.lastToolCallId })
+            toolParts.push({ part, isLast: part.callID === state.lastToolCallId })
           }
-          // User "## TODO" snapshots
+          // User "## TODO" snapshot text parts
           if (part.type === "text" && typeof part.text === "string" && part.text.startsWith("## TODO")) {
-            todoTextParts.push({ part })
+            snapshotParts.add(part)
+            lastSnapshotPart = part
           }
         }
       }
 
+      if (toolParts.length === 0 && snapshotParts.size === 0) return
+
       // 2. Check if all tasks are done by parsing the latest snapshot
-      const lastSnapshot = todoTextParts[todoTextParts.length - 1]?.part.text || ""
-      const doneMatch = lastSnapshot.match(/\[(\d+)\/(\d+) done/)
+      const lastSnapshotText = lastSnapshotPart?.text || ""
+      const doneMatch = lastSnapshotText.match(/\[(\d+)\/(\d+) done/)
       const allDone = doneMatch && doneMatch[1] === doneMatch[2] && parseInt(doneMatch[1]) > 0
 
-      // 3. Prune tool outputs
-      for (const { part, isLast } of todoToolParts) {
+      // 3. Prune old tool parts — clear BOTH input and output
+      for (const { part, isLast } of toolParts) {
         if (allDone || !isLast) {
-          part.state.output = "[TODO output pruned - see latest]"
+          part.state.output = "[TODO pruned]"
+          if (part.state.input) part.state.input = {}
         }
       }
 
-      // 4. Prune "## TODO" user messages
-      if (allDone) {
-        // All done → prune everything, no need to keep any snapshot
-        for (const { part } of todoTextParts) {
-          part.text = "[TODO completed - all tasks done]"
-        }
-      } else if (todoTextParts.length > 1) {
-        // Active tasks → keep only the last snapshot
-        for (let i = 0; i < todoTextParts.length - 1; i++) {
-          todoTextParts[i].part.text = "[TODO snapshot pruned - see latest]"
+      // 4. Remove old "## TODO" snapshot parts entirely from messages
+      //    Keep only the last snapshot (or none if allDone)
+      const snapshotsToRemove = new Set(snapshotParts)
+      if (!allDone && lastSnapshotPart) {
+        snapshotsToRemove.delete(lastSnapshotPart)
+      }
+
+      if (snapshotsToRemove.size > 0) {
+        for (const msg of messages) {
+          if (!Array.isArray(msg.parts)) continue
+          msg.parts = msg.parts.filter((p: any) => !snapshotsToRemove.has(p))
         }
       }
+
+      // 5. Remove messages that became empty after part removal
+      output.messages = messages.filter((msg: any) => (msg.parts || []).length > 0)
     },
 
     tool: {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@comfanion/usethis_todo",
-  "version": "0.1.7-dev.2",
+  "version": "0.1.7-dev.3",
   "description": "OpenCode plugin: enhanced TODO tools (dual storage + dependency graph)",
   "type": "module",
   "main": "./index.ts",