bone-agent 1.3.0

package/src/ui/displays.py

@@ -0,0 +1,213 @@
+"""UI display functions for command outputs.""" 
+
+from rich.table import Table
+from rich.panel import Panel
+from rich import box
+from llm import config
+
+
+def show_provider_table(current_provider: str, console):
+    """Display provider status table.
+
+    Args:
+        current_provider: Name of the currently active provider.
+        console: Rich Console instance for output.
+    """
+    table = Table("Provider", "Status", "Details", title="Providers", box=box.SIMPLE_HEAD)
+    for provider in config.get_providers():
+        cfg = config.get_provider_config(provider)
+        model = cfg.get("model", "N/A")
+        if provider == "local":
+            status = "✅" if cfg.get("model") else "❌ (set model path)"
+        else:
+            status = "✅" if cfg.get("api_key") else "❌ (set API key)"
+        active = " [green](active)[/green]" if provider == current_provider else ""
+        table.add_row(provider.capitalize(), status, f"{model[:40]}{active}")
+
+    console.print(table)
+
+    help_text = """Usage: /provider <name>
+
+Opens an editor to configure model, API key, and costs.
+
+Examples:
+  /provider openrouter
+  /provider glm
+  /provider local
+  /provider gemini
+  /provider minimax
+  /provider anthropic
+  /provider kimi"""
+    console.print(Panel(help_text, title="[bold #5F9EA0]Provider Settings[/bold #5F9EA0]", border_style="grey23", padding=(0, 2)))
+    console.print("")
+
+
+def show_help_table(console):
+    """Display command help table.
+
+    Args:
+        console: Rich Console instance for output.
+    """
+    console.print("")
+    table = Table(show_header=True, box=box.SIMPLE_HEAD)
+    table.add_column("Command", no_wrap=True)
+    table.add_column("Description")
+
+    table.add_row("[bold #5F9EA0]/help[/bold #5F9EA0]", "Show help")
+    table.add_row("[bold #5F9EA0]/exit[/bold #5F9EA0]", "Exit chat")
+    table.add_row("[bold #5F9EA0]/config[/bold #5F9EA0]", "Show all configuration settings")
+    table.add_row("[bold #5F9EA0]/provider[/bold #5F9EA0] [name]", "Configure provider settings (model, key, costs)")
+    table.add_row("[bold #5F9EA0]/key[/bold #5F9EA0] <key>", "Set API key for current provider")
+    table.add_row("[bold #5F9EA0]/model[/bold #5F9EA0] <name>", "Set model for current provider")
+    table.add_row("[bold #5F9EA0]/usage[/bold #5F9EA0] [provider] [in|out] <cost>", "Set/view provider-specific token cost")
+    table.add_row("[bold #5F9EA0]/compact[/bold #5F9EA0]", "Compact context with an AI summary")
+
+
+    table.add_row("[bold #5F9EA0]/cd[/bold #5F9EA0] [path]", "Change working directory (no args to show current)")
+    table.add_row("[bold #5F9EA0]/edit[/bold #5F9EA0], [bold #5F9EA0]/e[/bold #5F9EA0]", "Open editor for multi-line input")
+    table.add_row("[bold #5F9EA0]/review[/bold #5F9EA0] [args], [bold #5F9EA0]/r[/bold #5F9EA0]", "Code review git changes (e.g. /review --staged, /review main..HEAD)")
+    table.add_row("[bold #5F9EA0]/obsidian[/bold #5F9EA0] [set|enable|disable|status|init]", "Manage vault integration, scaffold project folders")
+    table.add_row("[bold #5F9EA0]/tools[/bold #5F9EA0] [list|enable|disable|enable-group|disable-group]", "Toggle tools or groups (e.g. file_ops, task_mgmt)")
+    table.add_row("[bold #5F9EA0]/setup[/bold #5F9EA0]", "Re-run the first-run setup wizard")
+    table.add_row("[bold #5F9EA0]/cron[/bold #5F9EA0] [list|add|remove|enable|disable|run]", "Manage scheduled cron jobs")
+
+
+    console.print(Panel(table, title="[bold #5F9EA0]Commands[/bold #5F9EA0]", border_style="grey23", padding=(0, 2)))
+
+    # Account management section
+    console.print()
+    acct_table = Table(show_header=True, box=box.SIMPLE_HEAD)
+    acct_table.add_column("Command", no_wrap=True)
+    acct_table.add_column("Description")
+
+    acct_table.add_row("[bold #5F9EA0]/signup[/bold #5F9EA0] <email>", "Create bone-agent account and get API key")
+    acct_table.add_row("[bold #5F9EA0]/login[/bold #5F9EA0]", "Log in to an existing bone-agent account")
+    acct_table.add_row("[bold #5F9EA0]/account[/bold #5F9EA0]", "View account info and subscription status")
+    acct_table.add_row("[bold #5F9EA0]/plan[/bold #5F9EA0]", "View available plans and pricing")
+    acct_table.add_row("[bold #5F9EA0]/upgrade[/bold #5F9EA0]", "Upgrade or change your plan")
+    acct_table.add_row("[bold #5F9EA0]/manage[/bold #5F9EA0]", "Cancel subscription or update payment (Stripe portal)")
+    acct_table.add_row("[bold #5F9EA0]/rotate-key[/bold #5F9EA0]", "Invalidate current API key and generate a new one")
+    acct_table.add_row("[bold #5F9EA0]/reset-key[/bold #5F9EA0]", "Get a new API key emailed to you (lost key recovery)")
+
+    console.print(Panel(acct_table, title="[bold #5F9EA0]Account[/bold #5F9EA0]", border_style="grey23", padding=(0, 2)))
+
+    # Keybinds section
+    console.print()
+    keybinds = Table(show_header=True, box=box.SIMPLE_HEAD)
+    keybinds.add_column("Keybind", no_wrap=True)
+    keybinds.add_column("Action")
+
+    keybinds.add_row("Shift+Tab", "Cycle approval mode")
+    keybinds.add_row("Ctrl+C", "Interrupt response")
+    keybinds.add_row("Ctrl+C (2x)", "Exit program")
+
+    console.print(Panel(keybinds, title="[bold #5F9EA0]Keybinds[/bold #5F9EA0]", border_style="grey23", padding=(0, 2)))
+    console.print("")
+
+
+def show_cron_help_table(console):
+    """Display cron command help table.
+
+    Args:
+        console: Rich Console instance for output.
+    """
+    console.print("")
+    table = Table(show_header=True, box=box.SIMPLE_HEAD)
+    table.add_column("Command", no_wrap=True)
+    table.add_column("Description")
+
+    table.add_row("[bold #5F9EA0]/cron list[/bold #5F9EA0]", "Show all cron jobs (default)")
+    table.add_row("[bold #5F9EA0]/cron add[/bold #5F9EA0] <id> <schedule> <cmd>", "Add a new cron job")
+    table.add_row("[bold #5F9EA0]/cron remove[/bold #5F9EA0] <id>", "Remove a cron job")
+    table.add_row("[bold #5F9EA0]/cron enable[/bold #5F9EA0] <id>", "Enable a cron job")
+    table.add_row("[bold #5F9EA0]/cron disable[/bold #5F9EA0] <id>", "Disable a cron job")
+    table.add_row("[bold #5F9EA0]/cron run[/bold #5F9EA0] <id>", "Run a job immediately (interactive)")
+    table.add_row("[bold #5F9EA0]/cron allowlist[/bold #5F9EA0] [list|add|remove|clear]", "Manage allowed commands for a job")
+
+    console.print(Panel(table, title="[bold #5F9EA0]Commands[/bold #5F9EA0]", border_style="grey23", padding=(0, 2)))
+
+    # Schedule formats section
+    console.print()
+    sched_table = Table(show_header=True, box=box.SIMPLE_HEAD)
+    sched_table.add_column("Format")
+    sched_table.add_column("Example")
+
+    sched_table.add_row("every <n> <unit>", "every 5 minutes, every 1 hour, every 3 days")
+    sched_table.add_row("daily at <time>", "daily at 8am, daily at 17:30")
+    sched_table.add_row("<day>s at <time>", "weekdays at 9am, mondays at 10:30pm")
+    sched_table.add_row("<time>", "08:00, 17:30")
+
+    console.print(Panel(sched_table, title="[bold #5F9EA0]Schedule Formats[/bold #5F9EA0]", border_style="grey23", padding=(0, 2)))
+    console.print("")
+
+
+def show_config_overview(chat_manager, console, debug_mode_container, current_provider):
+    """Display comprehensive configuration overview.
+
+    Args:
+        chat_manager: ChatManager instance for runtime state
+        console: Rich Console instance for output
+        debug_mode_container: Dict with debug key for debug mode state
+        current_provider: Name of the currently active provider
+    """
+    from core.config_manager import ConfigManager
+    config_manager = ConfigManager()
+    config_data = config_manager.load()
+
+    console.print()
+
+    # ===== Runtime Settings =====
+    runtime_table = Table("Setting", "Status", title="Runtime Settings", box=box.SIMPLE_HEAD)
+    debug_status = "[green]ON[/green]" if debug_mode_container.get("debug") else "[dim]OFF[/dim]"
+    runtime_table.add_row("Debug Mode", debug_status)
+    logging_status = "[green]ON[/green]" if chat_manager.markdown_logger else "[dim]OFF[/dim]"
+    runtime_table.add_row("Conversation Logging", logging_status)
+    approve_labels = {"safe": "SAFE", "accept_edits": "ACCEPT EDITS", "danger": "DANGER"}
+    approve_colors = {"safe": "green", "accept_edits": "yellow", "danger": "red"}
+    approve_mode = chat_manager.approve_mode
+    approve_color = approve_colors.get(approve_mode, "white")
+    runtime_table.add_row("Approval Mode", f"[{approve_color}]{approve_labels.get(approve_mode, approve_mode.upper())}[/{approve_color}]")
+    console.print(runtime_table)
+
+    # ===== Provider Settings =====
+    console.print()
+    provider_table = Table("Provider", "Model", "$ in/out", "API Key", title="Providers", box=box.SIMPLE_HEAD)
+
+    active_provider = config_data.get("LAST_PROVIDER", "Not set").upper()
+    provider_table.add_row("[green]Active[/green]", f"[green]{active_provider}[/green]", "", "")
+
+    def fmt(v, max_len=35):
+        return v[:max_len-3] + "..." if len(v) > max_len else v
+
+    # Local provider
+    local_model = config_data.get("LOCAL_MODEL_PATH", "Not set")
+    provider_table.add_row("Local", fmt(local_model), "N/A", "N/A")
+
+    # API providers
+    for provider in ["OpenRouter", "GLM", "OpenAI", "Gemini", "MiniMax", "Anthropic", "Kimi"]:
+        model = config_data.get(f"{provider.upper()}_MODEL", "Not set")
+        key = config_data.get(f"{provider.upper()}_API_KEY", "")
+        key_status = "[green]✓[/green]" if key else "[red]✗[/red]"
+
+        # Check for model-specific pricing
+        model_prices = config_data.get("MODEL_PRICES", {})
+        if model and model in model_prices:
+            cost_in = model_prices[model].get("cost_in", 0)
+            cost_out = model_prices[model].get("cost_out", 0)
+            if cost_in > 0 or cost_out > 0:
+                cost_str = f"${cost_in:.2f}/${cost_out:.2f}"
+            else:
+                cost_str = "Not set"
+        else:
+            cost_str = "Not set"
+
+        provider_table.add_row(provider, fmt(model), cost_str, key_status)
+
+    console.print(provider_table)
+
+    # ===== Quick Commands Reference =====
+    console.print()
+    help_text = """[bold #5F9EA0]Commands:[/bold #5F9EA0] [bold #5F9EA0]/provider[/bold #5F9EA0] <name>  [bold #5F9EA0]/model[/bold #5F9EA0] <path>  [bold #5F9EA0]/key[/bold #5F9EA0] <key>
+[#5F9EA0]         :[/#5F9EA0] [bold #5F9EA0]/usage[/bold #5F9EA0] [provider] [in|out] <$>  [bold #5F9EA0]/config[/bold #5F9EA0]"""
+    console.print(Panel(help_text, title="[#5F9EA0]Quick Reference[/#5F9EA0]"))
+    console.print()

