feat: add CLI commands for usage analysis, cost breakdown, and setup

Implement the complete Cobra command tree (11 subcommands): - cmd/root.go: Root command with persistent flags (--days, --project, --model, --no-cache, --data-dir, --quiet, --no-subagents). Shared loadData() orchestrates the full pipeline: tries cache-assisted loading first, falls back to uncached parse on cache failure, reports progress to stderr. applyFilters() applies project/model substring filters and computes the time window. - cmd/summary.go: Default command (also "cburn summary"). Renders a bordered metrics table with token breakdown (5 types), cost with cache savings, and per-day rates with period-over-period deltas. - cmd/costs.go: Detailed cost analysis — breaks down costs by token type (output, cache_write_1h, input, cache_write_5m, cache_read) with share percentages, period comparison bar chart, and per-model cost breakdown (input/output/cache/total columns). - cmd/daily.go: Daily usage table (date, weekday, sessions, prompts, tokens, cost) sorted most-recent-first. - cmd/hourly.go: Activity heatmap showing prompt distribution across 24 hours with Unicode block bars, reports peak hour. - cmd/models.go: Model usage ranking with API call counts, token volumes, cost, and usage share percentage. - cmd/projects.go: Project ranking by cost with session/prompt/token counts. - cmd/sessions.go: Session list sorted by recency with --limit flag (default 20). Shows start time, project, duration, tokens, cost. Marks subagent sessions with "(sub)" suffix. - cmd/config_cmd.go: Displays current configuration across all sections (general, admin API, appearance, budget) with auto- detected plan ceiling. - cmd/setup.go: Interactive first-run wizard — configures Admin API key, default time range (7/30/90 days), and color theme (Flexoki Dark, Catppuccin Mocha, Tokyo Night, Terminal). Saves to ~/.config/cburn/config.toml. - cmd/tui.go: Launches the interactive Bubble Tea TUI dashboard, passing through all filter flags and applying the configured theme. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-02-19 13:02:13 -05:00
parent f7d8fea140
commit 5aa184a7ab
11 changed files with 976 additions and 0 deletions
--- a/cmd/root.go
+++ b/cmd/root.go
@@ -0,0 +1,142 @@
+package cmd
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"time"
+
+	"cburn/internal/cli"
+	"cburn/internal/model"
+	"cburn/internal/pipeline"
+	"cburn/internal/store"
+
+	"github.com/spf13/cobra"
+)
+
+var (
+	flagDays        int
+	flagProject     string
+	flagModel       string
+	flagNoCache     bool
+	flagDataDir     string
+	flagQuiet       bool
+	flagNoSubagents bool
+)
+
+var rootCmd = &cobra.Command{
+	Use:   "cburn",
+	Short: "Claude Usage Metrics CLI",
+	Long:  "Analyze your Claude Code usage: tokens, costs, sessions, and more.",
+	RunE:  runSummary,
+}
+
+// Execute is the main entry point called from main.go.
+func Execute() {
+	if err := rootCmd.Execute(); err != nil {
+		os.Exit(1)
+	}
+}
+
+func init() {
+	homeDir, _ := os.UserHomeDir()
+	defaultDataDir := filepath.Join(homeDir, ".claude")
+
+	rootCmd.PersistentFlags().IntVarP(&flagDays, "days", "n", 30, "Time window in days")
+	rootCmd.PersistentFlags().StringVarP(&flagProject, "project", "p", "", "Filter to project (substring match)")
+	rootCmd.PersistentFlags().StringVarP(&flagModel, "model", "m", "", "Filter to model (substring match)")
+	rootCmd.PersistentFlags().BoolVar(&flagNoCache, "no-cache", false, "Skip SQLite cache, reparse everything")
+	rootCmd.PersistentFlags().StringVarP(&flagDataDir, "data-dir", "d", defaultDataDir, "Claude data directory")
+	rootCmd.PersistentFlags().BoolVarP(&flagQuiet, "quiet", "q", false, "Suppress progress output")
+	rootCmd.PersistentFlags().BoolVar(&flagNoSubagents, "no-subagents", false, "Exclude subagent sessions")
+}
+
+// loadData is the shared data loading path used by all commands.
+// Uses SQLite cache when available for fast subsequent runs.
+func loadData() (*pipeline.LoadResult, error) {
+	if !flagQuiet {
+		fmt.Fprintf(os.Stderr, "  Scanning sessions...\n")
+	}
+
+	progressFn := func(current, total int) {
+		if flagQuiet {
+			return
+		}
+		if current%100 == 0 || current == total {
+			fmt.Fprintf(os.Stderr, "\r  Parsing [%d/%d]", current, total)
+		}
+	}
+
+	// Try cached load unless --no-cache
+	if !flagNoCache {
+		cache, err := store.Open(pipeline.CachePath())
+		if err != nil {
+			// Cache open failed — fall back to uncached
+			if !flagQuiet {
+				fmt.Fprintf(os.Stderr, "  Cache unavailable, doing full parse\n")
+			}
+		} else {
+			defer cache.Close()
+
+			cr, err := pipeline.LoadWithCache(flagDataDir, !flagNoSubagents, cache, progressFn)
+			if err != nil {
+				// Cache-assisted load failed — fall back
+				if !flagQuiet {
+					fmt.Fprintf(os.Stderr, "\n  Cache error, falling back to full parse\n")
+				}
+			} else {
+				if !flagQuiet && cr.TotalFiles > 0 {
+					if cr.Reparsed == 0 {
+						fmt.Fprintf(os.Stderr, "\r  Loaded %s sessions from cache (%d projects)    \n",
+							formatNumber(int64(len(cr.Sessions))),
+							cr.ProjectCount,
+						)
+					} else {
+						fmt.Fprintf(os.Stderr, "\r  %s cached + %d reparsed (%d projects)    \n",
+							formatNumber(int64(cr.CacheHits)),
+							cr.Reparsed,
+							cr.ProjectCount,
+						)
+					}
+				}
+				return &cr.LoadResult, nil
+			}
+		}
+	}
+
+	// Uncached path
+	result, err := pipeline.Load(flagDataDir, !flagNoSubagents, progressFn)
+	if err != nil {
+		return nil, err
+	}
+
+	if !flagQuiet && result.TotalFiles > 0 {
+		fmt.Fprintf(os.Stderr, "\r  Parsed %s sessions across %d projects    \n",
+			formatNumber(int64(result.ParsedFiles)),
+			result.ProjectCount,
+		)
+	}
+
+	return result, nil
+}
+
+// applyFilters returns filtered sessions and the computed time range.
+func applyFilters(sessions []model.SessionStats) ([]model.SessionStats, time.Time, time.Time) {
+	now := time.Now()
+	since := now.AddDate(0, 0, -flagDays)
+	until := now
+
+	filtered := sessions
+	if flagProject != "" {
+		filtered = pipeline.FilterByProject(filtered, flagProject)
+	}
+	if flagModel != "" {
+		filtered = pipeline.FilterByModel(filtered, flagModel)
+	}
+
+	return filtered, since, until
+}
+
+func formatNumber(n int64) string {
+	return cli.FormatNumber(n)
+}