code-puppy-core-plugins 0.0.2__py3-none-any.whl

code_puppy_core_plugins/context_indicator/register_callbacks.py

@@ -0,0 +1,249 @@
+"""Register callbacks for the ``context_indicator`` plugin.
+
+Hooks:
+
+* ``startup`` — wraps the bottom bar's token-summary writer so the status
+  row shows a colored circle reflecting current context-window usage,
+  e.g. ``80.7k/500k tokens (16%)`` gains a green/yellow/red badge.
+* ``custom_command`` / ``custom_command_help`` — exposes ``/context`` for a
+  detailed token-usage breakdown.
+
+Idempotent: re-installing the status patch is a no-op.
+"""
+
+from __future__ import annotations
+
+from typing import List, Tuple
+
+from code_puppy.callbacks import register_callback
+from code_puppy.token_usage import (
+    ContextUsage,
+    get_current_usage,
+)
+
+_COMMAND_NAME = "context"
+_PATCH_ATTR = "_context_indicator_original"
+
+
+# ---------------------------------------------------------------------------
+# Messaging helpers (lazy-imported to dodge circular imports at boot)
+# ---------------------------------------------------------------------------
+def _emit_info(message: str) -> None:
+    from code_puppy.messaging import emit_info
+
+    emit_info(message)
+
+
+def _emit_system_message(message: str) -> None:
+    from code_puppy.messaging import emit_system_message
+
+    emit_system_message(message)
+
+
+def _emit_error(message: str) -> None:
+    from code_puppy.messaging import emit_error
+
+    emit_error(message)
+
+
+# ---------------------------------------------------------------------------
+# Status-line patching
+# ---------------------------------------------------------------------------
+def _decorate_status(info: str) -> str:
+    """Prepend the usage light to a non-empty token summary.
+
+    Empty strings pass through untouched: those are "clear the row" calls
+    and must stay empty so the idle prompt isn't haunted by a lone circle.
+    """
+    if not info:
+        return info
+    usage = get_current_usage()
+    if usage is None:
+        return info
+    return f"{usage.indicator} {info}"
+
+
+def _install_status_patch() -> None:
+    """Monkey-patch ``_compaction.update_spinner_context`` once.
+
+    The token summary on the bottom bar's status row is written by
+    ``agents._compaction`` through its import-time binding of
+    ``update_spinner_context`` — so that binding (not the ``spinner``
+    module attribute) is the seam we wrap. The badge rides the status
+    row, next to the numbers it describes, instead of the prompt line.
+    """
+    from code_puppy.agents import _compaction
+
+    if getattr(_compaction, _PATCH_ATTR, None) is not None:
+        return  # Already patched
+
+    original = _compaction.update_spinner_context
+    setattr(_compaction, _PATCH_ATTR, original)
+
+    def patched(info: str) -> None:
+        original(_decorate_status(info))
+
+    _compaction.update_spinner_context = patched
+
+
+_LEGEND_TEXT = (
+    "Context indicator: 🟢 <30%   🟡 30–<65%   🔴 ≥65%  "
+    "(use /context for a detailed breakdown)"
+)
+
+
+def _announce_legend() -> None:
+    try:
+        _emit_system_message(_LEGEND_TEXT)
+    except Exception:
+        # Never crash startup over a banner line — fail gracefully per plugin rules.
+        pass
+
+
+def _on_startup() -> None:
+    try:
+        _install_status_patch()
+    except Exception as exc:
+        _emit_error(f"context_indicator: failed to install status patch — {exc}")
+    _announce_legend()
+
+
+# ---------------------------------------------------------------------------
+# /context slash command
+# ---------------------------------------------------------------------------
+def _custom_help() -> List[Tuple[str, str]]:
+    return [
+        (
+            _COMMAND_NAME,
+            "Show context-window usage (tokens used vs. model capacity)",
+        )
+    ]
+
+
+# Bar glyphs — kept as module constants so the legend stays DRY.
+_BAR_GLYPH_OVERHEAD = "▒"  # system prompt + tool schema baseline
+_BAR_GLYPH_MESSAGES = "█"  # conversation tokens
+_BAR_GLYPH_EMPTY = "░"  # unused capacity
+_BAR_GLYPH_THRESHOLD = "┃"  # vertical marker for compaction trigger
+_BAR_WIDTH = 30
+
+
+def _render_usage_bar(usage: ContextUsage, threshold: float) -> str:
+    """Render the ASCII usage bar with three zones + a compaction marker.
+
+    Zones: overhead | messages | empty. The compaction-threshold marker
+    overwrites whichever cell it lands on (cosmetic; the underlying token
+    counts are unchanged).
+    """
+    capacity = max(1, usage.capacity)
+    overhead_cells = min(
+        _BAR_WIDTH, int(round(usage.overhead_tokens / capacity * _BAR_WIDTH))
+    )
+    total_cells = min(
+        _BAR_WIDTH, int(round(usage.total_tokens / capacity * _BAR_WIDTH))
+    )
+    # Guarantee messages cells start strictly after overhead cells.
+    message_cells = max(0, total_cells - overhead_cells)
+    empty_cells = _BAR_WIDTH - overhead_cells - message_cells
+
+    cells = (
+        [_BAR_GLYPH_OVERHEAD] * overhead_cells
+        + [_BAR_GLYPH_MESSAGES] * message_cells
+        + [_BAR_GLYPH_EMPTY] * empty_cells
+    )
+
+    threshold_clamped = max(0.0, min(1.0, threshold))
+    marker_idx = min(_BAR_WIDTH - 1, int(round(threshold_clamped * _BAR_WIDTH)))
+    cells[marker_idx] = _BAR_GLYPH_THRESHOLD
+    return "".join(cells)
+
+
+def _get_compaction_threshold() -> float:
+    """Fetch compaction threshold defensively; fall back to 0.85 on any error."""
+    try:
+        from code_puppy.config import get_compaction_threshold
+
+        return float(get_compaction_threshold())
+    except Exception:
+        return 0.85
+
+
+def _format_overhead_breakdown(usage: ContextUsage) -> str:
+    """Render the per-bucket overhead breakdown.
+
+    Each non-zero bucket gets its own indented line. Zero-valued buckets are
+    hidden so users with no MCP servers / no AGENTS.md aren't staring at
+    noise.  We *always* show the aggregate ``Overhead`` line so the report
+    structure stays consistent.
+    """
+    # (label, token_count) — order matters for readability.
+    rows = (
+        ("System prompt", usage.system_prompt_tokens),
+        ("AGENTS.md     ", usage.agents_md_tokens),
+        ("Kennel memory", usage.kennel_memory_tokens),
+        ("Pydantic tools", usage.pydantic_tools_tokens),
+        ("MCP toolsets  ", usage.mcp_tokens),
+    )
+    lines = [
+        f"    └─ {label}: {tokens:,} tokens" for label, tokens in rows if tokens > 0
+    ]
+    return "\n".join(lines)
+
+
+def _format_usage_report(usage: ContextUsage) -> str:
+    threshold = _get_compaction_threshold()
+    bar = _render_usage_bar(usage, threshold)
+    legend = (
+        f"  Legend   : {_BAR_GLYPH_OVERHEAD} overhead  "
+        f"{_BAR_GLYPH_MESSAGES} messages  "
+        f"{_BAR_GLYPH_EMPTY} free  "
+        f"{_BAR_GLYPH_THRESHOLD} compaction @ {threshold:.0%}"
+    )
+    breakdown = _format_overhead_breakdown(usage)
+    breakdown_block = f"\n{breakdown}" if breakdown else ""
+    return (
+        f"{usage.indicator} Context usage: {usage.percent:.1f}%\n"
+        f"  [{bar}]\n"
+        f"{legend}\n"
+        f"  Messages : {usage.used_tokens:,} tokens\n"
+        f"  Overhead : {usage.overhead_tokens:,} tokens (system prompt + AGENTS.md + kennel memory + tools + MCP)"
+        f"{breakdown_block}\n"
+        f"  Total    : {usage.total_tokens:,} / {usage.capacity:,} tokens\n"
+        f"  Buckets  : 🟢 <30%   🟡 30–65%   🔴 ≥65%"
+    )
+
+
+def _handle_context_command(command: str) -> bool:
+    usage = get_current_usage()
+    if usage is None:
+        _emit_info("🐶 No context info yet — load an agent and send a message first.")
+        return True
+    _emit_info(_format_usage_report(usage))
+    return True
+
+
+def _handle_custom_command(command: str, name: str):
+    if name != _COMMAND_NAME:
+        return None
+    return _handle_context_command(command)
+
+
+# ---------------------------------------------------------------------------
+# Registration
+# ---------------------------------------------------------------------------
+register_callback("startup", _on_startup)
+register_callback("custom_command", _handle_custom_command)
+register_callback("custom_command_help", _custom_help)
+
+
+__all__ = [
+    "_announce_legend",
+    "_custom_help",
+    "_decorate_status",
+    "_format_overhead_breakdown",
+    "_format_usage_report",
+    "_handle_context_command",
+    "_handle_custom_command",
+    "_install_status_patch",
+    "_on_startup",
+]