package/src/ui/loader.py

@@ -0,0 +1,284 @@
+"""Animated terminal effects for bone-agent — loading spinners, progress bars, and visual flair."""
+
+import time
+import random
+import threading
+from rich.console import Console
+from rich.text import Text
+from rich.live import Live
+from rich.layout import Layout
+from rich.panel import Panel
+from rich.table import Table
+from rich.align import Align
+
+console = Console()
+
+
+# ── ASCII art logo ──────────────────────────────────────────────────────────
+
+BONE_AGENT_LOGO = r"""
+   ╦ ╦┌─┐┌┐ ╔╦╗┬┬  ┌─┐
+   ║║║├┤ ├┴┐ ║ ││  ├┤
+   ╚╩╝└─┘└─┘ ╩ ┴┴─┘└─┘
+"""
+
+SPINNER_FRAMES = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"]
+DOT_FRAMES = ["   ", ".  ", ".. ", "..."]
+WAVE_FRAMES = ["▁", "▂", "▃", "▄", "▅", "▆", "▇", "█", "▇", "▆", "▅", "▄", "▃", "▂", "▁"]
+BAR_FRAMES = ["◐", "◓", "◑", "◒"]
+
+
+def _wave_string(text: str, frame_idx: int, color: str = "#5F9EA0") -> Text:
+    """Create a wave-animated string where each character oscillates vertically."""
+    result = Text()
+    n = len(WAVE_FRAMES)
+    for i, ch in enumerate(text):
+        offset = (frame_idx + i * 2) % n
+        bar_char = WAVE_FRAMES[offset]
+        # Fade intensity based on wave position (center = bright)
+        intensity = abs(offset - n // 2) / (n // 2)
+        if ch == " ":
+            result.append(" ")
+        else:
+            result.append(bar_char, style=color)
+    return result
+
+
+def display_intro_animation(provider: str = "", model: str = ""):
+    """Play a cinematic intro animation on startup.
+
+    Shows the bone-agent logo with a typing effect, a wave animation underneath,
+    and provider/model info fading in.
+    """
+    logo_lines = BONE_AGENT_LOGO.strip("\n").split("\n")
+
+    try:
+        with Live(console=console, transient=False, refresh_per_second=24) as live:
+            # Phase 1: Logo reveal (typewriter effect)
+            revealed_lines = []
+            for line_idx, line in enumerate(logo_lines):
+                revealed_lines.append("")
+                for ch_idx, ch in enumerate(line):
+                    revealed_lines[line_idx] = line[: ch_idx + 1]
+                    layout = Layout()
+                    logo_text = Text("\n".join(revealed_lines), style="bold #5F9EA0")
+                    layout.update(Align.center(Panel(
+                        logo_text,
+                        border_style="grey30",
+                        padding=(1, 4),
+                        subtitle=Text("  ", style="dim"),
+                    )))
+                    live.update(layout)
+                    time.sleep(0.008)
+                time.sleep(0.04)
+
+            # Phase 2: Wave animation beneath the logo (runs for ~2 seconds)
+            wave_line = "━" * 42
+            start = time.time()
+            frame = 0
+            while time.time() - start < 1.8:
+                wave = _wave_string(wave_line, frame, color="#3a7ca5")
+                layout = Layout()
+                full = Text()
+                full.append("\n".join(logo_lines) + "\n", style="bold #5F9EA0")
+                full.append(wave)
+                layout.update(Align.center(Panel(
+                    full,
+                    border_style="grey30",
+                    padding=(1, 4),
+                )))
+                live.update(layout)
+                frame += 1
+                time.sleep(0.05)
+
+            # Phase 3: Show tagline + provider info
+            tagline = Text("  local-first  ·  agent-powered  ·  terminal-native", style="dim grey60")
+            if provider and model:
+                info = Text.assemble(
+                    ("  ", ""),
+                    (f"● {provider.upper()} ", "bold #5F9EA0"),
+                    (f"{model}", "grey70"),
+                    style="",
+                )
+            else:
+                info = Text("")
+
+            layout = Layout()
+            full = Text()
+            full.append("\n".join(logo_lines), style="bold #5F9EA0")
+            full.append("\n")
+            full.append(wave_line, style="#3a7ca5")
+            layout.update(Align.center(Panel(
+                Align.center(
+                    Table.grid(padding=(0, 0)),
+                ),
+                border_style="grey30",
+                padding=(1, 4),
+                subtitle=tagline,
+            )))
+            live.update(layout)
+            time.sleep(0.6)
+
+            # Final frame — static logo with tagline
+            final = Text()
+            final.append("\n".join(logo_lines), style="bold #5F9EA0")
+            final.append("\n")
+            final.append(wave_line, style="#3a7ca5")
+            layout.update(Align.center(Panel(
+                final,
+                border_style="grey30",
+                padding=(1, 4),
+                subtitle=tagline,
+            )))
+            live.update(layout)
+            time.sleep(0.3)
+    except Exception:
+        # Fallback: if Live fails (e.g. non-TTY), just print static logo
+        console.print(Panel(
+            Text(BONE_AGENT_LOGO.strip("\n"), style="bold #5F9EA0"),
+            border_style="grey30",
+            subtitle="  local-first  ·  agent-powered  ·  terminal-native",
+        ))
+
+
+# ── Spinner context manager ─────────────────────────────────────────────────
+
+class Spinner:
+    """A rich-based status spinner for long-running operations.
+
+    Usage:
+        with Spinner("Indexing files..."):
+            do_expensive_work()
+    """
+
+    def __init__(self, message: str, style: str = "#5F9EA0", done_message: str = "done"):
+        self.message = message
+        self.style = style
+        self.done_message = done_message
+        self._stop = threading.Event()
+        self._thread = None
+
+    def _spin(self):
+        frame = 0
+        n = len(SPINNER_FRAMES)
+        with Live(console=console, transient=True, refresh_per_second=12) as live:
+            while not self._stop.is_set():
+                spinner = SPINNER_FRAMES[frame % n]
+                live.update(Text(f"  {spinner} {self.message}", style=self.style))
+                frame += 1
+                self._stop.wait(0.08)
+
+    def __enter__(self):
+        self._stop.clear()
+        self._thread = threading.Thread(target=self._spin, daemon=True)
+        self._thread.start()
+        return self
+
+    def __exit__(self, *args):
+        self._stop.set()
+        if self._thread:
+            self._thread.join(timeout=1)
+        console.print(f"  ✓ {self.message} {self.done_message}", style="dim green")
+
+    def ok(self, msg: str = ""):
+        """Mark as done with a custom message."""
+        self.done_message = msg
+
+
+# ── Progress bar ────────────────────────────────────────────────────────────
+
+class ProgressBar:
+    """A lightweight animated progress bar for terminal output.
+
+    Usage:
+        with ProgressBar("Loading", total=100) as bar:
+            for i in range(100):
+                bar.update(i + 1)
+    """
+
+    def __init__(self, label: str = "", total: int = 100, width: int = 30,
+                 fill: str = "█", empty: str = "░", color: str = "#5F9EA0"):
+        self.label = label
+        self.total = total
+        self.width = width
+        self.fill = fill
+        self.empty = empty
+        self.color = color
+        self.current = 0
+        self._stop = threading.Event()
+
+    def update(self, value: int):
+        self.current = min(value, self.total)
+
+    def _render(self) -> Text:
+        pct = self.current / self.total if self.total else 0
+        filled = int(self.width * pct)
+        bar = self.fill * filled + self.empty * (self.width - filled)
+        result = Text()
+        if self.label:
+            result.append(f"  {self.label} ", style="dim")
+        result.append(bar, style=self.color)
+        result.append(f" {pct:>5.1%}", style="dim grey70")
+        return result
+
+    def _animate(self):
+        with Live(console=console, transient=True, refresh_per_second=8) as live:
+            while not self._stop.is_set():
+                live.update(self._render())
+                self._stop.wait(0.1)
+
+    def __enter__(self):
+        self._stop.clear()
+        self._thread = threading.Thread(target=self._animate, daemon=True)
+        self._thread.start()
+        return self
+
+    def __exit__(self, *args):
+        self._stop.set()
+        if self._thread:
+            self._thread.join(timeout=1)
+        # Print final state
+        console.print(self._render())
+
+
+# ── Scan-line / matrix effect (decorative) ─────────────────────────────────
+
+def matrix_rain(duration: float = 1.5, cols: int = 60, rows: int = 8, chars: str = "01アイウエオカキクケコ"):
+    """Print a brief Matrix-style rain effect to the terminal.
+
+    Purely decorative — great for transitions between sections.
+    """
+    random.seed()
+    try:
+        with Live(console=console, transient=True, refresh_per_second=16) as live:
+            start = time.time()
+            # Each column has a falling "drop" at a random row
+            drops = [random.randint(0, rows - 1) for _ in range(cols)]
+            speeds = [random.uniform(0.3, 1.0) for _ in range(cols)]
+            phases = [random.random() * duration for _ in range(cols)]
+
+            while time.time() - start < duration:
+                grid = []
+                t = time.time() - start
+                for r in range(rows):
+                    row = Text()
+                    for c in range(cols):
+                        # Determine if this cell is "active"
+                        drop_pos = drops[c] + int((t - phases[c]) * speeds[c] * rows / duration)
+                        drop_pos = drop_pos % (rows + 4)  # wrap around
+                        dist = drop_pos - r
+                        if 0 <= dist <= 3:
+                            ch = random.choice(chars)
+                            if dist == 0:
+                                row.append(ch, style="bold white")
+                            else:
+                                fade = f"#{max(0, 0x10):02x}{max(0, 0x40 + (3 - dist) * 0x20):02x}{max(0, 0x10):02x}"
+                                row.append(ch, style=fade)
+                        else:
+                            row.append(" ")
+                    grid.append(row)
+
+                live.update(Text("\n").join(grid))
+                time.sleep(0.04)
+    except Exception:
+        pass  # silently skip if terminal doesn't support it