pi-context 1.0.3 → 1.0.5

package/README.md

@@ -4,7 +4,8 @@ A Git-like context management tool that allows AI agents to proactively manage t
 
 Inspired by kimi-cli d-mail, implementing lossless time travel on the Pi session tree.
 
-Blog: [中文](https://blog.xlab.app/p/6a966aeb/) [EN](https://blog.xlab.app/p/51d26495/)
+For the design philosophy, see the [blog post](https://blog.xlab.app/p/51d26495/)
+ ([中文版本](https://blog.xlab.app/p/6a966aeb/)).
 
 ## Installation
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-context",
-  "version": "1.0.3",
+  "version": "1.0.5",
   "description": "agent-driven context management tools",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -29,7 +29,8 @@
   },
   "pi": {
     "extensions": [
-      "./src/index.ts"
+      "./src/tools.ts",
+      "./src/commands.ts"
     ],
     "skills": [
       "./skills"

package/skills/context-management/SKILL.md

@@ -39,15 +39,15 @@ Follow this cycle for every major task:
 
 1.  **CHECK:** Verify state.
     `context_log`
-2.  **START:** Tag the beginning.
-    `context_tag({ name: "task-start" })`
+2.  **START:** Tag the beginning with a semantic name.
+    `context_tag({ name: "<task-slug>-start" })`  // e.g., `auth-login-start`
 3.  **WORK:** Execute steps.
-4.  **MILESTONE:** Tag intermediate stable states (e.g., "Plan done", "Part 1 done").
-    `context_tag({ name: "task-plan-done" })`
+4.  **MILESTONE:** Tag intermediate stable states.
+    `context_tag({ name: "<task-slug>-plan" })`  // e.g., `auth-login-plan`
 5.  **SQUASH (Autonomous):** If history becomes noisy or low-density, **Squash with Backup**.
-    *   *Action:* `context_checkout({ target: "task-start", message: "Summarized debugging steps...", backupTag: "pre-squash-backup" })`
-    *   *Action (Optional):* `context_tag({ name: "clean-state" })`
-    *   *Safety:* If you need the details later, you can checkout `pre-squash-backup`.
+    *   *Action:* `context_checkout({ target: "<task-slug>-start", message: "...", backupTag: "<task-slug>-raw-history" })`
+    *   *Action (Optional):* `context_tag({ name: "<task-slug>-done" })`
+    *   *Safety:* If you need the details later, checkout the backup tag.
 
 ## Tool Reference
 
@@ -60,11 +60,32 @@ Follow this cycle for every major task:
 ## Critical Rules
 
 ### Tag Wisely (Build The Skeleton)
-Tags are the "Table of Contents". Use them to mark **Start** and **Backups**.
-*   **Start:** `task-start`
-*   **Backup:** `task-raw-history` (Created automatically by `backupTag`)
-*   **Milestone:** `phase-1-done`
 
+Tags are the "Table of Contents". Name them so you can understand the history at a glance.
+
+**Naming Formula:** `<task-slug>-<phase>`
+
+*   **task-slug**: Short, kebab-case identifier for the task (e.g., `auth-login`, `db-migration`)
+*   **phase**: The stage of work (`start`, `plan`, `impl`, `done`, `fail`, `backup`)
+
+| Bad (Generic) | Good (Semantic) | Why |
+| :--- | :--- | :--- |
+| `task-start` | `auth-oauth-start` | Describes WHAT task |
+| `pre-research` | `error-log-analysis-start` | Future-you knows the topic |
+| `phase-1-done` | `db-schema-plan-done` | Know which phase of which task |
+| `debug-retry` | `null-pointer-fix-retry` | What bug? |
+
+**Tag Categories:**
+
+| Category | Pattern | Examples |
+| :--- | :--- | :--- |
+| **Start** | `<task>-start` | `auth-jwt-start`, `docker-setup-start` |
+| **Plan** | `<task>-plan` | `api-v2-plan`, `migration-plan` |
+| **Milestone** | `<task>-<milestone>` | `auth-jwt-impl-done`, `tests-passed` |
+| **Backup** | `<task>-raw-history` | `auth-jwt-raw-history` |
+| **Failure** | `<task>-fail-<reason>` | `auth-jwt-fail-timeout` |
+
+**How to generate:** Ask yourself "What is the task?" → Extract 1-3 keywords (e.g., "fix login timeout" → `login-timeout-fix-start`)
 
 ### Squash Noise, Keep Signal, Focus on Goal (Context Hygiene)
 Think of your conversation as a "Feature Branch" full of messy thoughts.
@@ -94,7 +115,7 @@ If you fail 3 times:
 
 | Situation | Action | Reason |
 | :--- | :--- | :--- |
-| **Starting Task** | `context_tag({ name: "task-X-start" })` | Create a rollback point. |
+| **Starting Task** | `context_tag({ name: "<task-slug>-start" })` | Create a rollback point. |
 | **Research / Logs** | `context_checkout` (Squash) | **Process is Noise.** Read 2000 lines -> Keep result. |
 | **Messy Debugging** | **Squash w/ Backup** | **Cleanup.** The error logs are noise once fixed. |
 | **Task Done (Candidate)**| **Squash w/ Backup** | **Assume Success.** Summary is usually enough. Backup exists if not. |
@@ -117,17 +138,16 @@ If you cannot answer these, run `context_log`:
 The `message` is your lifeline to your past self.
 A good message preserves critical context that would otherwise be lost.
 
-Structure: `[Status] + [Reason] + [Carryover Data] + [Important Changes]`
+Structure: `[Key Finding/Status] + [Reason] + [Important Changes]`
 
-*   **Status**: What did you just finish or stop doing?
-*   **Reason**: Why are you branching/moving? (e.g., "Too much noise", "Task complete", "Failed attempt")
-*   **Carryover**: What specific details (e.g., IDs, file paths, user constraints) must be remembered?
+*   **Key Finding/Status**: What did you discover or complete? Include specific numbers, errors, or outcomes.
+*   **Reason**: Why are you branching/moving? (e.g., "Task complete", "Approach failed", "Need raw logs")
 *   **Important Changes**: What files or logic have been modified? (This checkout only resets *conversation history*, NOT disk files, so you must remember what changed.)
 
 Examples:
 
-*   *Good (Resetting after failure)*: "Abandoning the recursive approach (infinite loop). Switching back to iterative. **Important Changes**: Modified `utils/recursion.ts`. **Carryover**: The test case `test_retry_logic` is the one failing."
-*   *Good (Cleaning up)*: "Completed authentication module. All tests passed. **Important Changes**: Created `auth/` directory and updated `routes.ts`. **Carryover**: The user ID is stored in `localStorage` under `auth_token`. Moving to Dashboard UI."
+*   *Good (Resetting after failure)*: "Recursive parser hit stack overflow at depth 8000. Switching to iterative. **Reason**: Stack limit reached. **Important Changes**: Modified `utils/recursion.ts`."
+*   *Good (Cleaning up)*: "Auth module complete: JWT + OAuth2 + RBAC. 23 tests passing. **Reason**: Task done, cleaning context. **Important Changes**: Created `auth/`, modified `routes.ts` and `middleware.ts`."
 *   *Bad*: "Switching context." (Too vague - you will forget why)
 *   *Bad*: "Done." (What is done? What did we learn?)
 
@@ -141,6 +161,7 @@ Examples:
 | **Panic** (Apologize repeatedly for errors) | **Revert** (`context_checkout`) to before the error. |
 | **Blind Checkout** (Guessing IDs) | **Look** (`context_log`) first to get valid IDs. |
 | **Vague Summaries** ("Done", "Fixed") | **Detailed Summaries** ("Found bug in line 40. Fixed with patch X.") |
+| **Generic Tag Names** (`task-start`, `phase-1`) | **Semantic Names** (`auth-jwt-start`, `db-schema-plan`) |
 
 ## Recipes (Copy-Paste)
 
@@ -148,19 +169,21 @@ Examples:
 **Goal:** Pure information gathering (Reading files, Searching web).
 **Why:** The *process* of searching is irrelevant. Only the *result* matters.
 
+**Example Task:** Analyzing error logs to find root cause of timeout
+
 ```javascript
-// 1. Tag BEFORE starting the noisy work
-context_tag({ name: "pre-research" });
+// 1. Tag BEFORE starting the noisy work (use descriptive name)
+context_tag({ name: "timeout-analysis-start" });
 
-// ... (Read 5 files, search 3 sites, find 1 key fact) ...
+// ... (Read 5 log files, search 3 docs, find DB connection pool exhaustion) ...
 
 // 2. Squash IMMEDIATELY. Do not wait for user.
 context_checkout({
-  target: "pre-research",
-  message: "Researched logs. Found root cause: DB timeout. Irrelevant logs discarded.",
-  backupTag: "raw-research-logs" // Safety backup
+  target: "timeout-analysis-start",
+  message: "Found DB connection pool exhaustion as root cause (pool size: 10, peak load: 1000 req/s). Recommended fix: increase to 50. **Reason**: Context cleanup after research. **Important Changes**: None (read-only).",
+  backupTag: "timeout-analysis-raw-history" // Safety backup
 });
-context_tag({ name: "research-done" });
+context_tag({ name: "timeout-analysis-done" });
 ```
 
 ### 2. The "Candidate" (Wait for Confirmation)
@@ -168,52 +191,60 @@ context_tag({ name: "research-done" });
 **Why:** The history is noisy. The result is clean.
 **Safety:** We create a backup tag automatically.
 
+**Example Task:** Implementing OAuth login flow
+
 ```javascript
 // Squash to Summary (Optimistic Cleanup)
 context_checkout({
-  target: "feature-a-start", // Squash range: Start -> Now
-  message: "Feature A implemented. 5 files changed. Tests passed. Revert to 'feature-a-raw-history' for full logs.",
-  backupTag: "feature-a-raw-history"
+  target: "oauth-impl-start", // Squash range: Start -> Now
+  message: "OAuth2 flow implemented with PKCE, Google + GitHub providers. All 12 tests passing. **Reason**: Task complete, cleaning up. **Important Changes**: Created `auth/oauth.ts`, modified `routes.ts`, `config.ts`. Backup at 'oauth-impl-raw-history'.",
+  backupTag: "oauth-impl-raw-history"
 });
-context_tag({ name: "feature-a-candidate" });
+context_tag({ name: "oauth-impl-candidate" });
 ```
 
 ### 3. The "Undo" (Revert Squash)
 **Goal:** User asks about a detail you squashed away.
 **Action:** Jump back to the backup tag.
 
+**Example Task:** Reviewing OAuth implementation details
+
 ```javascript
 // Jump back to the raw history
 context_checkout({
-  target: "feature-a-raw-history",
-  message: "Reverting to raw history to check specific error logs."
+  target: "oauth-impl-raw-history",
+  message: "Reviewing token refresh logic - user reports 401 after 15 min idle. Suspect refresh token not firing. **Reason**: Need raw logs to trace the bug. **Important Changes**: None."
 });
-context_tag({ name: "restored-history" });
+context_tag({ name: "oauth-review-start" });
 ```
 
 ### 4. Branching (Alternative Approach)
 **Scenario:** Method A failed (and was squashed). You want to try Method B from the clean state.
 **Action:** Checkout the start point.
 
+**Example Task:** Fixing memory leak - trying different approaches
+
 ```javascript
-// Jump back to start
+// Method A (weak references) failed, trying Method B (object pooling)
 context_checkout({
-  target: "task-start", 
-  message: "Method A failed (see summary in 'method-a-fail'). Starting Method B from clean state."
+  target: "memory-leak-fix-start", 
+  message: "WeakRef approach failed: objects GC'd within 30s (expected: 5min). Cache hit rate dropped from 95% to 12%. **Reason**: Switching to object pooling approach. **Important Changes**: `CacheManager.ts` modified (will revert)."
 });
-context_tag({ name: "method-b-start" });
+context_tag({ name: "memory-leak-pool-approach-start" });
 ```
 
 ### 5. The "Undo" (Failed Attempt)
 You tried to fix a bug but broke everything.
 **Goal:** Clean up a failed path.
 
+**Example Task:** Fixing race condition in async handler
+
 ```javascript
-// Jump back to safety.
+// Attempted mutex-based fix, but introduced deadlock
 context_checkout({
-  target: "pre-debug-tag",
-  message: "Attempted fix using Method A failed. Error: 'Timeout'.",
-  backupTag: "failed-attempt-1" // Save the failure just in case
+  target: "race-condition-fix-start",
+  message: "Mutex caused deadlock: Thread A holds mutex, awaits callback; callback needs mutex held by B; B waits for A. Circular wait detected. **Reason**: Trying lock-free CAS approach next. **Important Changes**: `AsyncQueue.ts` lines 70-90 modified (backup saved).",
+  backupTag: "race-condition-mutex-fail" // Save the failure for reference
 });
-context_tag({ name: "debug-retry" });
+context_tag({ name: "race-condition-lockfree-start" });
 ```

package/src/commands.ts

@@ -0,0 +1,153 @@
+import {
+  type ExtensionAPI,
+  type SessionManager,
+  DynamicBorder,
+} from "@mariozechner/pi-coding-agent";
+import { Container, Text, Spacer } from "@mariozechner/pi-tui";
+import { formatTokens } from "./utils.js";
+
+export default function (pi: ExtensionAPI) {
+  pi.registerCommand("context", {
+    description: "Show context usage visualization",
+    handler: async (args, ctx) => {
+      const usage = await ctx.getContextUsage();
+      if (!usage) {
+        ctx.ui.notify("Context usage info not available.", "warning");
+        return;
+      }
+
+      const sm = ctx.sessionManager as SessionManager;
+      const branch = sm.getBranch();
+      const systemPrompt = ctx.getSystemPrompt();
+      const tools = pi.getActiveTools();
+      const allTools = pi.getAllTools();
+      const activeToolDefs = allTools.filter(t => tools.includes(t.name));
+
+      const estimateTokens = (text: string) => Math.ceil(text.length / 4);
+
+      let msgTokensRaw = 0;
+      let toolUseTokensRaw = 0;
+      let toolResultTokensRaw = 0;
+
+      for (const entry of branch) {
+        if (entry.type === "message") {
+          const m = entry.message;
+          if (m.role === "user") {
+            if (typeof m.content === "string") msgTokensRaw += estimateTokens(m.content);
+            else if (Array.isArray(m.content)) {
+              for (const p of m.content) if (p.type === "text") msgTokensRaw += estimateTokens(p.text);
+            }
+          } else if (m.role === "assistant") {
+            if (typeof m.content === "string") msgTokensRaw += estimateTokens(m.content);
+            else if (Array.isArray(m.content)) {
+              for (const p of m.content) {
+                if (p.type === "text") msgTokensRaw += estimateTokens(p.text);
+                if (p.type === "toolCall") toolUseTokensRaw += estimateTokens(JSON.stringify(p));
+              }
+            }
+          } else if (m.role === "toolResult") {
+            if (Array.isArray(m.content)) {
+              for (const p of m.content) if (p.type === "text") toolResultTokensRaw += estimateTokens(p.text);
+            }
+          } else if (m.role === "bashExecution") {
+            toolUseTokensRaw += estimateTokens(m.command || "");
+          }
+        } else if (entry.type === "branch_summary" || entry.type === "compaction") {
+          msgTokensRaw += estimateTokens(entry.summary || "");
+        }
+      }
+
+      const systemTokensRaw = estimateTokens(systemPrompt);
+      const toolDefTokensRaw = estimateTokens(JSON.stringify(activeToolDefs));
+      const totalActual = usage.tokens;
+      const limit = usage.contextWindow;
+
+      const totalRaw = systemTokensRaw + toolDefTokensRaw + msgTokensRaw + toolUseTokensRaw + toolResultTokensRaw;
+      const ratio = totalRaw > 0 ? (totalActual / totalRaw) : 1;
+
+      const systemTokens = Math.round(systemTokensRaw * ratio);
+      const toolDefTokens = Math.round(toolDefTokensRaw * ratio);
+      const msgTokens = Math.round(msgTokensRaw * ratio);
+      const toolUseTokens = Math.round(toolUseTokensRaw * ratio);
+      const toolResultTokens = Math.round(toolResultTokensRaw * ratio);
+
+      await ctx.ui.custom((tui, theme, kb, done) => {
+        const container = new Container();
+        container.addChild(new DynamicBorder((s: string) => theme.fg("accent", s)));
+        container.addChild(new Text(theme.fg("accent", theme.bold(" Context Usage")), 1, 0));
+        container.addChild(new Spacer(1));
+
+        // Grouped by function and color
+        const categories = [
+          { label: "System Prompt", value: systemTokens, color: "muted" },
+          { label: "System Tools", value: toolDefTokens, color: "dim" },
+          { label: "Tool Call", value: toolUseTokens + toolResultTokens, color: "success" },
+          { label: "Messages", value: msgTokens, color: "accent" },
+        ];
+
+        const otherTokens = Math.max(0, totalActual - (systemTokens + toolDefTokens + msgTokens + toolUseTokens + toolResultTokens));
+        if (otherTokens > 10) categories.push({ label: "Other", value: otherTokens, color: "dim" });
+
+        categories.push({ label: "Available", value: Math.max(0, limit - totalActual), color: "borderMuted" });
+
+        const gridWidth = 10;
+        const gridHeight = 5;
+        const totalBlocks = gridWidth * gridHeight;
+
+        const blocks: { color: string, filled: boolean }[] = [];
+        categories.forEach((cat) => {
+          if (cat.label === "Available") return;
+          let count = Math.round((cat.value / limit) * totalBlocks);
+          if (count === 0 && cat.value > 0) count = 1;
+          for (let i = 0; i < count && blocks.length < totalBlocks; i++) {
+            blocks.push({ color: cat.color, filled: true });
+          }
+        });
+
+        while (blocks.length < totalBlocks) {
+          blocks.push({ color: "borderMuted", filled: false });
+        }
+
+        const gridLines: string[] = [];
+        for (let r = 0; r < gridHeight; r++) {
+          let rowStr = "";
+          for (let c = 0; c < gridWidth; c++) {
+            const b = blocks[r * gridWidth + c];
+            rowStr += theme.fg(b.color as any, b.filled ? "■ " : "□ ");
+          }
+          gridLines.push(rowStr.trimEnd());
+        }
+
+        const totalUsageTitle = `${theme.fg("text", theme.bold("Total Usage".padEnd(16)))} ${theme.fg("text", theme.bold(formatTokens(totalActual).padStart(7)))} ${theme.fg("text", theme.bold(`(${usage.percent.toFixed(1).padStart(5)}%)`))}`;
+
+        const catDetailLines = categories.map(cat => {
+          const labelStr = cat.label.padEnd(14);
+          const valStr = formatTokens(cat.value).padStart(7);
+          const rowPercent = ((cat.value / limit) * 100).toFixed(1).padStart(5);
+          const icon = cat.label === "Available" ? "□" : "■";
+          return `${theme.fg(cat.color as any, icon)} ${theme.fg("text", labelStr)} ${theme.fg("accent", valStr)} (${rowPercent}%)`;
+        });
+
+        const allDetailLines = [totalUsageTitle, "", ...catDetailLines];
+
+        const leftSideWidth = 20;
+        const maxH = Math.max(gridLines.length, allDetailLines.length);
+        for (let i = 0; i < maxH; i++) {
+          const left = (gridLines[i] || "").padEnd(leftSideWidth);
+          const right = allDetailLines[i] || "";
+          container.addChild(new Text(`    ${left}      ${right}`, 1, 0));
+        }
+
+        container.addChild(new Spacer(1));
+        container.addChild(new Text(theme.fg("dim", " Press any key to close"), 1, 0));
+        container.addChild(new DynamicBorder((s: string) => theme.fg("accent", s)));
+
+        return {
+          render: (w) => container.render(w),
+          invalidate: () => container.invalidate(),
+          handleInput: (data) => done(undefined),
+        };
+      }, { overlay: true });
+    }
+  });
+}

package/src/{index.ts → tools.ts}

@@ -1,23 +1,15 @@
 import {
   type ExtensionAPI,
-  type SessionEntry,
   type SessionManager,
-  DynamicBorder,
+  type SessionEntry,
 } from "@mariozechner/pi-coding-agent";
 import type {
-  ToolCall,
   TextContent,
   ImageContent,
+  ToolCall,
 } from "@mariozechner/pi-ai";
 import { Type, type Static } from "@sinclair/typebox";
-import { Container, Text, Spacer } from "@mariozechner/pi-tui";
-
-// Define missing types locally as they are not exported from the main entry point
-interface SessionTreeNode {
-  entry: SessionEntry;
-  children: SessionTreeNode[];
-  label?: string;
-}
+import { isInternal, resolveTargetId, formatTokens } from "./utils.js";
 
 const ContextLogParams = Type.Object({
   limit: Type.Optional(Type.Number({ description: "History limit for visible entries (default: 50)." })),
@@ -35,32 +27,6 @@ const ContextTagParams = Type.Object({
   target: Type.Optional(Type.String({ description: "The commit ID to tag. Defaults to HEAD (current state)." })),
 });
 
-const isInternal = (name: string) => ["context_tag", "context_log", "context_checkout"].includes(name);
-
-const resolveTargetId = (sm: SessionManager, target: string): string => {
-  if (target.toLowerCase() === "root") {
-    const tree = sm.getTree();
-    return tree.length > 0 ? tree[0].entry.id : target;
-  }
-  if (/^[0-9a-f]{8,}$/i.test(target)) return target;
-  const find = (nodes: SessionTreeNode[]): string | null => {
-    for (const n of nodes) {
-      if (sm.getLabel(n.entry.id) === target) return n.entry.id;
-      const r = find(n.children);
-      if (r) return r;
-    }
-    return null;
-  };
-  // sm.getTree() returns the SDK's SessionTreeNode[], which is structurally compatible
-  return find(sm.getTree()) || target;
-};
-
-const formatTokens = (n: number) => {
-  if (n >= 1_000_000) return (n / 1_000_000).toFixed(1) + "M";
-  if (n >= 1_000) return Math.round(n / 1_000) + "k";
-  return n.toString();
-};
-
 export default function (pi: ExtensionAPI) {
   pi.registerTool({
     name: "context_tag",
@@ -343,149 +309,4 @@ export default function (pi: ExtensionAPI) {
       return { content: [{ type: "text", text: `Checked out ${tid}\nBackup tag created: ${params.backupTag || "none"}\nmessage: ${enrichedMessage}` }], details: {} };
     },
   });
-
-
-  pi.registerCommand("context", {
-    description: "Show context usage visualization",
-    handler: async (args, ctx) => {
-      const usage = await ctx.getContextUsage();
-      if (!usage) {
-        ctx.ui.notify("Context usage info not available.", "warning");
-        return;
-      }
-
-      const sm = ctx.sessionManager as SessionManager;
-      const branch = sm.getBranch();
-      const systemPrompt = ctx.getSystemPrompt();
-      const tools = pi.getActiveTools();
-      const allTools = pi.getAllTools();
-      const activeToolDefs = allTools.filter(t => tools.includes(t.name));
-
-      const estimateTokens = (text: string) => Math.ceil(text.length / 4);
-
-      let msgTokensRaw = 0;
-      let toolUseTokensRaw = 0;
-      let toolResultTokensRaw = 0;
-
-      for (const entry of branch) {
-        if (entry.type === "message") {
-          const m = entry.message;
-          if (m.role === "user") {
-            if (typeof m.content === "string") msgTokensRaw += estimateTokens(m.content);
-            else if (Array.isArray(m.content)) {
-              for (const p of m.content) if (p.type === "text") msgTokensRaw += estimateTokens(p.text);
-            }
-          } else if (m.role === "assistant") {
-            if (typeof m.content === "string") msgTokensRaw += estimateTokens(m.content);
-            else if (Array.isArray(m.content)) {
-              for (const p of m.content) {
-                if (p.type === "text") msgTokensRaw += estimateTokens(p.text);
-                if (p.type === "toolCall") toolUseTokensRaw += estimateTokens(JSON.stringify(p));
-              }
-            }
-          } else if (m.role === "toolResult") {
-            if (Array.isArray(m.content)) {
-              for (const p of m.content) if (p.type === "text") toolResultTokensRaw += estimateTokens(p.text);
-            }
-          } else if (m.role === "bashExecution") {
-            toolUseTokensRaw += estimateTokens(m.command || "");
-          }
-        } else if (entry.type === "branch_summary" || entry.type === "compaction") {
-          msgTokensRaw += estimateTokens(entry.summary || "");
-        }
-      }
-
-      const systemTokensRaw = estimateTokens(systemPrompt);
-      const toolDefTokensRaw = estimateTokens(JSON.stringify(activeToolDefs));
-      const totalActual = usage.tokens;
-      const limit = usage.contextWindow;
-
-      const totalRaw = systemTokensRaw + toolDefTokensRaw + msgTokensRaw + toolUseTokensRaw + toolResultTokensRaw;
-      const ratio = totalRaw > 0 ? (totalActual / totalRaw) : 1;
-
-      const systemTokens = Math.round(systemTokensRaw * ratio);
-      const toolDefTokens = Math.round(toolDefTokensRaw * ratio);
-      const msgTokens = Math.round(msgTokensRaw * ratio);
-      const toolUseTokens = Math.round(toolUseTokensRaw * ratio);
-      const toolResultTokens = Math.round(toolResultTokensRaw * ratio);
-
-      await ctx.ui.custom((tui, theme, kb, done) => {
-        const container = new Container();
-        container.addChild(new DynamicBorder((s: string) => theme.fg("accent", s)));
-        container.addChild(new Text(theme.fg("accent", theme.bold(" Context Usage")), 1, 0));
-        container.addChild(new Spacer(1));
-
-        // Grouped by function and color
-        const categories = [
-          { label: "System Prompt", value: systemTokens, color: "muted" },
-          { label: "System Tools", value: toolDefTokens, color: "dim" },
-          { label: "Tool Call", value: toolUseTokens + toolResultTokens, color: "success" },
-          { label: "Messages", value: msgTokens, color: "accent" },
-        ];
-
-        const otherTokens = Math.max(0, totalActual - (systemTokens + toolDefTokens + msgTokens + toolUseTokens + toolResultTokens));
-        if (otherTokens > 10) categories.push({ label: "Other", value: otherTokens, color: "dim" });
-
-        categories.push({ label: "Available", value: Math.max(0, limit - totalActual), color: "borderMuted" });
-
-        const gridWidth = 10;
-        const gridHeight = 5;
-        const totalBlocks = gridWidth * gridHeight;
-
-        const blocks: { color: string, filled: boolean }[] = [];
-        categories.forEach((cat) => {
-          if (cat.label === "Available") return;
-          let count = Math.round((cat.value / limit) * totalBlocks);
-          if (count === 0 && cat.value > 0) count = 1;
-          for (let i = 0; i < count && blocks.length < totalBlocks; i++) {
-            blocks.push({ color: cat.color, filled: true });
-          }
-        });
-
-        while (blocks.length < totalBlocks) {
-          blocks.push({ color: "borderMuted", filled: false });
-        }
-
-        const gridLines: string[] = [];
-        for (let r = 0; r < gridHeight; r++) {
-          let rowStr = "";
-          for (let c = 0; c < gridWidth; c++) {
-            const b = blocks[r * gridWidth + c];
-            rowStr += theme.fg(b.color as any, b.filled ? "■ " : "□ ");
-          }
-          gridLines.push(rowStr.trimEnd());
-        }
-
-        const totalUsageTitle = `${theme.fg("text", theme.bold("Total Usage".padEnd(16)))} ${theme.fg("text", theme.bold(formatTokens(totalActual).padStart(7)))} ${theme.fg("text", theme.bold(`(${usage.percent.toFixed(1).padStart(5)}%)`))}`;
-
-        const catDetailLines = categories.map(cat => {
-          const labelStr = cat.label.padEnd(14);
-          const valStr = formatTokens(cat.value).padStart(7);
-          const rowPercent = ((cat.value / limit) * 100).toFixed(1).padStart(5);
-          const icon = cat.label === "Available" ? "□" : "■";
-          return `${theme.fg(cat.color as any, icon)} ${theme.fg("text", labelStr)} ${theme.fg("accent", valStr)} (${rowPercent}%)`;
-        });
-
-        const allDetailLines = [totalUsageTitle, "", ...catDetailLines];
-
-        const leftSideWidth = 20;
-        const maxH = Math.max(gridLines.length, allDetailLines.length);
-        for (let i = 0; i < maxH; i++) {
-          const left = (gridLines[i] || "").padEnd(leftSideWidth);
-          const right = allDetailLines[i] || "";
-          container.addChild(new Text(`    ${left}      ${right}`, 1, 0));
-        }
-
-        container.addChild(new Spacer(1));
-        container.addChild(new Text(theme.fg("dim", " Press any key to close"), 1, 0));
-        container.addChild(new DynamicBorder((s: string) => theme.fg("accent", s)));
-
-        return {
-          render: (w) => container.render(w),
-          invalidate: () => container.invalidate(),
-          handleInput: (data) => done(undefined),
-        };
-      }, { overlay: true });
-    }
-  });
 }

package/src/utils.ts

@@ -0,0 +1,37 @@
+import {
+  type SessionEntry,
+  type SessionManager,
+} from "@mariozechner/pi-coding-agent";
+
+// Define missing types locally as they are not exported from the main entry point
+export interface SessionTreeNode {
+  entry: SessionEntry;
+  children: SessionTreeNode[];
+  label?: string;
+}
+
+export const isInternal = (name: string) => ["context_tag", "context_log", "context_checkout"].includes(name);
+
+export const resolveTargetId = (sm: SessionManager, target: string): string => {
+  if (target.toLowerCase() === "root") {
+    const tree = sm.getTree();
+    return tree.length > 0 ? tree[0].entry.id : target;
+  }
+  if (/^[0-9a-f]{8,}$/i.test(target)) return target;
+  const find = (nodes: SessionTreeNode[]): string | null => {
+    for (const n of nodes) {
+      if (sm.getLabel(n.entry.id) === target) return n.entry.id;
+      const r = find(n.children);
+      if (r) return r;
+    }
+    return null;
+  };
+  // sm.getTree() returns the SDK's SessionTreeNode[], which is structurally compatible
+  return find(sm.getTree()) || target;
+};
+
+export const formatTokens = (n: number) => {
+  if (n >= 1_000_000) return (n / 1_000_000).toFixed(1) + "M";
+  if (n >= 1_000) return Math.round(n / 1_000) + "k";
+  return n.toString();
+};

package/dist/index.js

@@ -1,443 +0,0 @@
-import { DynamicBorder, } from "@mariozechner/pi-coding-agent";
-import { Type } from "@sinclair/typebox";
-import { Container, Text, Spacer } from "@mariozechner/pi-tui";
-const ContextLogParams = Type.Object({
-    limit: Type.Optional(Type.Number({ description: "History limit for visible entries (default: 50)." })),
-    verbose: Type.Optional(Type.Boolean({ description: "If true, show ALL messages. If false (default), collapses intermediate AI steps and only shows 'milestones': User messages, Tags, Branch Points, and Summaries." })),
-});
-const ContextCheckoutParams = Type.Object({
-    target: Type.String({ description: "Where to jump/squash to. Can be a tag name (e.g., 'task-start'), a commit ID, or 'root'. This is the base for your new branch." }),
-    message: Type.String({ description: "The 'Carryover Message' for the new branch. A summary of your *current* progress/lessons that you want to bring with you to the new state. This ensures you don't lose key information when switching contexts. Good summary message: '[Status] + [Reason] + [Important Changes] + [Carryover Data]'" }),
-    backupTag: Type.Optional(Type.String({ description: "Optional tag name to apply to the CURRENT state before checking out. Use this to create an automatic backup of the history you are about to leave/squash." })),
-});
-const ContextTagParams = Type.Object({
-    name: Type.String({ description: "The tag/milestone name. Use meaningful names." }),
-    target: Type.Optional(Type.String({ description: "The commit ID to tag. Defaults to HEAD (current state)." })),
-});
-const isInternal = (name) => ["context_tag", "context_log", "context_checkout"].includes(name);
-const resolveTargetId = (sm, target) => {
-    if (target.toLowerCase() === "root") {
-        const tree = sm.getTree();
-        return tree.length > 0 ? tree[0].entry.id : target;
-    }
-    if (/^[0-9a-f]{8,}$/i.test(target))
-        return target;
-    const find = (nodes) => {
-        for (const n of nodes) {
-            if (sm.getLabel(n.entry.id) === target)
-                return n.entry.id;
-            const r = find(n.children);
-            if (r)
-                return r;
-        }
-        return null;
-    };
-    // sm.getTree() returns the SDK's SessionTreeNode[], which is structurally compatible
-    return find(sm.getTree()) || target;
-};
-const formatTokens = (n) => {
-    if (n >= 1_000_000)
-        return (n / 1_000_000).toFixed(1) + "M";
-    if (n >= 1_000)
-        return Math.round(n / 1_000) + "k";
-    return n.toString();
-};
-export default function (pi) {
-    pi.registerTool({
-        name: "context_tag",
-        label: "Context Tag",
-        description: "Creates a 'Save Point' (Bookmark) in the history. Use this before trying risky changes or when a feature is stable. 'Untagged progress is risky'.",
-        parameters: ContextTagParams,
-        async execute(_id, params, _signal, _onUpdate, ctx) {
-            const sm = ctx.sessionManager;
-            let id = params.target ? resolveTargetId(sm, params.target) : undefined;
-            if (!id) {
-                // Auto-resolve: Find the last "interesting" node to tag.
-                // We skip ToolResults (which look ugly tagged) and internal-only Assistant messages (which look empty).
-                const branch = sm.getBranch();
-                for (let i = branch.length - 1; i >= 0; i--) {
-                    const entry = branch[i];
-                    // 1. Check ToolResults
-                    if (entry.type === 'message' && entry.message.role === 'toolResult') {
-                        const tr = entry.message;
-                        if (isInternal(tr.toolName))
-                            continue;
-                        // Public tool result is a valid target
-                        id = entry.id;
-                        break;
-                    }
-                    // 2. Check Assistant messages for visibility
-                    if (entry.type === 'message' && entry.message.role === 'assistant') {
-                        const m = entry.message;
-                        let hasText = false;
-                        let hasPublicTools = false;
-                        const content = m.content;
-                        if (typeof content === 'string') {
-                            hasText = content.trim().length > 0;
-                        }
-                        else if (Array.isArray(content)) {
-                            hasText = content.some((c) => c.type === 'text' && typeof c.text === 'string' && c.text.trim().length > 0);
-                            const toolCalls = content.filter((c) => c.type === 'toolCall');
-                            hasPublicTools = toolCalls.some((tc) => !isInternal(tc.name));
-                        }
-                        // If it has visible text or public tools, it's a valid target
-                        if (hasText || hasPublicTools) {
-                            id = entry.id;
-                            break;
-                        }
-                        // Otherwise (only internal tools like context_tag), skip it
-                        continue;
-                    }
-                    // 3. User messages, Summaries, etc. are always valid targets
-                    id = entry.id;
-                    break;
-                }
-                // Fallback to leaf if search failed
-                if (!id)
-                    id = sm.getLeafId() ?? "";
-            }
-            pi.setLabel(id, params.name);
-            return { content: [{ type: "text", text: `Created tag '${params.name}' at ${id.slice(0, 7)}` }], details: {} };
-        },
-    });
-    pi.registerTool({
-        name: "context_log",
-        label: "Context Log",
-        description: "Show the entire history structure (status, message, tags, milestones). Analogous to 'git log --graph --oneline --decorate'",
-        parameters: ContextLogParams,
-        async execute(_id, params, _signal, _onUpdate, ctx) {
-            const sm = ctx.sessionManager;
-            const branch = sm.getBranch();
-            const currentLeafId = sm.getLeafId();
-            const verbose = params.verbose ?? false;
-            const limit = params.limit ?? 50;
-            const backboneIds = new Set(branch.map((e) => e.id));
-            const sequence = [];
-            branch.forEach((entry) => {
-                sequence.push(entry);
-                // Preserve side-summary logic: Show branch summaries/compactions that are off-path
-                const children = sm.getChildren(entry.id);
-                children.forEach((child) => {
-                    if ((child.type === "branch_summary" || child.type === "compaction") && !backboneIds.has(child.id)) {
-                        sequence.push(child);
-                    }
-                });
-            });
-            const getMsgContent = (entry) => {
-                if (entry.type === "branch_summary" || entry.type === "compaction") {
-                    const e = entry;
-                    return e.summary || "[No summary provided]";
-                }
-                if (entry.type === "label") {
-                    return `tag: ${entry.label}`;
-                }
-                if (entry.type === "message") {
-                    const msg = entry.message;
-                    if (msg.role === "toolResult") {
-                        const tr = msg;
-                        if (!verbose && isInternal(tr.toolName))
-                            return "";
-                        const extractText = (content) => {
-                            return content
-                                .map((p) => (p.type === "text" ? p.text : ""))
-                                .join(" ")
-                                .trim();
-                        };
-                        let resText = extractText(tr.content);
-                        const details = tr.details;
-                        if ((tr.toolName === "read" || tr.toolName === "edit") && details && "path" in details && typeof details.path === "string") {
-                            resText = `${details.path}: ${resText}`;
-                        }
-                        return `(${tr.toolName}) ${resText}`;
-                    }
-                    if (msg.role === "bashExecution") {
-                        return `[Bash] ${msg.command}`;
-                    }
-                    if (msg.role === "user" || msg.role === "assistant") {
-                        let text = "";
-                        if (typeof msg.content === "string") {
-                            text = msg.content;
-                        }
-                        else if (Array.isArray(msg.content)) {
-                            text = msg.content
-                                .map((p) => {
-                                if (typeof p === "object" && p !== null && "text" in p)
-                                    return p.text;
-                                return "";
-                            })
-                                .join(" ")
-                                .trim();
-                        }
-                        let toolCallsText = "";
-                        if (msg.role === "assistant") {
-                            const toolCalls = msg.content.filter((c) => c.type === "toolCall");
-                            toolCallsText = toolCalls
-                                .filter((tc) => verbose || !isInternal(tc.name))
-                                .map((tc) => `call: ${tc.name}(${JSON.stringify(tc.arguments)})`)
-                                .join("; ");
-                        }
-                        return [text, toolCallsText].filter(Boolean).join(" ");
-                    }
-                }
-                return "";
-            };
-            const isInteresting = (entry) => {
-                // 1. HEAD and Root
-                if (entry.id === currentLeafId)
-                    return true;
-                if (branch.length > 0 && entry.id === branch[0].id)
-                    return true;
-                // 2. Explicit Tags (Labels) - Only show the TAGGED node, not the label node itself
-                if (sm.getLabel(entry.id))
-                    return true;
-                if (entry.type === 'label')
-                    return false; // Hide label nodes, they are redundant
-                // 3. Structural Milestones (Summaries)
-                if (entry.type === 'branch_summary' || entry.type === 'compaction')
-                    return true;
-                // 4. Branch Points (Forks)
-                if (sm.getChildren(entry.id).length > 1)
-                    return true;
-                // 5. Natural Milestones (User Messages) - This is the key auto-tagging mechanism
-                if (entry.type === 'message' && entry.message.role === 'user')
-                    return true;
-                return false;
-            };
-            const visibleSequenceIds = new Set();
-            sequence.forEach(e => {
-                if (verbose || isInteresting(e)) {
-                    visibleSequenceIds.add(e.id);
-                }
-            });
-            let visibleEntries = sequence.filter(e => visibleSequenceIds.has(e.id));
-            if (visibleEntries.length > limit) {
-                const allowedIds = new Set(visibleEntries.slice(-limit).map(e => e.id));
-                visibleSequenceIds.clear();
-                allowedIds.forEach(id => visibleSequenceIds.add(id));
-            }
-            const lines = [];
-            let hiddenCount = 0;
-            sequence.forEach((entry) => {
-                if (!visibleSequenceIds.has(entry.id)) {
-                    hiddenCount++;
-                    return;
-                }
-                if (hiddenCount > 0) {
-                    lines.push(`  :  ... (${hiddenCount} hidden messages) ...`);
-                    hiddenCount = 0;
-                }
-                const isHead = entry.id === currentLeafId;
-                const label = sm.getLabel(entry.id);
-                const content = getMsgContent(entry).replace(/\s+/g, " ");
-                let role = entry.type.toUpperCase();
-                if (entry.type === "message") {
-                    const m = entry.message;
-                    role =
-                        m.role === "assistant"
-                            ? "AI"
-                            : m.role === "user"
-                                ? "USER"
-                                : m.role === "bashExecution"
-                                    ? "BASH"
-                                    : "TOOL";
-                }
-                else if (entry.type === "branch_summary" || entry.type === "compaction") {
-                    role = "SUMMARY";
-                }
-                const id = entry.id;
-                const isRoot = branch.length > 0 && entry.id === branch[0].id;
-                const meta = [isRoot ? "ROOT" : null, isHead ? "HEAD" : null, label ? `tag: ${label}` : null].filter(Boolean).join(", ");
-                const body = content.length > 100 ? content.slice(0, 100) + "..." : content;
-                const marker = isHead ? "*" : (role === "USER" ? "•" : "|");
-                lines.push(`${marker} ${id}${meta ? ` (${meta})` : ""} [${role}] ${body}`);
-            });
-            if (hiddenCount > 0) {
-                lines.push(`  :  ... (${hiddenCount} hidden messages) ...`);
-            }
-            // --- Context Dashboard (HUD) ---
-            const usage = await ctx.getContextUsage();
-            let usageStr = "Unknown";
-            if (usage) {
-                usageStr = `${usage.percent.toFixed(1)}% (${formatTokens(usage.tokens)}/${formatTokens(usage.contextWindow)})`;
-            }
-            // Find the distance to the nearest tag
-            let stepsSinceTag = 0;
-            let nearestTagName = "None";
-            for (let i = branch.length - 1; i >= 0; i--) {
-                const id = branch[i].id;
-                const label = sm.getLabel(id);
-                if (label) {
-                    nearestTagName = label;
-                    break;
-                }
-                stepsSinceTag++;
-            }
-            const hud = [
-                `[Context Dashboard]`,
-                `• Context Usage:    ${usageStr}`,
-                `• Segment Size:     ${stepsSinceTag} steps since last tag '${nearestTagName}'`,
-                `---------------------------------------------------`
-            ].join("\n");
-            return { content: [{ type: "text", text: hud + "\n" + (lines.join("\n") || "(Root Path Only)") }], details: {} };
-        },
-    });
-    pi.registerTool({
-        name: "context_checkout",
-        label: "Context Checkout",
-        description: "Navigate to ANY point in the conversation history. This checkout only resets *conversation history*, NOT disk files. ALWAYS provide a detailed 'message' to bridge context.",
-        parameters: ContextCheckoutParams,
-        async execute(_id, params, _signal, _onUpdate, ctx) {
-            const sm = ctx.sessionManager;
-            const tid = resolveTargetId(sm, params.target);
-            const currentLeaf = sm.getLeafId();
-            if (currentLeaf === tid) {
-                return { content: [{ type: "text", text: `Already at target ${tid}` }], details: {} };
-            }
-            if (params.backupTag && currentLeaf) {
-                pi.setLabel(currentLeaf, params.backupTag);
-            }
-            const currentLabel = currentLeaf ? sm.getLabel(currentLeaf) : undefined;
-            const origin = currentLabel ? `tag: ${currentLabel}` : (currentLeaf || "unknown");
-            const enrichedMessage = `(summary from ${origin})\n${params.message}`;
-            await sm.branchWithSummary(tid, enrichedMessage);
-            return { content: [{ type: "text", text: `Checked out ${tid}\nBackup tag created: ${params.backupTag || "none"}\nmessage: ${enrichedMessage}` }], details: {} };
-        },
-    });
-    pi.registerCommand("context", {
-        description: "Show context usage visualization",
-        handler: async (args, ctx) => {
-            const usage = await ctx.getContextUsage();
-            if (!usage) {
-                ctx.ui.notify("Context usage info not available.", "warning");
-                return;
-            }
-            const sm = ctx.sessionManager;
-            const branch = sm.getBranch();
-            const systemPrompt = ctx.getSystemPrompt();
-            const tools = pi.getActiveTools();
-            const allTools = pi.getAllTools();
-            const activeToolDefs = allTools.filter(t => tools.includes(t.name));
-            const estimateTokens = (text) => Math.ceil(text.length / 4);
-            let msgTokensRaw = 0;
-            let toolUseTokensRaw = 0;
-            let toolResultTokensRaw = 0;
-            for (const entry of branch) {
-                if (entry.type === "message") {
-                    const m = entry.message;
-                    if (m.role === "user") {
-                        if (typeof m.content === "string")
-                            msgTokensRaw += estimateTokens(m.content);
-                        else if (Array.isArray(m.content)) {
-                            for (const p of m.content)
-                                if (p.type === "text")
-                                    msgTokensRaw += estimateTokens(p.text);
-                        }
-                    }
-                    else if (m.role === "assistant") {
-                        if (typeof m.content === "string")
-                            msgTokensRaw += estimateTokens(m.content);
-                        else if (Array.isArray(m.content)) {
-                            for (const p of m.content) {
-                                if (p.type === "text")
-                                    msgTokensRaw += estimateTokens(p.text);
-                                if (p.type === "toolCall")
-                                    toolUseTokensRaw += estimateTokens(JSON.stringify(p));
-                            }
-                        }
-                    }
-                    else if (m.role === "toolResult") {
-                        if (Array.isArray(m.content)) {
-                            for (const p of m.content)
-                                if (p.type === "text")
-                                    toolResultTokensRaw += estimateTokens(p.text);
-                        }
-                    }
-                    else if (m.role === "bashExecution") {
-                        toolUseTokensRaw += estimateTokens(m.command || "");
-                    }
-                }
-                else if (entry.type === "branch_summary" || entry.type === "compaction") {
-                    msgTokensRaw += estimateTokens(entry.summary || "");
-                }
-            }
-            const systemTokensRaw = estimateTokens(systemPrompt);
-            const toolDefTokensRaw = estimateTokens(JSON.stringify(activeToolDefs));
-            const totalActual = usage.tokens;
-            const limit = usage.contextWindow;
-            const totalRaw = systemTokensRaw + toolDefTokensRaw + msgTokensRaw + toolUseTokensRaw + toolResultTokensRaw;
-            const ratio = totalRaw > 0 ? (totalActual / totalRaw) : 1;
-            const systemTokens = Math.round(systemTokensRaw * ratio);
-            const toolDefTokens = Math.round(toolDefTokensRaw * ratio);
-            const msgTokens = Math.round(msgTokensRaw * ratio);
-            const toolUseTokens = Math.round(toolUseTokensRaw * ratio);
-            const toolResultTokens = Math.round(toolResultTokensRaw * ratio);
-            await ctx.ui.custom((tui, theme, kb, done) => {
-                const container = new Container();
-                container.addChild(new DynamicBorder((s) => theme.fg("accent", s)));
-                container.addChild(new Text(theme.fg("accent", theme.bold(" Context Usage")), 1, 0));
-                container.addChild(new Spacer(1));
-                // Grouped by function and color
-                const categories = [
-                    { label: "System Prompt", value: systemTokens, color: "muted" },
-                    { label: "System Tools", value: toolDefTokens, color: "dim" },
-                    { label: "Tool Call", value: toolUseTokens + toolResultTokens, color: "success" },
-                    { label: "Messages", value: msgTokens, color: "accent" },
-                ];
-                const otherTokens = Math.max(0, totalActual - (systemTokens + toolDefTokens + msgTokens + toolUseTokens + toolResultTokens));
-                if (otherTokens > 10)
-                    categories.push({ label: "Other", value: otherTokens, color: "dim" });
-                categories.push({ label: "Available", value: Math.max(0, limit - totalActual), color: "borderMuted" });
-                const gridWidth = 10;
-                const gridHeight = 5;
-                const totalBlocks = gridWidth * gridHeight;
-                const blocks = [];
-                categories.forEach((cat) => {
-                    if (cat.label === "Available")
-                        return;
-                    let count = Math.round((cat.value / limit) * totalBlocks);
-                    if (count === 0 && cat.value > 0)
-                        count = 1;
-                    for (let i = 0; i < count && blocks.length < totalBlocks; i++) {
-                        blocks.push({ color: cat.color, filled: true });
-                    }
-                });
-                while (blocks.length < totalBlocks) {
-                    blocks.push({ color: "borderMuted", filled: false });
-                }
-                const gridLines = [];
-                for (let r = 0; r < gridHeight; r++) {
-                    let rowStr = "";
-                    for (let c = 0; c < gridWidth; c++) {
-                        const b = blocks[r * gridWidth + c];
-                        rowStr += theme.fg(b.color, b.filled ? "■ " : "□ ");
-                    }
-                    gridLines.push(rowStr.trimEnd());
-                }
-                const totalUsageTitle = `${theme.fg("text", theme.bold("Total Usage".padEnd(16)))} ${theme.fg("text", theme.bold(formatTokens(totalActual).padStart(7)))} ${theme.fg("text", theme.bold(`(${usage.percent.toFixed(1).padStart(5)}%)`))}`;
-                const catDetailLines = categories.map(cat => {
-                    const labelStr = cat.label.padEnd(14);
-                    const valStr = formatTokens(cat.value).padStart(7);
-                    const rowPercent = ((cat.value / limit) * 100).toFixed(1).padStart(5);
-                    const icon = cat.label === "Available" ? "□" : "■";
-                    return `${theme.fg(cat.color, icon)} ${theme.fg("text", labelStr)} ${theme.fg("accent", valStr)} (${rowPercent}%)`;
-                });
-                const allDetailLines = [totalUsageTitle, "", ...catDetailLines];
-                const leftSideWidth = 20;
-                const maxH = Math.max(gridLines.length, allDetailLines.length);
-                for (let i = 0; i < maxH; i++) {
-                    const left = (gridLines[i] || "").padEnd(leftSideWidth);
-                    const right = allDetailLines[i] || "";
-                    container.addChild(new Text(`    ${left}      ${right}`, 1, 0));
-                }
-                container.addChild(new Spacer(1));
-                container.addChild(new Text(theme.fg("dim", " Press any key to close"), 1, 0));
-                container.addChild(new DynamicBorder((s) => theme.fg("accent", s)));
-                return {
-                    render: (w) => container.render(w),
-                    invalidate: () => container.invalidate(),
-                    handleInput: (data) => done(undefined),
-                };
-            }, { overlay: true });
-        }
-    });
-}