@mohndoe/pi-atlas 0.1.1

package/src/tabs/__tests__/Usage.test.ts

@@ -0,0 +1,174 @@
+import { afterEach, beforeEach, describe, expect, it, vi } from "bun:test";
+import { makeMockTUI, makeTheme } from "../../__tests__/components.fixtures";
+import { type ToolStat } from "../../types";
+import { Usage } from "../Usage";
+
+describe("Usage", () => {
+  const mockTui = makeMockTUI();
+
+  const tokenUsage = {
+    total: 10000,
+    input: 5000,
+    output: 4000,
+    cacheRead: 500,
+    cacheWrite: 500,
+  };
+
+  const tools: ToolStat[] = [
+    { name: "bash", count: 150 },
+    { name: "read", count: 120 },
+    { name: "edit", count: 45 },
+  ];
+
+  it("renders token section with StatCards", () => {
+    const tab = new Usage(tools, tokenUsage, makeTheme(), mockTui, 10);
+    const text = tab.render(80).join("\n");
+
+    expect(text).toContain("Tokens");
+    expect(text).toContain("10.0k");
+    expect(text).toContain("Input");
+    expect(text).toContain("5.0k");
+    expect(text).toContain("Output");
+    expect(text).toContain("4.0k");
+    expect(text).toContain("Cache Read");
+    expect(text).toContain("500");
+    expect(text).toContain("Cache Write");
+  });
+
+  it("renders tool table with formatted data", () => {
+    const tab = new Usage(tools, tokenUsage, makeTheme(), mockTui, 10);
+    const lines = tab.render(80).slice(4);
+
+    const text = lines.join("\n");
+
+    expect(lines[0]).toContain("Tools");
+    expect(lines[0]).toContain(tools.length.toString());
+
+    // Headers
+    expect(text).toContain("Command");
+    expect(text).toContain("Calls");
+    expect(text).toContain("Share %");
+
+    // Tool names
+    expect(text).toContain("bash");
+    expect(text).toContain("read");
+    expect(text).toContain("edit");
+
+    // Call counts
+    expect(text).toContain("150");
+    expect(text).toContain("120");
+    expect(text).toContain("45");
+  });
+
+  it("does NOT show 'Tool Calls' title", () => {
+    const tab = new Usage(tools, tokenUsage, makeTheme(), mockTui, 10);
+    const text = tab.render(80).join("\n");
+
+    expect(text).not.toContain("Tool Calls");
+  });
+
+  it("shows empty state when tools is empty", () => {
+    const tab = new Usage([], tokenUsage, makeTheme(), mockTui, 10);
+    const lines = tab.render(80).slice(4);
+    const text = lines.join("\n");
+
+    expect(lines[0]).toContain("Tools");
+    // don't display 0 counter
+    expect(lines[0]).not.toContain("0");
+    expect(text).toContain("No tools data for this time range");
+  });
+
+  it("renders tool names with control characters as single lines", () => {
+    const dirtyTools: ToolStat[] = [
+      { name: "ls -la agent/\n</parameter", count: 2 },
+      { name: "bash", count: 100 },
+    ];
+    const tab = new Usage(dirtyTools, tokenUsage, makeTheme(), mockTui, 10);
+    const lines = tab.render(80);
+    // Every line should be a single-line render — no extra lines from \n
+    const text = lines.join("\n");
+    // Tool column is ~20 chars wide — the \n is stripped, leaving truncated "ls -la agent/</param…"
+    expect(text).not.toContain("agent/\n");
+    expect(text).toContain("bash");
+    // Verify no broken rows (each rendered line is a single table row)
+    for (const line of lines) {
+      // Each line should be a continuous visible string without embedded newlines
+      expect(line).not.toContain("\n");
+    }
+  });
+
+  it("renders within width", () => {
+    const tab = new Usage(tools, tokenUsage, makeTheme(), mockTui, 10);
+    const lines = tab.render(50);
+    for (const line of lines) {
+      const visLen = line.replace(/\x1b\[[0-9;]*m/g, "").length;
+      expect(visLen).toBeLessThanOrEqual(50);
+    }
+  });
+
+  it("shows sort indicator on Calls column", () => {
+    const tab = new Usage(tools, tokenUsage, makeTheme(), mockTui, 10);
+    const lines = tab.render(80);
+    const text = lines.join("\n");
+    expect(text).toContain("Calls ▼");
+  });
+
+  it("invalidates render cache", () => {
+    const tab = new Usage(tools, tokenUsage, makeTheme(), mockTui, 10);
+    tab.render(80);
+    tab.invalidate();
+    const lines = tab.render(60);
+    for (const line of lines) {
+      const visLen = line.replace(/\x1b\[[0-9;]*m/g, "").length;
+      expect(visLen).toBeLessThanOrEqual(60);
+    }
+  });
+
+  it("supports re-render after invalidation (lifecycle path)", () => {
+    const tab = new Usage(tools, tokenUsage, makeTheme(), mockTui, 10);
+
+    const lines1 = tab.render(80);
+    expect(lines1.join("\n")).toContain("bash");
+
+    tab.invalidate();
+
+    const lines2 = tab.render(80);
+    const text = lines2.join("\n");
+    expect(text).toContain("bash");
+    expect(text).toContain("Command");
+    expect(text).toContain("Calls ▼");
+    // Token section still intact
+    expect(text).toContain("5.0k");
+    expect(text).toContain("4.0k");
+
+    for (const line of lines2) {
+      const visLen = line.replace(/\x1b\[[0-9;]*m/g, "").length;
+      expect(visLen).toBeLessThanOrEqual(80);
+    }
+  });
+
+  describe("marquee lifecycle", () => {
+    beforeEach(() => {
+      vi.useFakeTimers();
+    });
+
+    afterEach(() => {
+      vi.useRealTimers();
+    });
+
+    it("clears marquee timers on invalidate", () => {
+      const longTool: ToolStat[] = [{ name: "a-very-long-tool-name-x", count: 150 }];
+      const tab = new Usage(longTool, tokenUsage, makeTheme(), mockTui, 10);
+
+      tab.render(30);
+      expect(vi.getTimerCount()).toBe(1);
+
+      tab.invalidate();
+      expect(vi.getTimerCount()).toBe(0);
+
+      const lines = tab.render(80);
+      const text = lines.join("\n");
+      expect(text).toContain("a-very-long-tool");
+    });
+  });
+});

package/src/types.ts

@@ -0,0 +1,99 @@
+export interface DayAgg {
+  date: string; // "YYYY-MM-DD"
+  cost: number;
+  hourCost: Record<number, number>; // accumulated cost per UTC hour 0-23
+  inTok: number;
+  outTok: number;
+  crTok: number;
+  cwTok: number;
+  userMsgs: number;
+  asstMsgs: number;
+  toolResults: number;
+  sessionIds: Set<string>;
+  langLines: Record<string, number>;
+  langEdits: Record<string, number>;
+  modelCost: Record<string, number>;
+  modelCount: Record<string, number>;
+  providerCost: Record<string, number>;
+  providerCount: Record<string, number>;
+  modelToProvider: Map<string, string>;
+  projectCost: Record<string, number>;
+  projectSessions: Record<string, Set<string>>;
+  toolCount: Record<string, number>;
+  // New fields tracking pi session entry types beyond session+message
+  compactionCount: number;
+  compactedTokens: number;
+  modelChanges: number;
+  thinkingLevelCount: Record<string, number>;
+}
+
+export type TimeRange = "1d" | "7d" | "30d" | "All";
+
+export interface DaySpend {
+  date: string; // "YYYY-MM-DD"
+  cost: number;
+}
+
+export interface HourSpend {
+  hour: number; // 0-23
+  cost: number;
+}
+
+export interface LangStat {
+  language: string;
+  lines: number;
+  edits: number;
+}
+
+export interface ModelStat {
+  provider?: string;
+  model: string;
+  cost: number;
+  calls: number;
+}
+
+export interface ProjectStat {
+  project: string;
+  cost: number;
+  sessions: number;
+}
+
+export interface ToolStat {
+  name: string;
+  count: number;
+}
+
+export interface StatsSummary {
+  totalCost: number;
+  sessionCount: number;
+  totalMessages: number;
+  totalTokens: number;
+  totalInputTokens: number;
+  totalOutputTokens: number;
+  totalCacheReadTokens: number;
+  totalCacheWriteTokens: number;
+  daysActive: number;
+  avgCostPerDay: number;
+  todayCost: number;
+  languages: LangStat[];
+  models: ModelStat[];
+  projects: ProjectStat[];
+  tools: ToolStat[];
+  dailySpend: DaySpend[];
+  hourlySpend: HourSpend[];
+}
+
+export interface CachePayload {
+  signature: string;
+  generatedAt: string;
+  days: SerializedDayAgg[];
+}
+
+export interface SerializedDayAgg extends Omit<
+  DayAgg,
+  "sessionIds" | "projectSessions" | "modelToProvider"
+> {
+  sessionIds: string[];
+  projectSessions: Record<string, string[]>;
+  modelToProvider: Record<string, string>;
+}

package/tsconfig.json

@@ -0,0 +1,30 @@
+{
+  "compilerOptions": {
+    // Environment setup & latest features
+    "lib": ["ESNext"],
+    "target": "ESNext",
+    "module": "Preserve",
+    "moduleDetection": "force",
+    "jsx": "react-jsx",
+    "allowJs": true,
+    "types": ["bun"],
+
+    // Bundler mode
+    "moduleResolution": "bundler",
+    "allowImportingTsExtensions": true,
+    "verbatimModuleSyntax": true,
+    "noEmit": true,
+
+    // Best practices
+    "strict": true,
+    "skipLibCheck": true,
+    "noFallthroughCasesInSwitch": true,
+    "noUncheckedIndexedAccess": true,
+    "noImplicitOverride": true,
+
+    // Some stricter flags (disabled by default)
+    "noUnusedLocals": false,
+    "noUnusedParameters": false,
+    "noPropertyAccessFromIndexSignature": false
+  }
+}