@gabrihhh/jarvis 2.6.0 → 2.6.2

package/README.md

@@ -31,9 +31,13 @@ Reinicie o Claude Code após o setup.
 | `jarvis --trigger prompt` | Hook de memória roda a cada prompt |
 | `jarvis --trigger off` | Desativa o carregamento automático de memória |
 | `jarvis --theme` | Mostra o tema atual da status bar |
-| `jarvis --theme <name>:<#hex>` | Define a cor de um box (`context`, `trigger`, `memory`) |
+| `jarvis --theme <name>:<#hex>` | Define a cor de um box (`context`, `trigger`, `memory`, `tokens`) |
 | `jarvis --theme <name>:reset` | Reseta a cor de um box para o padrão |
 | `jarvis --theme reset` | Reseta todas as cores para o padrão |
+| `jarvis --token` | Mostra o modo de exibição de tokens atual |
+| `jarvis --token on` | Ativa box com total de tokens do último turno (`◈`) |
+| `jarvis --token complete` | Ativa box `◈` + linha de breakdown abaixo da status bar |
+| `jarvis --token off` | Desativa exibição de tokens |
 | `jarvis --line` | Saída de uma linha usada internamente pela status bar |
 | `jarvis --help` | Lista todos os comandos |
 
@@ -79,3 +83,18 @@ Reinicie o Claude Code após o setup.
 │ CONTEXT ████░░░░ 52% │
 ╰──────────────────────╯
 ```
+
+**Status bar com `--token on`** (box com total de tokens do último turno):
+```
+╭──────────────────────╮╭──────────╮
+│ CONTEXT ████░░░░ 52% ││ ◈ 50.0K  │
+╰──────────────────────╯╰──────────╯
+```
+
+**Status bar com `--token complete`** (box + breakdown detalhado):
+```
+╭──────────────────────╮╭──────────╮
+│ CONTEXT ████░░░░ 52% ││ ◈ 50.0K  │
+╰──────────────────────╯╰──────────╯
+INPUT 800 │ HISTORY 45.0K │ CACHE 1.2K │ RESPONSE 3.2K
+```

package/bin/jarvis.js

@@ -78,9 +78,13 @@ if (args.includes('--help') || args.includes('-h')) {
     jarvis --trigger prompt      Hook runs on every prompt
     jarvis --trigger off         Disable automatic memory loading
     jarvis --theme               Show current statusline theme
-    jarvis --theme <name>:<hex>  Set a box color (context, trigger, memory)
+    jarvis --theme <name>:<hex>  Set a box color (context, trigger, memory, tokens)
     jarvis --theme <name>:reset  Reset a single box to default color
     jarvis --theme reset         Reset all colors to default
+    jarvis --token               Show current token display mode
+    jarvis --token on            Show total tokens box (◈)
+    jarvis --token complete      Show token box + INPUT/HISTORY/CACHE/RESPONSE breakdown
+    jarvis --token off           Disable token display
     jarvis --help                Show this help
 
   Slash commands (inside Claude Code):
@@ -253,6 +257,25 @@ if (args.includes('--graph')) {
   writeTheme(theme);
   console.log(`\n  ✓ ${name} set to ${chalk.hex(color).bold(color)}\n`);
   process.exit(0);
+} else if (args.includes('--token')) {
+  const value = args[args.indexOf('--token') + 1];
+  const validValues = ['on', 'off', 'complete'];
+
+  if (!value || !validValues.includes(value)) {
+    const cfg = loadMemoryConfig();
+    const current = cfg.tokenDisplay || 'off';
+    console.log(`\n  Token display: ${current}\n`);
+    console.log(`  Usage: jarvis --token <on|complete|off>\n`);
+    process.exit(0);
+  }
+
+  const cfg = loadMemoryConfig();
+  cfg.tokenDisplay = value === 'on' ? 'simple' : value;
+  saveMemoryConfig(cfg);
+
+  const labels = { simple: 'on (◈ total tokens)', off: 'off', complete: 'complete (breakdown)' };
+  console.log(`\n  ✓ Token display set to: ${labels[cfg.tokenDisplay]}\n`);
+  process.exit(0);
 } else if (args.includes('--query')) {
   try {
     const { queryByPath } = await import('../src/memory/query-by-path.js');

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gabrihhh/jarvis",
-  "version": "2.6.0",
+  "version": "2.6.2",
   "description": "Claude Code terminal dashboard + semantic memory graph via Neo4j",
   "bin": {
     "jarvis": "bin/jarvis.js",

package/src/calculator.js

@@ -109,6 +109,18 @@ export function formatTokens(n) {
   return `${n}`;
 }
 
+export function getLastTurnTokens(entries) {
+  if (!entries.length) return null;
+  const last = entries[entries.length - 1];
+  return {
+    input:    last.inputTokens,
+    history:  last.cacheReadTokens,
+    cache:    last.cacheWriteTokens,
+    response: last.outputTokens,
+    total:    last.inputTokens + last.outputTokens + last.cacheReadTokens + last.cacheWriteTokens,
+  };
+}
+
 export function formatCost(n) {
   if (n >= 1) return `$${n.toFixed(2)}`;
   if (n >= 0.01) return `$${n.toFixed(3)}`;

package/src/reader.js

@@ -135,29 +135,7 @@ export function getCurrentSessionFile() {
       pid = getParentPid(pid);
     }
 
-    // Fallback: find a session whose sessionId appears in JSONL files
-    const sessionFiles = readdirSync(sessionsDir)
-      .map(f => {
-        try { return JSON.parse(readFileSync(join(sessionsDir, f), 'utf-8')); }
-        catch { return null; }
-      })
-      .filter(Boolean);
-
-    if (!sessionFiles.length) return null;
-
-    const allFiles = getAllSessionFiles();
-    const sessionIdSet = new Set();
-    for (const file of allFiles) {
-      try {
-        const lines = readFileSync(file, 'utf-8').split('\n').filter(l => l.trim());
-        for (const line of lines.slice(-5)) {
-          try { const obj = JSON.parse(line); if (obj.sessionId) sessionIdSet.add(obj.sessionId); }
-          catch { /* skip */ }
-        }
-      } catch { /* skip */ }
-    }
-
-    return sessionFiles.find(s => sessionIdSet.has(s.sessionId)) || sessionFiles[0];
+    return null;
   } catch {
     return null;
   }

package/src/statusline.js

@@ -3,7 +3,7 @@ import { readFileSync, existsSync, unlinkSync } from 'fs';
 import { join } from 'path';
 import { homedir, tmpdir } from 'os';
 import { readAllUsage, getCurrentSessionFile, readCurrentSessionUsage } from './reader.js';
-import { aggregateStats, aggregateSession } from './calculator.js';
+import { aggregateStats, aggregateSession, getLastTurnTokens, formatTokens } from './calculator.js';
 import { readTheme } from './theme.js';
 
 const chalk = new Chalk({ level: 3 });
@@ -21,6 +21,13 @@ function readTriggerMode() {
   } catch { return 'off'; }
 }
 
+function readTokenMode() {
+  try {
+    const cfg = JSON.parse(readFileSync(join(homedir(), '.claude-memory.json'), 'utf8'));
+    return cfg.tokenDisplay || 'off';
+  } catch { return 'off'; }
+}
+
 function buildBox(inner, color, width = inner.length) {
   return [
     chalk.hex(color).bold(`╭${'─'.repeat(width)}╮`),
@@ -54,6 +61,7 @@ function isMemoryLoaded(sessionId) {
 
 export function renderLine() {
   const mode = readTriggerMode();
+  const tokenMode = readTokenMode();
   const theme = readTheme();
 
   const sessionMeta = getCurrentSessionFile();
@@ -63,28 +71,44 @@ export function renderLine() {
 
   const allEntries = readAllUsage();
 
-  const boxes = (contextBox) => {
+  const buildOutput = (contextBox, turnTokens) => {
     const toJoin = [contextBox];
     if (mode !== 'off') toJoin.push(buildBox(` TRIGGER ${mode.toUpperCase()} `, theme.trigger));
     if (loadedBox) toJoin.push(loadedBox);
-    return joinBoxes(...toJoin);
+    if (turnTokens) toJoin.push(buildBox(` ◈ ${formatTokens(turnTokens.total)} `, theme.tokens));
+
+    let out = joinBoxes(...toJoin);
+
+    if (tokenMode === 'complete' && turnTokens) {
+      const col = (s) => chalk.hex(theme.tokens).bold(s);
+      const parts = [
+        `INPUT ${formatTokens(turnTokens.input)}`,
+        `HISTORY ${formatTokens(turnTokens.history)}`,
+        `CACHE ${formatTokens(turnTokens.cache)}`,
+        `RESPONSE ${formatTokens(turnTokens.response)}`,
+      ];
+      out += '\n' + parts.map(col).join(col(' │ '));
+    }
+
+    return out;
   };
 
   if (!allEntries.length) {
     const contextBox = buildBox(` CONTEXT ${'░'.repeat(8)} 0% `, theme.context);
-    process.stdout.write(boxes(contextBox));
+    process.stdout.write(buildOutput(contextBox, null));
     return;
   }
 
   const sessionEntries = sessionId ? readCurrentSessionUsage(sessionId) : [];
   const session = aggregateSession(sessionEntries);
+  const turnTokens = tokenMode !== 'off' ? getLastTurnTokens(sessionEntries) : null;
 
   if (!session) {
     const contextBox = buildBox(` CONTEXT ${'░'.repeat(8)} 0% `, theme.context);
-    process.stdout.write(boxes(contextBox));
+    process.stdout.write(buildOutput(contextBox, turnTokens));
     return;
   }
 
   const contextBox = buildBox(` CONTEXT ${bar(session.percent)} ${session.percent}% `, theme.context);
-  process.stdout.write(boxes(contextBox));
+  process.stdout.write(buildOutput(contextBox, turnTokens));
 }

package/src/theme.js

@@ -8,6 +8,7 @@ export const DEFAULT_COLORS = {
   context: '#22d3ee',
   trigger: '#f472b6',
   memory:  '#4ade80',
+  tokens:  '#ef4444',
 };
 
 export const VALID_NAMES = Object.keys(DEFAULT_COLORS);