usage-cli 0.29.32__py3-none-any.whl

tui.py

@@ -0,0 +1,288 @@
+# SPDX-License-Identifier: AGPL-3.0-only
+# Copyright (C) 2026 lollapalooza <https://github.com/aqua5230>
+#
+# Part of "usage". Free software licensed under the GNU Affero General Public
+# License v3.0 only; see the LICENSE file for full terms and the warranty disclaimer.
+
+from __future__ import annotations
+
+import json
+import math
+import time
+from dataclasses import dataclass, field
+from functools import lru_cache
+from pathlib import Path
+
+from rich.align import Align
+from rich.console import Console, Group, RenderableType
+from rich.panel import Panel
+from rich.table import Table
+from rich.text import Text
+
+from i18n import packaged_resource_path
+from tui_sprite import render_sprite
+from usage_client import PollState, UsageSnapshot
+from usage_lang import detect_lang
+
+BG = "#000000"
+PANEL = "#1f1f1e"
+TEXT = "#faf9f5"
+DIM = "#b0aea5"
+ACCENT = "#d97757"
+GREEN = "#788c5d"
+RED = "#c0392b"
+BAR_BG = "#2a2a28"
+
+SPINNER_FRAMES = ["·", "✻", "✽", "✶", "✳", "✢"]
+SPINNER_PHASES = [0, 1, 2, 3, 4, 5, 4, 3, 2, 1]
+SPINNER_PHASE_MS = [260, 130, 130, 130, 130, 260, 130, 130, 130, 130]
+LOADING_INTERVAL_MS = 4000
+I18N_PATH = packaged_resource_path("i18n.json", Path(__file__).with_name("i18n.json"))
+
+
+@lru_cache(maxsize=1)
+def _load_i18n_bundle() -> dict[str, dict[str, str]]:
+    data = json.loads(I18N_PATH.read_text(encoding="utf-8"))
+    return {
+        str(lang): {str(key): str(value) for key, value in values.items()}
+        for lang, values in data.items()
+    }
+
+
+def _t(language: str, key: str, **kwargs: object) -> str:
+    bundle = _load_i18n_bundle()
+    table = bundle.get(language) or bundle["en"]
+    template = table.get(key) or bundle["en"].get(key) or key
+    return template.format(**kwargs)
+
+
+def _loading_phrases(language: str) -> list[str]:
+    return _t(language, "loading_phrases").split("|")
+
+
+@dataclass(slots=True)
+class AppViewState:
+    language: str = field(default_factory=detect_lang)
+    poll_state: PollState = PollState.LOADING
+    snapshot: UsageSnapshot | None = None
+    message: str = ""
+    fatal_message: str | None = None
+    started_at: float = field(default_factory=time.monotonic)
+    rate_group: int = 0
+    codex_credits: tuple[str | None, bool] | None = None
+
+
+def format_countdown(reset_at: float, language: str, now: float | None = None) -> str:
+    current_time = now if now is not None else time.time()
+    remaining = max(0, math.ceil(reset_at - current_time))
+    days, rem = divmod(remaining, 86400)
+    hours, rem = divmod(rem, 3600)
+    minutes, seconds = divmod(rem, 60)
+
+    if days > 0:
+        return _t(language, "resets_in_days", days=days, hours=hours)
+    if hours > 0:
+        return _t(language, "resets_in_hours", hours=hours, minutes=minutes)
+    return _t(language, "resets_in_minutes", minutes=minutes, seconds=seconds)
+
+
+def _bar_style(percent: int) -> str:
+    if percent < 50:
+        return GREEN
+    if percent <= 80:
+        return ACCENT
+    return RED
+
+
+def _build_progress_line(percent: int, width: int = 28) -> Text:
+    filled = round((percent / 100) * width)
+    text = Text()
+    text.append("▄" * filled, style=f"bold {_bar_style(percent)}")
+    text.append("▄" * max(0, width - filled), style=f"bold {BAR_BG}")
+    return text
+
+
+def _chip(label: str) -> Text:
+    return Text(f" {label} ", style=f"bold {TEXT} on {PANEL}")
+
+
+def _usage_block(
+    percent: int,
+    label: str,
+    reset_at: float,
+    now: float,
+    language: str,
+) -> RenderableType:
+    row = Table.grid(expand=False, padding=(0, 1))
+    row.add_column(width=4)
+    row.add_column(width=16)
+    row.add_column(width=11)
+    row.add_row(
+        Text(f"{percent}%", style=f"bold {_bar_style(percent)}"),
+        _build_progress_line(percent, width=15),
+        _chip(label),
+    )
+
+    countdown = Text.assemble("    ", (format_countdown(reset_at, language, now), DIM))
+    return Group(row, countdown)
+
+
+def _missing_usage_block(label: str, language: str) -> RenderableType:
+    row = Table.grid(expand=False, padding=(0, 1))
+    row.add_column(width=4)
+    row.add_column(width=16)
+    row.add_column(width=11)
+    row.add_row(
+        Text("--", style=f"bold {DIM}"),
+        _build_progress_line(0, width=15),
+        _chip(label),
+    )
+    return Group(row, Text(f"    {_t(language, 'resets_in_placeholder')}", style=DIM))
+
+
+def _spinner_frame(now: float, started_at: float) -> str:
+    elapsed_ms = int((now - started_at) * 1000)
+    total_ms = sum(SPINNER_PHASE_MS)
+    phase_ms = elapsed_ms % total_ms
+    accumulated = 0
+
+    for phase_index, duration in enumerate(SPINNER_PHASE_MS):
+        accumulated += duration
+        if phase_ms < accumulated:
+            return SPINNER_FRAMES[SPINNER_PHASES[phase_index]]
+    return SPINNER_FRAMES[SPINNER_PHASES[-1]]
+
+
+def _status_line(state: AppViewState, now: float) -> Text:
+    spinner = _spinner_frame(now, state.started_at)
+    elapsed_ms = (now - state.started_at) * 1000
+    phrases = _loading_phrases(state.language)
+    phrase_index = int(elapsed_ms / LOADING_INTERVAL_MS) % len(phrases)
+
+    color = ACCENT
+    phrase = phrases[phrase_index]
+
+    if state.poll_state == PollState.RATE_LIMITED:
+        phrase = _t(state.language, "status_rate_limited")
+        color = RED
+    elif state.poll_state == PollState.TOKEN_ERROR:
+        phrase = _t(state.language, "status_token_unavailable")
+        color = ACCENT
+    elif state.poll_state in (PollState.CONNECTION_ERROR, PollState.FATAL):
+        phrase = _t(state.language, "status_api_offline")
+        color = RED
+
+    return Text.assemble(
+        (f"{spinner} ", f"bold {color}"),
+        (phrase, f"bold {color}"),
+    )
+
+
+def render_screen(state: AppViewState, frame_index: int) -> Panel:
+    now = time.time()
+    anim_now = time.monotonic()
+    panel_width = min(Console().size.width, 60)
+
+    # Group label and colors
+    groups = [
+        _t(state.language, "group_idle"),
+        _t(state.language, "group_normal"),
+        _t(state.language, "group_active"),
+        _t(state.language, "group_heavy"),
+    ]
+    group_colors = [DIM, TEXT, ACCENT, RED]
+    group_label = groups[state.rate_group]
+    group_color = group_colors[state.rate_group]
+
+    # Status dot color
+    dot_color = ACCENT
+    if state.poll_state == PollState.SUCCESS:
+        dot_color = GREEN
+    elif state.poll_state in (
+        PollState.TOKEN_ERROR,
+        PollState.CONNECTION_ERROR,
+        PollState.RATE_LIMITED,
+        PollState.FATAL,
+    ):
+        dot_color = RED
+
+    title_table = Table.grid(expand=True, padding=(0, 0))
+    title_table.add_column(width=10)
+    title_table.add_column(ratio=1, justify="left")
+    title_table.add_column(width=4, justify="right")
+
+    title_left = Text.assemble(
+        (f"[{group_label}]", f"bold {group_color}"),
+        (" •", f"bold {dot_color}"),
+    )
+
+    title_table.add_row(
+        render_sprite(frame_index),
+        Text.assemble((f"  {_t(state.language, 'usage_title')}  ", f"bold {TEXT}"), title_left),
+        Text("🔋"),
+    )
+
+    body_items: list[RenderableType] = []
+
+    if state.snapshot is None:
+        loading_block = Group(
+            _usage_block(0, _t(state.language, "current_label"), now + 3600, now, state.language),
+            Text(""),
+            _usage_block(0, _t(state.language, "weekly_label"), now + 86400, now, state.language),
+        )
+        body_items.append(loading_block)
+    else:
+        current_block = (
+            _usage_block(
+                state.snapshot.current_percent,
+                _t(state.language, "current_label"),
+                state.snapshot.current_reset_at,
+                now,
+                state.language,
+            )
+            if state.snapshot.current_percent is not None
+            else _missing_usage_block(_t(state.language, "current_label"), state.language)
+        )
+        weekly_block = (
+            _usage_block(
+                state.snapshot.weekly_percent,
+                _t(state.language, "weekly_label"),
+                state.snapshot.weekly_reset_at,
+                now,
+                state.language,
+            )
+            if state.snapshot.weekly_percent is not None
+            else _missing_usage_block(_t(state.language, "weekly_label"), state.language)
+        )
+        body_items.append(
+            Group(
+                current_block,
+                Text(""),
+                weekly_block,
+            )
+        )
+
+    if state.codex_credits is not None:
+        balance, unlimited = state.codex_credits
+        text = (
+            _t(state.language, "codex_credits_unlimited")
+            if unlimited
+            else _t(state.language, "codex_credits", balance=balance or "--")
+        )
+        body_items.append(Text(text, style=DIM))
+
+    screen = Group(
+        title_table,
+        Text(""),
+        *body_items,
+        Text(""),
+        Align.center(_status_line(state, anim_now)),
+    )
+
+    return Panel(
+        screen,
+        border_style=PANEL,
+        padding=(1, 2),
+        width=panel_width,
+        style=f"on {BG}",
+    )

