@comfanion/usethis_todo 0.1.6 → 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.6",
+  "version": "0.1.7-dev.2",
   "description": "OpenCode plugin: enhanced TODO tools (dual storage + dependency graph)",
   "type": "module",
   "main": "./index.ts",

package/tools.ts

@@ -29,6 +29,45 @@ import fs from "fs/promises"
 // Types
 // ============================================================================
 
+interface TodoFile {
+  path: string                      // File path (relative to project root)
+  role: "input" | "output" | "test" | "doc" | "config" | "main" | "graph"
+  addedAt: string                   // ISO timestamp when linked
+  addedBy: "manual" | "auto"       // How it was added
+  lastAccessed?: string             // Last read/write timestamp
+  toolIds?: number[]                // Tool call IDs that touched this file
+  preserve?: boolean                // Don't prune even on TODO complete
+}
+
+interface TodoTool {
+  toolName: string                  // Tool name (read, write, search, etc.)
+  callId: number                    // Tool call ID in session history
+  timestamp: string                 // ISO timestamp
+  filePath?: string                 // If tool operated on file
+  status: "success" | "error"       // Tool execution status
+  tokens?: number                   // Tokens used by this tool result
+}
+
+interface TodoSearch {
+  query: string                     // Search query text
+  toolId: number                    // search() tool call ID
+  resultsCount: number              // Number of results returned
+  timestamp: string                 // ISO timestamp
+  topResults?: string[]             // Top file paths found
+}
+
+interface TodoContext {
+  totalTokens: number               // Total tokens in all linked tools
+  createdAt: string                 // Task creation timestamp
+  startedAt?: string                // When status → in_progress
+  completedAt?: string              // When status → completed
+  estimatedCleanup: number          // Estimated tokens prunable on complete
+  
+  // Tracking
+  activeTracking: boolean           // Is Mind currently tracking this TODO?
+  lastSync: string                  // Last time Mind synced metadata
+}
+
 interface Todo {
   id: string              // E01-S01-T01
   content: string         // Short task summary
@@ -39,6 +78,19 @@ interface Todo {
   blockedBy?: string[]    // IDs of blocking tasks
   createdAt?: number
   updatedAt?: number
+  
+  // Mind extensions (backward compatible - optional)
+  files?: TodoFile[]                // Associated files
+  tools?: TodoTool[]                // Associated tool calls
+  searches?: TodoSearch[]           // Associated searches
+  context?: TodoContext             // Context metadata
+  
+  // Cleanup configuration (optional)
+  cleanup?: {
+    onComplete?: "auto" | "manual" | "extract"
+    preserveFiles?: string[]        // Glob patterns to preserve
+    preserveTools?: string[]        // Tool names to preserve
+  }
 }
 
 interface NativeTodo {
@@ -260,11 +312,21 @@ function normalizeTodo(input: any): Todo {
   // Auto transition: ready -> done when releases exist
   const promotedStatus = status === "ready" && releases?.length ? "done" : status
 
-  return {
+  // Initialize Mind extensions if provided (backward compatible)
+  const normalized: Todo = {
     ...input,
     status: promotedStatus,
     releases,
   }
+
+  // Preserve existing Mind extensions if they exist
+  if (input.files) normalized.files = input.files
+  if (input.tools) normalized.tools = input.tools
+  if (input.searches) normalized.searches = input.searches
+  if (input.context) normalized.context = input.context
+  if (input.cleanup) normalized.cleanup = input.cleanup
+
+  return normalized
 }
 
 function prioRank(p?: string): number {
@@ -294,9 +356,14 @@ function todoLine(todo: Todo, byId: Map<string, Todo>): string {
   const desc = todo.description?.trim() ? ` — ${todo.description.trim()}` : ""
   const rel = todo.releases?.length ? ` [rel: ${todo.releases.join(", ")}]` : ""
   const deps = todo.blockedBy?.length ? ` ← ${todo.blockedBy.join(", ")}` : ""
+  
+  // Mind extensions
+  const files = todo.files?.length ? ` 📄${todo.files.length}` : ""
+  const tokens = todo.context?.totalTokens ? ` (${todo.context.totalTokens}t)` : ""
+  
   const ns = normalizeStatus(todo.status)
   const icon = isBlocked(todo, byId) ? "⊗" : SI(ns)
-  return `${icon} ${PE(todo.priority)} ${todo.id}: ${todo.content}${desc}${rel}${deps}`
+  return `${icon} ${PE(todo.priority)} ${todo.id}: ${todo.content}${desc}${rel}${files}${tokens}${deps}`
 }
 
 function renderNestedTodoList(todos: Todo[], allTodos?: Todo[]): string {
@@ -411,17 +478,17 @@ function formatGraph(graph: TodoGraph): string {
 // ============================================================================
 
 export const write = tool({
-  description: "Create or update TODO list. TODOv2 (Prefer this instead of TODO)",
+  description: "Create or update TODO list. Use this for TODO. For better performance use ID for task relation",
   args: {
     todos: tool.schema.array(
       tool.schema.object({
-        id: tool.schema.string().describe("Task ID in concat format: E01-S01-T01"),
-        content: tool.schema.string().describe("Short task summary"),
-        description: tool.schema.string().optional().describe("Full task description"),
-        releases: tool.schema.array(tool.schema.string()).optional().describe("Release identifiers"),
+        id: tool.schema.string().describe("Task ID in concat format: E01-S01-T01(for graph dependency)"),
+        content: tool.schema.string().describe("Short task summary or title"),
+        description: tool.schema.string().optional().describe("Full task description. Helps to remember what should be done"),
+        releases: tool.schema.array(tool.schema.string()).optional().describe("Set IDs for AUTO Release task from ready -> done(after review)"),
         status: tool.schema.string().describe("todo | in_progress | ready | done"),
         priority: tool.schema.string().describe("CRIT | HIGH | MED | LOW"),
-        blockedBy: tool.schema.array(tool.schema.string()).optional().describe("IDs of blocking tasks"),
+        blockedBy: tool.schema.array(tool.schema.string()).optional().describe("IDs of blocking tasks. this help you understand scope better"),
       }),
     ).describe("Array of todos"),
   },