remath proportional bargraph display

Author: spoons-and-mirrors

lib/tools/compress.ts

@@ -406,6 +406,12 @@ export function createCompressTool(ctx: ToolContext): ReturnType<typeof tool> {
                     compressedToolIds: compressedToolIds.length,
                 })
 
+                // Build token weight map for all messages (for proportional bar graph)
+                const weights = new Map<string, number>()
+                for (const msg of messages) {
+                    weights.set(msg.info.id, countAllMessageTokens(msg))
+                }
+
                 await sendCompressNotification(
                     client,
                     logger,
@@ -420,6 +426,7 @@ export function createCompressTool(ctx: ToolContext): ReturnType<typeof tool> {
                     totalSessionTokens,
                     estimatedCompressedTokens,
                     messages.map((m) => m.info.id),
+                    weights,
                     messages.length,
                     currentParams,
                 )

lib/ui/notification.ts

@@ -142,6 +142,7 @@ export async function sendCompressNotification(
     totalSessionTokens: number,
     compressedTokens: number,
     sessionMessageIds: string[],
+    weights: Map<string, number>,
     totalMessages: number,
     params: any,
 ): Promise<boolean> {
@@ -168,7 +169,13 @@ export async function sendCompressNotification(
         })
 
         const newIds = new Set(messageIds)
-        const progressBar = formatSessionMap(sessionMessageIds, state.prune.messages, newIds, 50)
+        const progressBar = formatSessionMap(
+            sessionMessageIds,
+            state.prune.messages,
+            newIds,
+            weights,
+            50,
+        )
         const reduction =
             totalSessionTokens > 0 ? Math.round((compressedTokens / totalSessionTokens) * 100) : 0
         message += `\n\n${progressBar}`

lib/ui/utils.ts

@@ -63,17 +63,25 @@ export function formatSessionMap(
     messageIds: string[],
     prunedMessages: Map<string, number>,
     newPrunedIds: Set<string>,
+    weights: Map<string, number>,
     width: number = 50,
 ): string {
     const total = messageIds.length
     if (total === 0) return `│${"░".repeat(width)}│`
 
+    // Build cumulative token weights for proportional positioning
+    const cum = [0]
+    for (const id of messageIds) {
+        cum.push(cum[cum.length - 1] + (weights.get(id) || 1))
+    }
+    const totalWeight = cum[cum.length - 1]
+
     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)
+            const start = Math.floor((cum[m] / totalWeight) * width)
+            const end = Math.floor((cum[m + 1] / totalWeight) * width)
             const char = newPrunedIds.has(messageIds[m]) ? "▓" : "░"
             for (let i = start; i < end; i++) {
                 bar[i] = char