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

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,69 @@ 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" user messages and all tool outputs
+      const todoTextParts: { part: any }[] = []
+      const todoToolParts: { part: any; isLast: boolean }[] = []
+
+      for (const msg of messages) {
+        const parts = Array.isArray(msg.parts) ? msg.parts : []
+        for (const part of parts) {
+          // Tool outputs
+          if (part.type === "tool" && prunedToolNames.has(part.tool) && part.state?.status === "completed") {
+            todoToolParts.push({ part, isLast: part.callID === state.lastToolCallId })
+          }
+          // User "## TODO" snapshots
+          if (part.type === "text" && typeof part.text === "string" && part.text.startsWith("## TODO")) {
+            todoTextParts.push({ part })
+          }
+        }
+      }
+
+      // 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 allDone = doneMatch && doneMatch[1] === doneMatch[2] && parseInt(doneMatch[1]) > 0
+
+      // 3. Prune tool outputs
+      for (const { part, isLast } of todoToolParts) {
+        if (allDone || !isLast) {
+          part.state.output = "[TODO output pruned - see latest]"
+        }
+      }
+
+      // 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]"
+        }
+      }
+    },
+
     tool: {
       usethis_todo_write: write,
       usethis_todo_read: read,
@@ -22,7 +91,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.2",
   "description": "OpenCode plugin: enhanced TODO tools (dual storage + dependency graph)",
   "type": "module",
   "main": "./index.ts",