code_puppy_core_plugins/context_indicator/usage.py

@@ -0,0 +1,40 @@
+"""Compatibility shim for the relocated token-usage module.
+
+Token accounting moved from this plugin into the core module
+``code_puppy.token_usage`` so that more than one consumer (the
+``context_indicator`` plugin, the ``statusline`` plugin, and the ``herdr``
+integration) can share it without cross-plugin imports.
+
+This module remains as a thin re-export so existing import sites and tests
+that reference ``code_puppy_core_plugins.context_indicator.usage`` keep working
+and observe the *same* objects as the core module (identity is preserved).
+New code should import from :mod:`code_puppy.token_usage` directly.
+"""
+
+from __future__ import annotations
+
+from code_puppy.token_usage import (
+    GREEN_CIRCLE,
+    GREEN_THRESHOLD,
+    RED_CIRCLE,
+    YELLOW_CIRCLE,
+    YELLOW_THRESHOLD,
+    ContextUsage,
+    OverheadBreakdown,
+    compute_overhead_breakdown,
+    get_current_usage,
+    pick_indicator,
+)
+
+__all__ = [
+    "ContextUsage",
+    "OverheadBreakdown",
+    "get_current_usage",
+    "compute_overhead_breakdown",
+    "pick_indicator",
+    "GREEN_THRESHOLD",
+    "YELLOW_THRESHOLD",
+    "GREEN_CIRCLE",
+    "YELLOW_CIRCLE",
+    "RED_CIRCLE",
+]

