bargraph v2

Author: spoons-and-mirrors

lib/tools/compress.ts

@@ -362,6 +362,11 @@ export function createCompressTool(ctx: ToolContext): ReturnType<typeof tool> {
                     }
                 }
 
+                // Cap estimate — countAllMessageTokens can inflate beyond API count
+                if (totalSessionTokens > 0 && estimatedCompressedTokens > totalSessionTokens) {
+                    estimatedCompressedTokens = Math.round(totalSessionTokens * 0.95)
+                }
+
                 clog.info(C.COMPRESS, `Token Accounting`, {
                     totalSessionTokens,
                     estimatedCompressedTokens,
@@ -414,8 +419,7 @@ export function createCompressTool(ctx: ToolContext): ReturnType<typeof tool> {
                     summaryTokens,
                     totalSessionTokens,
                     estimatedCompressedTokens,
-                    rawStartResult,
-                    rawEndResult,
+                    messages.map((m) => m.info.id),
                     messages.length,
                     currentParams,
                 )

lib/ui/notification.ts

@@ -4,7 +4,7 @@ import {
     formatPrunedItemsList,
     formatStatsHeader,
     formatTokenCount,
-    formatCompressBar,
+    formatSessionMap,
 } from "./utils"
 import { ToolParameterEntry } from "../state"
 import { PluginConfig } from "../config"
@@ -141,8 +141,7 @@ export async function sendCompressNotification(
     summaryTokens: number,
     totalSessionTokens: number,
     compressedTokens: number,
-    startResult: any,
-    endResult: any,
+    sessionMessageIds: string[],
     totalMessages: number,
     params: any,
 ): Promise<boolean> {
@@ -168,10 +167,11 @@ export async function sendCompressNotification(
                 totalSessionTokens > 0 ? (compressedTokens / totalSessionTokens).toFixed(4) : "N/A",
         })
 
-        const progressBar = formatCompressBar(totalSessionTokens, compressedTokens, 25)
+        const progressBar = formatSessionMap(sessionMessageIds, state.prune.messages, 50)
         const reduction =
             totalSessionTokens > 0 ? Math.round((compressedTokens / totalSessionTokens) * 100) : 0
-        message += `\n\n▣ Compressing (${pruneTokenCounterStr} removed, ${reduction}% reduction) ${progressBar}`
+        message += `\n\n▣ Compressing (${pruneTokenCounterStr} removed, ${reduction}% reduction)`
+        message += `\n${progressBar}`
         message += `\n→ Topic: ${topic}`
         message += `\n→ Items: ${messageIds.length} messages`
         if (toolIds.length > 0) {

lib/ui/utils.ts

@@ -59,32 +59,27 @@ export function formatProgressBar(
     return `│${bar}│`
 }
 
-export function formatCompressBar(
-    sessionContext: number,
-    compressedAmount: number,
-    width: number = 20,
+export function formatSessionMap(
+    messageIds: string[],
+    prunedMessages: Map<string, number>,
+    width: number = 50,
 ): string {
-    if (sessionContext <= 0) return `│${"░".repeat(width)}│ 0%`
-
-    // sessionContext already includes compressedAmount (it's the full session total)
-    const remainingRatio = (sessionContext - compressedAmount) / sessionContext
-    const remainingWidth = Math.round(remainingRatio * width)
-    const compressedWidth = width - remainingWidth
-    const reductionPercent = -Math.round((compressedAmount / sessionContext) * 100)
-
-    clog.info(C.COMPRESS, `formatCompressBar`, {
-        sessionContext,
-        compressedAmount,
-        remainingRatio: remainingRatio.toFixed(4),
-        remainingWidth,
-        compressedWidth,
-        reductionPercent,
-    })
-
-    const remainingBar = "█".repeat(Math.max(0, remainingWidth))
-    const compressedBar = "░".repeat(Math.max(0, compressedWidth))
-
-    return `│${remainingBar}${compressedBar}│`
+    const total = messageIds.length
+    if (total === 0) return `│${"░".repeat(width)}│`
+
+    const bar = new Array(width).fill("█")
+
+    for (let m = 0; m < total; m++) {
+        if (prunedMessages.has(messageIds[m])) {
+            const start = Math.floor((m / total) * width)
+            const end = Math.floor(((m + 1) / total) * width)
+            for (let i = start; i < end; i++) {
+                bar[i] = "░"
+            }
+        }
+    }
+
+    return `│${bar.join("")}│`
 }
 
 export function shortenPath(input: string, workingDirectory?: string): string {