Closes #278. Adds Charmbracelet Crush as a lazy-loaded provider: - src/providers/crush.ts: walks ~/.local/share/crush/projects.json (XDG_DATA_HOME and CRUSH_GLOBAL_DATA aware), opens each project's crush.db read-only, queries root sessions where parent_session_id IS NULL. Emits one ParsedProviderCall per session with real prompt_tokens, completion_tokens, cost (dollars), and the dominant model resolved from messages.model. - src/providers/index.ts: register crush alongside cursor, goose, opencode, antigravity, cursor-agent in the lazy import path. - tests/providers/crush.test.ts: 10 fixture-based tests covering discovery, parsing, missing-registry, malformed JSON, missing db, child session exclusion, dominant model selection, dedup, and array-shaped legacy registry. Schema source: charmbracelet/crush@v0.66.1 internal/db/migrations/20250424200609_initial.sql, verified by spawning a research agent against upstream. The schema *comments* in that migration claim millisecond timestamps but every actual INSERT/UPDATE uses strftime('%s', 'now') which returns Unix seconds; the parser treats values as seconds. Tokscale's parser (junhoyeo/tokscale#346) gets this wrong and is off by 1000x, plus its parser misses the prompt_tokens/completion_tokens columns that exist in Crush's schema. Our integration uses both, so Crush sessions get real per-model attribution. Menubar: - mac/Sources/CodeBurnMenubar/AppStore.swift: add .crush case to ProviderFilter and its cliArg switch. - mac/Sources/CodeBurnMenubar/Views/AgentTabStrip.swift: add Crush color to the per-tab color extension. The visibleFilters computed property already filters by detected providers, so the Crush tab appears automatically when a user has Crush data. README: - Replace the provider table with an icon-led layout. Icons live under assets/providers/<name>.<ext>. 14 icons sourced from junhoyeo/tokscale (MIT) under nominative fair use, 4 sourced separately: codex (OpenAI org avatar), cursor-agent (reuses the Cursor icon), kiro (kiro.dev favicon, ico->png via sips), omp (can1357/oh-my-pi icon.svg, MIT). Attribution line added. - Add Crush row. Docs: - docs/providers/crush.md: full per-provider doc with verified schema excerpt, the seconds-vs-milliseconds quirk, and a "when fixing a bug here" checklist. - docs/architecture.md: provider count 17 -> 18, test count 41 -> 42, and crush in the lazy list. - docs/providers/README.md: add Crush row to the lazy index. - CONTRIBUTING.md: bump test count to 568 (was 558). All 568 tests pass locally; swift build clean.
9.8 KiB
CodeBurn Architecture
A map of the codebase. Read this once before opening a non-trivial PR.
Three Surfaces
CodeBurn is one Node.js CLI plus two GUI clients that shell out to it.
+----------------------+ +-----------------+
| mac/ (Swift) | ---> | |
+----------------------+ | src/cli.ts |
| gnome/ (JavaScript) | ---> | (the CLI) |
+----------------------+ | |
| status |
| --format |
| menubar-json |
+-----------------+
|
v
+----------------------------+
| session files on disk |
| (JSONL, SQLite, protobuf) |
+----------------------------+
The macOS menubar (mac/) and the GNOME extension (gnome/) both invoke codeburn status --format menubar-json --period <p> and parse the JSON. They do not share code with the CLI; they only depend on its output contract.
CLI (src/)
src/cli.ts is the Commander.js entry point. The bin field in package.json points at dist/cli.js. Twelve commands are registered:
| Command | Line | Purpose |
|---|---|---|
report |
274 | Default. Interactive Ink TUI dashboard. |
status |
358 | Compact text status, plus --format menubar-json for clients. |
today |
524 | Today-only view of report. |
month |
542 | Month-only view of report. |
export |
560 | CSV or JSON dump of usage data. |
menubar |
621 | Downloads and launches the macOS menubar bundle. |
currency |
636 | Sets display currency. |
model-alias |
687 | Maps an unknown model name to a known one for pricing. |
plan |
737 | Configures a subscription plan for overage tracking. |
optimize |
857 | Runs all 14 waste detectors. |
compare |
870 | Compares two models side by side. |
yield |
882 | Tracks which sessions shipped to main vs. were reverted (experimental). |
Pipeline
provider.discoverSessions()
|
v
provider.createSessionParser(source, seenKeys)
|
v yields ParsedProviderCall (see src/providers/types.ts)
|
v
src/parser.ts: parseAllSessions()
|
v aggregates into ProjectSummary[]
|
v
src/daily-cache.ts: aggregate per day, persist
|
v
output formatter (Ink TUI, JSON, or menubar-json)
src/parser.ts is the central aggregator. Public exports: parseAllSessions, filterProjectsByName, extractMcpInventory. It owns the dedup Set (seenKeys) that is passed into every provider parser so a turn that surfaces in two providers (Claude logs vs. Cursor mirror, for instance) is counted once.
Cache Layers
Three caches under ~/.cache/codeburn/ (override with CODEBURN_CACHE_DIR):
| File | Owner | Invalidation |
|---|---|---|
codex-results.json |
src/codex-cache.ts |
mtimeMs + sizeBytes per Codex .jsonl. |
cursor-results.json |
src/cursor-cache.ts |
mtimeMs + sizeBytes of the Cursor SQLite db. |
daily-cache.json |
src/daily-cache.ts |
Tracks lastComputedDate; new days are backfilled, old days are reused. |
All three use atomic write (temp file + rename) and write with mode 0o600. All three carry a numeric version field; bumping it forces a recompute next run.
Optimize Detectors
src/optimize.ts exports 14 detectors. Each returns a WasteFinding | null. They are composed by runOptimize() which collects findings, ranks them by impact, and returns them with WasteAction objects (paste-to-CLAUDE.md, paste-to-session-opener, prompt-now, edit shell config).
| Detector | Line | What it catches |
|---|---|---|
detectJunkReads |
428 | Reads into node_modules, .git, dist, etc. |
detectDuplicateReads |
477 | Re-reads of the same file in a session. |
detectMcpToolCoverage |
795 | MCP servers with many tools but low usage. |
detectUnusedMcp |
855 | MCP servers configured but never invoked. |
detectBloatedClaudeMd |
944 | CLAUDE.md files past a healthy size. |
detectLowReadEditRatio |
987 | Edit-heavy sessions with too few prior reads. |
detectCacheBloat |
1048 | High cache_creation_input_tokens. |
detectGhostAgents |
1124 | Defined but never-invoked Claude agents. |
detectGhostSkills |
1154 | Defined but never-invoked skills. |
detectGhostCommands |
1184 | Defined but never-invoked slash commands. |
detectBashBloat |
1228 | Shell output limit set above the recommended 15K chars. |
detectLowWorthSessions |
1405 | Sessions with cost but no edits or git delivery. |
detectContextBloat |
1512 | Input:output token ratio above 25:1. |
detectSessionOutliers |
1558 | Sessions costing more than 2x the project average. |
Output Formats
| Command | --format choices |
Default |
|---|---|---|
report, today, month |
tui, json |
tui |
status |
terminal, menubar-json, json |
terminal |
export |
csv, json |
csv |
plan |
text, json |
text |
The macOS menubar and GNOME extension consume menubar-json. src/menubar-json.ts defines the contract; tests/menubar-json.test.ts pins it.
Providers (src/providers/)
Every provider implements the Provider interface in src/providers/types.ts:
type Provider = {
name: string
displayName: string
modelDisplayName(model: string): string
toolDisplayName(rawTool: string): string
discoverSessions(): Promise<SessionSource[]>
createSessionParser(source: SessionSource, seenKeys: Set<string>): SessionParser
}
src/providers/index.ts registers eighteen providers across two tiers:
- Eager:
claude,codex,copilot,droid,gemini,kilo-code,kiro,openclaw,pi,omp,qwen,roo-code. Imported at module load. - Lazy:
antigravity,goose,cursor,opencode,cursor-agent,crush. Imported via dynamicimport()so the heavy dependencies (SQLite, protobuf) do not touch users who do not have those tools installed.
Both lists hit the same getAllProviders() aggregator. A failed lazy import is silent and excludes that provider from the run.
src/providers/vscode-cline-parser.ts is a shared helper consumed by kilo-code and roo-code. It is not registered as a provider on its own.
For the per-provider data location, storage format, parser quirks, and test coverage, see docs/providers/.
macOS Menubar (mac/)
Swift package (mac/Package.swift), targets macOS 14, strict concurrency on. Layout under mac/Sources/CodeBurnMenubar/:
CodeBurnApp.swiftboots the SwiftUIAppand theNSStatusItem.AppStore.swiftis the single source of truth for UI state.Data/holds models, the CLI client, credential stores, and subscription services.DataClient.swiftspawns the CLI and decodesMenubarPayload. See file-level comment for why we never route through/bin/zsh -c.MenubarPayload.swiftmirrors the JSON the CLI emits; keep it in sync withsrc/menubar-json.ts.
Security/CodeburnCLI.swiftresolves the CLI binary (env overrideCODEBURN_BIN, fallbackcodeburn), validates each argv entry against an allowlist regex, and augments PATH for Homebrew and npm-global installs. The Process is launched via/usr/bin/env, never via a shell.Theme/holds color and typography constants and the dark/light state.Views/are the SwiftUI components rendered insideNSPopover.
Tests live in mac/Tests/CodeBurnMenubarTests/ (currently CapacityEstimatorTests.swift).
The build artifact is a zipped .app bundle produced by mac/Scripts/package-app.sh. See RELEASING.md for how the GitHub Actions workflow uses it.
GNOME Extension (gnome/)
Plain JavaScript, no bundler. Targets GNOME Shell 45-50 (metadata.json).
extension.jsis the entry point. Onenable()it constructs aCodeBurnIndicatorand adds it to the panel.indicator.jsis the popover. It owns the period selector, the insight tabs, and the provider filter.dataClient.jswrapsGio.Subprocessto call the CLI. It validates argv against the same allowlist pattern as the macOS client and augments PATH with~/.local/bin,~/.npm-global/bin,~/.volta/bin,~/.bun/bin,~/.cargo/bin,~/.asdf/shims, and a few others. Results are cached for 300 seconds.prefs.jsis the settings dialog backed byschemas/org.gnome.shell.extensions.codeburn.gschema.xml.install.shcopies the extension into~/.local/share/gnome-shell/extensions/.
Build (scripts/, tsup.config.ts)
npm run build is two steps:
node scripts/bundle-litellm.mjsfetches the latest litellm pricing JSON and writessrc/data/litellm-snapshot.json. The bundle script keeps a manual override for MiniMax variants. Direct (un-prefixed) entries win over prefixed ones. The result is checked in so the build is reproducible.tsupreadstsup.config.tsand emits a single ESM bundle atdist/cli.jswith a Node shebang banner. No source maps in publish builds; sourcemaps on for development.
The prepublishOnly hook in package.json runs npm run build so npm publish always ships fresh code.
Tests
npm test runs vitest. Forty-two test files live under tests/:
tests/root (27 files) covers CLI, parser, optimize, cache, format, models, plans.tests/security/(1 file) covers prototype-pollution guards.tests/providers/(14 files) covers per-provider parsing.tests/fixtures/holds redacted real-world session data.
Five providers ship without dedicated test files today: antigravity, claude, gemini, goose, qwen. Closing this gap is a standing good-first-issue.
CI runs Semgrep against .semgrep/rules/no-bracket-assign-hot-paths.yml over src/providers/ and src/parser.ts (.github/workflows/ci.yml). It does not run vitest in CI today; tests run locally before publish.