cc-context-stats 1.7.0 → 1.8.1

package/src/claude_statusline/cli/context_stats.py

@@ -1,512 +0,0 @@
-#!/usr/bin/env python3
-"""Context Stats Visualizer for Claude Code.
-
-Displays ASCII graphs of token consumption over time.
-
-Usage:
-    context-stats [session_id] [options]
-
-Options:
-    --type <cumulative|delta|io|both|all>  Graph type to display (default: both)
-    --watch, -w [interval]                  Real-time monitoring mode (default: 2s)
-    --no-color                              Disable color output
-    --version, -V                           Show version and exit
-    --help                                  Show this help
-"""
-
-from __future__ import annotations
-
-import argparse
-import signal
-import sys
-import time
-from pathlib import Path
-
-from claude_statusline import __version__
-from claude_statusline.core.colors import ColorManager
-from claude_statusline.core.config import Config
-from claude_statusline.core.state import StateFile, _validate_session_id
-from claude_statusline.graphs.renderer import GraphDimensions, GraphRenderer
-from claude_statusline.graphs.statistics import calculate_deltas
-from claude_statusline.ui.icons import get_activity_tier, get_tier_label
-from claude_statusline.ui.waiting import get_waiting_text, is_active
-
-# Cursor control sequences
-CURSOR_HOME = "\033[H"
-CLEAR_SCREEN = "\033[2J"
-HIDE_CURSOR = "\033[?25l"
-SHOW_CURSOR = "\033[?25h"
-CLEAR_TO_END = "\033[J"
-
-
-def show_help() -> None:
-    """Show help message."""
-    print(
-        """Context Stats Visualizer for Claude Code
-
-USAGE:
-    context-stats [session_id] [options]
-    context-stats explain
-
-ARGUMENTS:
-    session_id    Optional session ID. If not provided, uses the latest session.
-
-COMMANDS:
-    explain       Diagnostic dump of Claude Code's JSON context (pipe JSON to stdin)
-
-OPTIONS:
-    --type <type>  Graph type to display:
-                   - delta: Context growth per interaction (default)
-                   - cumulative: Total context usage over time
-                   - io: Input/output tokens over time
-                   - both: Show cumulative and delta graphs
-                   - all: Show all graphs including I/O
-    -w [interval]  Set refresh interval in seconds (default: 2)
-    --no-watch     Show graphs once and exit (disable live monitoring)
-    --no-color     Disable color output
-    --version, -V  Show version and exit
-    --help         Show this help message
-
-NOTE:
-    By default, context-stats runs in live monitoring mode, refreshing every 2 seconds.
-    Press Ctrl+C to exit. Use --no-watch to display graphs once and exit.
-
-EXAMPLES:
-    # Live monitoring (default, refreshes every 2s)
-    context-stats
-
-    # Live monitoring with custom interval
-    context-stats -w 5
-
-    # Show graphs once and exit
-    context-stats --no-watch
-
-    # Show graphs for specific session
-    context-stats abc123def
-
-    # Show only cumulative graph
-    context-stats --type cumulative
-
-    # Combine options
-    context-stats abc123 --type cumulative -w 3
-
-    # Output to file (no colors, single run)
-    context-stats --no-watch --no-color > output.txt
-
-    # Diagnostic dump (pipe Claude Code JSON context)
-    echo '{"model":{"display_name":"Opus"},...}' | context-stats explain
-
-DATA SOURCE:
-    Reads token history from ~/.claude/statusline/statusline.<session_id>.state
-"""
-    )
-
-
-def parse_args() -> argparse.Namespace:
-    """Parse command-line arguments."""
-    parser = argparse.ArgumentParser(
-        description="Context Stats Visualizer for Claude Code",
-        add_help=False,
-    )
-    parser.add_argument("session_id", nargs="?", default=None, help="Session ID")
-    parser.add_argument(
-        "--type",
-        choices=["cumulative", "delta", "io", "both", "all"],
-        default="delta",
-        help="Graph type to display (default: delta)",
-    )
-    parser.add_argument(
-        "--watch",
-        "-w",
-        nargs="?",
-        const=2,
-        type=int,
-        default=2,
-        help="Watch mode interval in seconds (default: 2, use --no-watch to disable)",
-    )
-    parser.add_argument(
-        "--no-watch",
-        action="store_true",
-        help="Disable watch mode (show graphs once and exit)",
-    )
-    parser.add_argument(
-        "--no-color",
-        action="store_true",
-        help="Disable color output",
-    )
-    parser.add_argument(
-        "--help",
-        "-h",
-        action="store_true",
-        help="Show help message",
-    )
-    parser.add_argument(
-        "--version",
-        "-V",
-        action="store_true",
-        help="Show version and exit",
-    )
-
-    args = parser.parse_args()
-
-    if args.version:
-        print(f"cc-context-stats {__version__}")
-        sys.exit(0)
-
-    if args.help:
-        show_help()
-        sys.exit(0)
-
-    if args.session_id is not None:
-        try:
-            _validate_session_id(args.session_id)
-        except ValueError as e:
-            sys.stderr.write(f"Error: {e}\n")
-            sys.exit(1)
-
-    return args
-
-
-def render_once(
-    state_file: StateFile,
-    graph_type: str,
-    renderer: GraphRenderer,
-    colors: ColorManager,
-    watch_mode: bool = False,
-    config: Config | None = None,
-    cycle_index: int = 0,
-) -> bool | str:
-    """Render graphs once.
-
-    Args:
-        state_file: StateFile instance
-        graph_type: Type of graphs to render
-        renderer: GraphRenderer instance
-        colors: ColorManager instance
-        watch_mode: Whether running in watch mode
-        config: Config instance for motion settings
-        cycle_index: Watch mode refresh counter for rotating text
-
-    Returns:
-        True if rendering was successful (non-watch mode),
-        buffered string if watch_mode is True,
-        False if not enough data
-    """
-    entries = state_file.read_history()
-
-    if len(entries) < 2:
-        msg = (
-            f"\n{colors.yellow}Need at least 2 data points to generate graphs.{colors.reset}\n"
-            f"{colors.dim}Found: {len(entries)} entry. Use Claude Code to accumulate more data.{colors.reset}"
-        )
-        if watch_mode:
-            return msg
-        print(msg)
-        return False
-
-    # In watch mode, buffer all output
-    lines: list[str] = []
-
-    def emit(line: str = "") -> None:
-        if watch_mode:
-            lines.append(line)
-        else:
-            print(line)
-
-    # Extract data for graphs
-    timestamps = [e.timestamp for e in entries]
-    # Current context window usage (what's actually in the context)
-    # This is: cache_read + cache_creation + current_input_tokens
-    context_used = [e.current_used_tokens for e in entries]
-    # Per-request I/O tokens from current_usage
-    current_input = [e.current_input_tokens for e in entries]
-    current_output = [e.current_output_tokens for e in entries]
-    deltas = calculate_deltas(context_used)
-    delta_times = timestamps[1:]  # Deltas start from second entry
-
-    # Get session name and project from entries
-    file_path = state_file.find_latest_state_file()
-    session_name = file_path.stem.replace("statusline.", "") if file_path else "unknown"
-
-    # Get project name from the last entry (most recent)
-    last_entry = entries[-1]
-    project_name = ""
-    if last_entry.workspace_project_dir:
-        # Extract just the project folder name from the path
-        project_name = Path(last_entry.workspace_project_dir).name
-
-    # Header
-    if not watch_mode:
-        emit()
-    if project_name:
-        emit(
-            f"{colors.bold}{colors.magenta}Context Stats{colors.reset} "
-            f"{colors.dim}({colors.cyan}{project_name}{colors.dim} • {session_name}){colors.reset}"
-        )
-    else:
-        emit(
-            f"{colors.bold}{colors.magenta}Context Stats{colors.reset} "
-            f"{colors.dim}(Session: {session_name}){colors.reset}"
-        )
-
-    # Activity indicator (waiting text + label)
-    reduced_motion = config.reduced_motion if config else False
-    tier = get_activity_tier(entries, last_entry.context_window_size)
-    label = get_tier_label(tier)
-    active = is_active(entries)
-
-    if active:
-        text = get_waiting_text(cycle_index, reduced_motion)
-        emit(f"  {colors.dim}{text} [{label}]{colors.reset}")
-    else:
-        emit(f"  {colors.dim}{label}{colors.reset}")
-
-    # In watch mode, enable renderer buffering
-    if watch_mode:
-        renderer.begin_buffering()
-
-    # Render requested graphs
-    if graph_type in ("cumulative", "both", "all"):
-        renderer.render_timeseries(
-            context_used, timestamps, "Context Usage Over Time", colors.green
-        )
-
-    if graph_type in ("delta", "both", "all"):
-        renderer.render_timeseries(
-            deltas, delta_times, "Context Growth Per Interaction", colors.cyan
-        )
-
-    if graph_type in ("io", "all"):
-        renderer.render_timeseries(
-            current_input, timestamps, "Input Tokens (per request)", colors.blue
-        )
-        renderer.render_timeseries(
-            current_output, timestamps, "Output Tokens (per request)", colors.magenta
-        )
-
-    # Summary and footer
-    renderer.render_summary(entries, deltas)
-    renderer.render_footer(__version__)
-
-    if watch_mode:
-        # Collect renderer buffer and combine with header lines
-        renderer_output = renderer.get_buffer()
-        lines.append(renderer_output)
-        return "\n".join(lines)
-
-    return True
-
-
-def run_watch_mode(
-    state_file: StateFile,
-    graph_type: str,
-    interval: int,
-    renderer: GraphRenderer,
-    colors: ColorManager,
-    config: Config | None = None,
-) -> None:
-    """Run in watch mode with continuous refresh.
-
-    Args:
-        state_file: StateFile instance
-        graph_type: Type of graphs to render
-        interval: Refresh interval in seconds
-        renderer: GraphRenderer instance
-        colors: ColorManager instance
-        config: Config instance for motion settings
-    """
-
-    # Signal handler for clean exit
-    def handle_signal(_signum: int, _frame: object) -> None:
-        sys.stdout.write(SHOW_CURSOR)
-        sys.stdout.flush()
-        print(f"\n{colors.dim}Watch mode stopped.{colors.reset}")
-        sys.exit(0)
-
-    signal.signal(signal.SIGINT, handle_signal)
-    signal.signal(signal.SIGTERM, handle_signal)
-
-    # Hide cursor and initial clear in one write
-    sys.stdout.write(f"{HIDE_CURSOR}{CLEAR_SCREEN}{CURSOR_HOME}")
-    sys.stdout.flush()
-
-    cycle_counter = 0
-
-    try:
-        while True:
-            # Update dimensions in case of terminal resize
-            renderer.dimensions = GraphDimensions.detect()
-
-            # Build all output into a buffer
-            buf_lines: list[str] = []
-
-            # Watch mode indicator
-            current_time = time.strftime("%H:%M:%S")
-            buf_lines.append(
-                f"{colors.dim}[LIVE {current_time}] Refresh: {interval}s | Ctrl+C to exit{colors.reset}"
-            )
-
-            # Check if state file exists now (may have been created since start)
-            file_path = state_file.find_latest_state_file()
-            if not file_path or not file_path.exists():
-                # Show waiting message for new session
-                reduced_motion = config.reduced_motion if config else False
-                text = get_waiting_text(cycle_counter, reduced_motion)
-                buf_lines.append(
-                    _format_waiting_message(
-                        colors,
-                        state_file.session_id,
-                        text,
-                    )
-                )
-            else:
-                # Render graphs (returns buffered string in watch mode)
-                result = render_once(
-                    state_file,
-                    graph_type,
-                    renderer,
-                    colors,
-                    watch_mode=True,
-                    config=config,
-                    cycle_index=cycle_counter,
-                )
-                if isinstance(result, str):
-                    buf_lines.append(result)
-
-            # Atomic write: CURSOR_HOME + content + CLEAR_TO_END (clean up stale trailing lines)
-            buffered_content = "\n".join(buf_lines)
-            sys.stdout.write(f"{CURSOR_HOME}{buffered_content}\n{CLEAR_TO_END}")
-            sys.stdout.flush()
-
-            cycle_counter += 1
-            time.sleep(interval)
-    finally:
-        sys.stdout.write(SHOW_CURSOR)
-        sys.stdout.flush()
-
-
-def _format_waiting_message(
-    colors: ColorManager,
-    session_id: str | None,
-    message: str = "Waiting for session data...",
-) -> str:
-    """Format a waiting message as a string.
-
-    Args:
-        colors: ColorManager instance
-        session_id: Session ID if specified
-        message: Message to display
-
-    Returns:
-        Formatted waiting message string
-    """
-    lines = [""]
-    if session_id:
-        lines.append(
-            f"{colors.bold}{colors.magenta}Context Stats{colors.reset} "
-            f"{colors.dim}(Session: {session_id}){colors.reset}"
-        )
-    else:
-        lines.append(f"{colors.bold}{colors.magenta}Context Stats{colors.reset}")
-    lines.append("")
-    lines.append(f"  {colors.cyan}⏳ {message}{colors.reset}")
-    lines.append("")
-    lines.append(
-        f"  {colors.dim}The session has just started and no data has been recorded yet.{colors.reset}"
-    )
-    lines.append(
-        f"  {colors.dim}Data will appear after the first Claude interaction.{colors.reset}"
-    )
-    lines.append("")
-    return "\n".join(lines)
-
-
-def show_waiting_message(
-    colors: ColorManager,
-    session_id: str | None,
-    message: str = "Waiting for session data...",
-) -> None:
-    """Show a friendly waiting message for new sessions.
-
-    Args:
-        colors: ColorManager instance
-        session_id: Session ID if specified
-        message: Message to display
-    """
-    print(_format_waiting_message(colors, session_id, message))
-
-
-def _ensure_utf8_stdout() -> None:
-    """Reconfigure stdout/stderr to UTF-8 on Windows where cp1252 is the default."""
-    if sys.stdout.encoding and sys.stdout.encoding.lower().replace("-", "") != "utf8":
-        sys.stdout.reconfigure(encoding="utf-8", errors="replace")  # type: ignore[union-attr]
-    if sys.stderr.encoding and sys.stderr.encoding.lower().replace("-", "") != "utf8":
-        sys.stderr.reconfigure(encoding="utf-8", errors="replace")  # type: ignore[union-attr]
-
-
-def main() -> None:
-    """Main entry point for context-stats CLI."""
-    _ensure_utf8_stdout()
-
-    # Handle 'explain' subcommand before argparse (it expects stdin JSON, not flags)
-    if len(sys.argv) > 1 and sys.argv[1] == "explain":
-        import json
-
-        from claude_statusline.cli.explain import run_explain
-
-        no_color = "--no-color" in sys.argv
-        try:
-            data = json.load(sys.stdin)
-        except json.JSONDecodeError as e:
-            sys.stderr.write(f"Error: invalid JSON on stdin: {e}\n")
-            sys.stderr.write("Usage: echo '{...}' | context-stats explain\n")
-            sys.exit(1)
-        run_explain(data, no_color=no_color)
-        return
-
-    args = parse_args()
-
-    # Load config for token_detail setting
-    config = Config.load()
-
-    # Setup colors with any user overrides from config
-    color_enabled = not args.no_color and sys.stdout.isatty()
-    colors = ColorManager(enabled=color_enabled, overrides=config.color_overrides)
-
-    # Setup state file
-    state_file = StateFile(args.session_id)
-
-    # Find state file
-    file_path = state_file.find_latest_state_file()
-
-    # Handle case where no state file exists yet
-    if not file_path or not file_path.exists():
-        if args.no_watch:
-            # Single run mode - show friendly message and exit
-            if args.session_id:
-                show_waiting_message(colors, args.session_id)
-            else:
-                print(f"{colors.yellow}No session data found.{colors.reset}")
-                print(f"{colors.dim}Run Claude Code to generate token usage data.{colors.reset}")
-            sys.exit(0)
-        else:
-            # Watch mode - continue and wait for data
-            pass
-
-    # Setup renderer
-    renderer = GraphRenderer(
-        colors=colors,
-        token_detail=config.token_detail,
-    )
-
-    # Run
-    if args.no_watch:
-        success = render_once(state_file, args.type, renderer, colors, config=config)
-        if not success:
-            sys.exit(1)
-    else:
-        run_watch_mode(state_file, args.type, args.watch, renderer, colors, config=config)
-
-
-if __name__ == "__main__":
-    main()