code_puppy_core_plugins/copilot_auth/__init__.py

@@ -0,0 +1,11 @@
+"""GitHub Copilot auth plugin for Code Puppy.
+
+Authenticates with GitHub (or GitHub Enterprise) via the browser-based
+Device Flow and exchanges the resulting OAuth token for a short-lived
+Copilot session token to access the Copilot API.
+
+Commands:
+- ``/copilot-login``  — authenticate via browser
+- ``/copilot-status`` — show auth & model status
+- ``/copilot-logout`` — remove tokens and registered models
+"""

code_puppy_core_plugins/copilot_auth/config.py

@@ -0,0 +1,93 @@
+"""Configuration constants for the GitHub Copilot auth plugin."""
+
+from __future__ import annotations
+
+from pathlib import Path
+from typing import Any, Dict
+
+from code_puppy import config
+
+# GitHub Copilot auth configuration
+COPILOT_AUTH_CONFIG: Dict[str, Any] = {
+    # Copilot session-token endpoints
+    "github_token_url": "https://api.github.com/copilot_internal/v2/token",
+    # GHE template: replace {host}
+    "ghe_token_url_template": "https://{host}/api/v3/copilot_internal/v2/token",
+    # OpenAI-compatible Copilot chat API (default; overridden per-host at runtime)
+    "api_base_url": "https://api.githubcopilot.com",
+    # Model prefix in Code Puppy
+    "prefix": "copilot-",
+    "default_context_length": 128000,
+    # Headers expected by the Copilot API
+    "editor_version": "JetBrains-IU/2024.3",
+    "editor_plugin_version": "copilot/2.0.0",
+    "copilot_integration_id": "vscode-chat",
+    "openai_intent": "conversation-panel",
+}
+
+# GitHub Device Flow configuration — browser-based auth with no IDE required.
+# Uses the VS Code Copilot extension's public client ID which supports Device Flow.
+DEVICE_FLOW_CONFIG: Dict[str, Any] = {
+    "client_id": "Iv1.b507a08c87ecfe98",
+    "scope": "read:user",
+    # Endpoint templates — {host} is replaced at runtime
+    "device_code_url": "https://{host}/login/device/code",
+    "access_token_url": "https://{host}/login/oauth/access_token",
+    # Polling defaults (server may override via response)
+    "default_poll_interval": 5,
+    "grant_type": "urn:ietf:params:oauth:grant-type:device_code",
+}
+
+# Models known to be available via Copilot (costs are $0 via Copilot).
+# The /copilot-login flow will also try to fetch the live model list.
+DEFAULT_COPILOT_MODELS = [
+    "gpt-4o",
+    "gpt-4.1",
+    "gpt-4.1-mini",
+    "o4-mini",
+    "o3",
+    "claude-opus-4.6",
+    "claude-opus-4",
+    "claude-sonnet-4",
+    "claude-sonnet-4.5",
+    "claude-haiku-4.5",
+    "gemini-2.5-pro",
+    "gemini-2.5-flash",
+]
+
+# Per-model context-length overrides (tokens).
+COPILOT_MODEL_CONTEXT_LENGTHS: Dict[str, int] = {
+    "gpt-4o": 64000,
+    "gpt-4.1": 128000,
+    "gpt-4.1-mini": 128000,
+    "o4-mini": 128000,
+    "o3": 128000,
+    "claude-opus-4.6": 200000,
+    "claude-opus-4": 200000,
+    "claude-sonnet-4": 128000,
+    "claude-sonnet-4.5": 128000,
+    "claude-haiku-4.5": 128000,
+    "gemini-2.5-pro": 128000,
+    "gemini-2.5-flash": 128000,
+}
+
+
+def get_copilot_models_path() -> Path:
+    """Return path for persisted copilot_models.json (XDG_DATA_HOME)."""
+    data_dir = Path(config.DATA_DIR)
+    data_dir.mkdir(parents=True, exist_ok=True, mode=0o700)
+    return data_dir / "copilot_models.json"
+
+
+def get_session_cache_path() -> Path:
+    """Return path for caching the short-lived Copilot session token."""
+    data_dir = Path(config.DATA_DIR)
+    data_dir.mkdir(parents=True, exist_ok=True, mode=0o700)
+    return data_dir / "copilot_session.json"
+
+
+def get_device_token_storage_path() -> Path:
+    """Return path for storing tokens obtained via the GitHub Device Flow."""
+    data_dir = Path(config.DATA_DIR)
+    data_dir.mkdir(parents=True, exist_ok=True, mode=0o700)
+    return data_dir / "copilot_device_tokens.json"