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

package/index.ts

@@ -2,6 +2,12 @@ import type { Plugin } from "@opencode-ai/plugin"
 
 import { write, read, read_five, read_by_id, update } from "./tools"
 
+interface TodoPruneState {
+  lastToolCallId: string | null
+}
+
+const pruneStates = new Map<string, TodoPruneState>()
+
 const UsethisTodoPlugin: Plugin = async ({ directory, client }) => {
   // Ensure storage directory exists on init
   try {
@@ -12,6 +18,77 @@ const UsethisTodoPlugin: Plugin = async ({ directory, client }) => {
   }
 
   return {
+    "experimental.chat.messages.transform": async (_input: unknown, output: { messages: any[] }) => {
+      const messages = output.messages || []
+      const sessionID = messages[0]?.info?.sessionID
+
+      if (!sessionID) return
+
+      const state = pruneStates.get(sessionID)
+      if (!state?.lastToolCallId) return
+
+      const prunedToolNames = new Set([
+        "usethis_todo_write",
+        "usethis_todo_update",
+        "usethis_todo_read",
+        "usethis_todo_read_five",
+        "usethis_todo_read_by_id",
+        "todowrite",
+        "todoread",
+      ])
+
+      // 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) {
+        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") {
+            toolParts.push({ part, isLast: part.callID === state.lastToolCallId })
+          }
+          // User "## TODO" snapshot text parts
+          if (part.type === "text" && typeof part.text === "string" && part.text.startsWith("## TODO")) {
+            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 lastSnapshotText = lastSnapshotPart?.text || ""
+      const doneMatch = lastSnapshotText.match(/\[(\d+)\/(\d+) done/)
+      const allDone = doneMatch && doneMatch[1] === doneMatch[2] && parseInt(doneMatch[1]) > 0
+
+      // 3. Prune old tool parts — clear BOTH input and output
+      for (const { part, isLast } of toolParts) {
+        if (allDone || !isLast) {
+          part.state.output = "[TODO pruned]"
+          if (part.state.input) part.state.input = {}
+        }
+      }
+
+      // 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: {
       usethis_todo_write: write,
       usethis_todo_read: read,
@@ -22,7 +99,15 @@ const UsethisTodoPlugin: Plugin = async ({ directory, client }) => {
 
     // UI niceties + publish snapshot into the chat
     "tool.execute.after": async (input, output) => {
-      if (!input.tool.startsWith("usethis_todo_")) return
+      if (!input.tool.startsWith("usethis_todo_") && input.tool !== "todowrite" && input.tool !== "todoread") return
+
+      // Update prune state with latest call ID
+      const sessionID = input.sessionID
+      if (sessionID) {
+        const state = pruneStates.get(sessionID) || { lastToolCallId: null }
+        state.lastToolCallId = input.callID
+        pruneStates.set(sessionID, state)
+      }
 
       const out = output.output || ""
 

package/package.json

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