tui_sprite.py

@@ -0,0 +1,206 @@
+# SPDX-License-Identifier: AGPL-3.0-only
+# Copyright (C) 2026 lollapalooza <https://github.com/aqua5230>
+#
+# Part of "usage". Free software licensed under the GNU Affero General Public
+# License v3.0 only; see the LICENSE file for full terms and the warranty disclaimer.
+
+from __future__ import annotations
+
+from rich.style import Style
+from rich.text import Text
+
+ORANGE = "#d97757"
+EYE = "#1A1A1A"
+
+Pixel = str | None
+Frame = list[list[Pixel]]
+
+TRANSPARENT: Pixel = None
+
+
+FRAMES: list[Frame] = [
+    [
+        [
+            TRANSPARENT,
+            TRANSPARENT,
+            ORANGE,
+            ORANGE,
+            ORANGE,
+            ORANGE,
+            ORANGE,
+            TRANSPARENT,
+            TRANSPARENT,
+        ],
+        [TRANSPARENT, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, TRANSPARENT],
+        [ORANGE, ORANGE, EYE, ORANGE, ORANGE, ORANGE, EYE, ORANGE, ORANGE],
+        [ORANGE, ORANGE, EYE, ORANGE, ORANGE, ORANGE, EYE, ORANGE, ORANGE],
+        [ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE],
+        [EYE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE],
+        [
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+        ],
+        [
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+        ],
+    ],
+    [
+        [
+            TRANSPARENT,
+            TRANSPARENT,
+            ORANGE,
+            ORANGE,
+            ORANGE,
+            ORANGE,
+            ORANGE,
+            TRANSPARENT,
+            TRANSPARENT,
+        ],
+        [TRANSPARENT, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, TRANSPARENT],
+        [ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE],
+        [ORANGE, ORANGE, EYE, ORANGE, ORANGE, ORANGE, EYE, ORANGE, ORANGE],
+        [ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE],
+        [EYE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE],
+        [
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+        ],
+        [
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+        ],
+    ],
+    [
+        [
+            TRANSPARENT,
+            TRANSPARENT,
+            ORANGE,
+            ORANGE,
+            ORANGE,
+            ORANGE,
+            ORANGE,
+            TRANSPARENT,
+            TRANSPARENT,
+        ],
+        [TRANSPARENT, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, TRANSPARENT],
+        [ORANGE, ORANGE, EYE, ORANGE, ORANGE, ORANGE, EYE, ORANGE, ORANGE],
+        [ORANGE, ORANGE, EYE, ORANGE, ORANGE, ORANGE, EYE, ORANGE, ORANGE],
+        [ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE],
+        [EYE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE],
+        [
+            TRANSPARENT,
+            TRANSPARENT,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+        ],
+        [
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+        ],
+    ],
+    [
+        [
+            TRANSPARENT,
+            TRANSPARENT,
+            ORANGE,
+            ORANGE,
+            ORANGE,
+            ORANGE,
+            ORANGE,
+            TRANSPARENT,
+            TRANSPARENT,
+        ],
+        [TRANSPARENT, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, TRANSPARENT],
+        [ORANGE, ORANGE, EYE, ORANGE, ORANGE, ORANGE, EYE, ORANGE, ORANGE],
+        [ORANGE, ORANGE, EYE, ORANGE, ORANGE, ORANGE, EYE, ORANGE, ORANGE],
+        [ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE],
+        [EYE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE, ORANGE],
+        [
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            TRANSPARENT,
+            TRANSPARENT,
+        ],
+        [
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+            ORANGE,
+            TRANSPARENT,
+        ],
+    ],
+]
+
+
+def render_sprite(frame_index: int) -> Text:
+    frame = FRAMES[frame_index % len(FRAMES)]
+    text = Text()
+
+    for row in range(0, len(frame), 2):
+        upper = frame[row]
+        lower = frame[row + 1] if row + 1 < len(frame) else [None] * len(upper)
+
+        for upper_pixel, lower_pixel in zip(upper, lower, strict=True):
+            if upper_pixel is None and lower_pixel is None:
+                text.append(" ")
+                continue
+            if upper_pixel is None:
+                text.append("▄", style=Style(color=lower_pixel))
+                continue
+            if lower_pixel is None:
+                text.append("▀", style=Style(color=upper_pixel))
+                continue
+            text.append("▀", style=Style(color=upper_pixel, bgcolor=lower_pixel))
+        text.append("\n")
+
+    return text

ui/__init__.py

@@ -0,0 +1,5 @@
+# SPDX-License-Identifier: AGPL-3.0-only
+# Copyright (C) 2026 lollapalooza <https://github.com/aqua5230>
+#
+# Part of "usage". Free software licensed under the GNU Affero General Public
+# License v3.0 only; see the LICENSE file for full terms and the warranty disclaimer.