code-puppy 0.0.214__py3-none-any.whl → 0.0.366__py3-none-any.whl

code_puppy/plugins/claude_code_oauth/SETUP.md

@@ -0,0 +1,93 @@
+# Claude Code OAuth Plugin Setup Guide
+
+This guide walks you through using the Claude Code OAuth plugin inside Code Puppy.
+
+## Quick Start
+
+1. Ensure the plugin files live under `code_puppy/plugins/claude_code_oauth/`
+2. Restart Code Puppy so it loads the plugin
+3. Run `/claude-code-auth` and follow the prompts
+
+## Why No Client Registration?
+
+Anthropic exposes a shared **public client** (`claude-cli`) for command-line tools. That means:
+- No client secret is needed
+- Everyone authenticates through Claude Console
+- Security is enforced with PKCE and per-user tokens
+
+## Authentication Flow
+
+1. Call `/claude-code-auth`
+2. Your browser opens the Claude OAuth consent flow at `https://claude.ai/oauth/authorize`
+3. Sign in (or pick an account) and approve the "Claude CLI" access request
+4. The browser closes automatically after the redirect is captured
+5. Tokens are stored locally at `~/.code_puppy/claude_code_oauth.json`
+6. Available Claude Code models are fetched and added to `extra_models.json`
+
+## Commands Recap
+
+- `/claude-code-auth` – Authenticate and sync models
+- `/claude-code-status` – Show auth status, expiry, configured models
+- `/claude-code-logout` – Remove tokens and any models added by the plugin
+
+## Configuration Defaults
+
+`config.py` ships with values aligned to llxprt-code:
+
+```python
+CLAUDE_CODE_OAUTH_CONFIG = {
+    "auth_url": "https://claude.ai/oauth/authorize",
+    "token_url": "https://claude.ai/api/oauth/token",
+    "api_base_url": "https://api.anthropic.com",
+    "client_id": "9d1c250a-e61b-44d9-88ed-5944d1962f5e",
+    "scope": "org:create_api_key user:profile user:inference",
+    "redirect_host": "http://localhost",
+    "redirect_path": "callback",
+    "callback_port_range": (8765, 8795),
+    "callback_timeout": 180,
+    "prefix": "claude-code-",
+    "default_context_length": 200000,
+    "api_key_env_var": "CLAUDE_CODE_ACCESS_TOKEN",
+}
+```
+
+Change these only if Anthropic updates their endpoints or scopes.
+
+## After Authentication
+
+- Models appear in `~/.code_puppy/extra_models.json` with the `claude-code-` prefix
+- The environment variable `CLAUDE_CODE_ACCESS_TOKEN` is used by those models
+- `/claude-code-status` shows token expiry when the API provides it
+
+## Troubleshooting Tips
+
+- **Browser did not open** – Copy the displayed URL into your browser manually
+- **Invalid code** – The code expires quickly; generate a new one in Claude Console
+- **State mismatch** – Rare, but rerun `/claude-code-auth` if the browser reports a mismatch
+- **No models added** – Your account might lack Claude Code access; tokens are still stored for later use
+
+## Files Created
+
+```
+~/.code_puppy/
+├── claude_code_oauth.json   # OAuth tokens (0600 permissions)
+└── extra_models.json        # Extended model registry
+```
+
+## Manual Testing
+
+Run the helper script for sanity checks:
+
+```bash
+python code_puppy/plugins/claude_code_oauth/test_plugin.py
+```
+
+It verifies imports, configuration values, and filesystem expectations without hitting the Anthropic API.
+
+## Security Notes
+
+- Tokens are stored locally and never transmitted elsewhere
+- PKCE protects the flow even without a client secret
+- HTTPS endpoints are enforced for all requests
+
+Enjoy hacking with Claude Code straight from Code Puppy! 🐶💻

code_puppy/plugins/claude_code_oauth/__init__.py

@@ -0,0 +1,6 @@
+"""
+Claude Code OAuth Plugin for Code Puppy
+
+This plugin provides OAuth authentication for Claude Code and automatically
+adds available models to the extra_models.json configuration.
+"""

code_puppy/plugins/claude_code_oauth/config.py

@@ -0,0 +1,50 @@
+from pathlib import Path
+from typing import Any, Dict
+
+from code_puppy import config
+
+# Claude Code OAuth configuration
+CLAUDE_CODE_OAUTH_CONFIG: Dict[str, Any] = {
+    # OAuth endpoints inferred from official Claude Code OAuth flow
+    "auth_url": "https://claude.ai/oauth/authorize",
+    "token_url": "https://console.anthropic.com/v1/oauth/token",
+    "api_base_url": "https://api.anthropic.com",
+    # OAuth client configuration observed in Claude Code CLI flow
+    "client_id": "9d1c250a-e61b-44d9-88ed-5944d1962f5e",
+    "scope": "org:create_api_key user:profile user:inference",
+    # Callback handling (we host a localhost callback to capture the redirect)
+    "redirect_host": "http://localhost",
+    "redirect_path": "callback",
+    "callback_port_range": (8765, 8795),
+    "callback_timeout": 180,
+    # Console redirect fallback (for manual flows, if needed)
+    "console_redirect_uri": "https://console.anthropic.com/oauth/code/callback",
+    # Local configuration (uses XDG_DATA_HOME)
+    "token_storage": None,  # Set dynamically in get_token_storage_path()
+    # Model configuration
+    "prefix": "claude-code-",
+    "default_context_length": 200000,
+    "api_key_env_var": "CLAUDE_CODE_ACCESS_TOKEN",
+    "anthropic_version": "2023-06-01",
+}
+
+
+def get_token_storage_path() -> Path:
+    """Get the path for storing OAuth tokens (uses XDG_DATA_HOME)."""
+    data_dir = Path(config.DATA_DIR)
+    data_dir.mkdir(parents=True, exist_ok=True, mode=0o700)
+    return data_dir / "claude_code_oauth.json"
+
+
+def get_config_dir() -> Path:
+    """Get the Code Puppy configuration directory (uses XDG_CONFIG_HOME)."""
+    config_dir = Path(config.CONFIG_DIR)
+    config_dir.mkdir(parents=True, exist_ok=True, mode=0o700)
+    return config_dir
+
+
+def get_claude_models_path() -> Path:
+    """Get the path to the dedicated claude_models.json file (uses XDG_DATA_HOME)."""
+    data_dir = Path(config.DATA_DIR)
+    data_dir.mkdir(parents=True, exist_ok=True, mode=0o700)
+    return data_dir / "claude_models.json"

code_puppy/plugins/claude_code_oauth/register_callbacks.py

@@ -0,0 +1,308 @@
+"""
+Claude Code OAuth Plugin for Code Puppy.
+"""
+
+from __future__ import annotations
+
+import logging
+import threading
+import time
+from http.server import BaseHTTPRequestHandler, HTTPServer
+from typing import Any, Dict, List, Optional, Tuple
+from urllib.parse import parse_qs, urlparse
+
+from code_puppy.callbacks import register_callback
+from code_puppy.config import set_model_name
+from code_puppy.messaging import emit_error, emit_info, emit_success, emit_warning
+
+from ..oauth_puppy_html import oauth_failure_html, oauth_success_html
+from .config import CLAUDE_CODE_OAUTH_CONFIG, get_token_storage_path
+from .utils import (
+    OAuthContext,
+    add_models_to_extra_config,
+    assign_redirect_uri,
+    build_authorization_url,
+    exchange_code_for_tokens,
+    fetch_claude_code_models,
+    load_claude_models_filtered,
+    load_stored_tokens,
+    prepare_oauth_context,
+    remove_claude_code_models,
+    save_tokens,
+)
+
+logger = logging.getLogger(__name__)
+
+
+class _OAuthResult:
+    def __init__(self) -> None:
+        self.code: Optional[str] = None
+        self.state: Optional[str] = None
+        self.error: Optional[str] = None
+
+
+class _CallbackHandler(BaseHTTPRequestHandler):
+    result: _OAuthResult
+    received_event: threading.Event
+
+    def do_GET(self) -> None:  # noqa: N802
+        logger.info("Callback received: path=%s", self.path)
+        parsed = urlparse(self.path)
+        params: Dict[str, List[str]] = parse_qs(parsed.query)
+
+        code = params.get("code", [None])[0]
+        state = params.get("state", [None])[0]
+
+        if code and state:
+            self.result.code = code
+            self.result.state = state
+            success_html = oauth_success_html(
+                "Claude Code",
+                "You're totally synced with Claude Code now!",
+            )
+            self._write_response(200, success_html)
+        else:
+            self.result.error = "Missing code or state"
+            failure_html = oauth_failure_html(
+                "Claude Code",
+                "Missing code or state parameter 🥺",
+            )
+            self._write_response(400, failure_html)
+
+        self.received_event.set()
+
+    def log_message(self, format: str, *args: Any) -> None:  # noqa: A003
+        return
+
+    def _write_response(self, status: int, body: str) -> None:
+        self.send_response(status)
+        self.send_header("Content-Type", "text/html; charset=utf-8")
+        self.end_headers()
+        self.wfile.write(body.encode("utf-8"))
+
+
+def _start_callback_server(
+    context: OAuthContext,
+) -> Optional[Tuple[HTTPServer, _OAuthResult, threading.Event]]:
+    port_range = CLAUDE_CODE_OAUTH_CONFIG["callback_port_range"]
+
+    for port in range(port_range[0], port_range[1] + 1):
+        try:
+            server = HTTPServer(("localhost", port), _CallbackHandler)
+            assign_redirect_uri(context, port)
+            result = _OAuthResult()
+            event = threading.Event()
+            _CallbackHandler.result = result
+            _CallbackHandler.received_event = event
+
+            def run_server() -> None:
+                with server:
+                    server.serve_forever()
+
+            threading.Thread(target=run_server, daemon=True).start()
+            return server, result, event
+        except OSError:
+            continue
+
+    emit_error("Could not start OAuth callback server; all candidate ports are in use")
+    return None
+
+
+def _await_callback(context: OAuthContext) -> Optional[str]:
+    timeout = CLAUDE_CODE_OAUTH_CONFIG["callback_timeout"]
+
+    started = _start_callback_server(context)
+    if not started:
+        return None
+
+    server, result, event = started
+    redirect_uri = context.redirect_uri
+    if not redirect_uri:
+        emit_error("Failed to assign redirect URI for OAuth flow")
+        server.shutdown()
+        return None
+
+    auth_url = build_authorization_url(context)
+
+    try:
+        import webbrowser
+
+        from code_puppy.tools.common import should_suppress_browser
+
+        if should_suppress_browser():
+            emit_info(
+                "[HEADLESS MODE] Would normally open browser for Claude Code OAuth…"
+            )
+            emit_info(f"In normal mode, would visit: {auth_url}")
+        else:
+            emit_info("Opening browser for Claude Code OAuth…")
+            webbrowser.open(auth_url)
+            emit_info(f"If it doesn't open automatically, visit: {auth_url}")
+    except Exception as exc:  # pragma: no cover
+        if not should_suppress_browser():
+            emit_warning(f"Failed to open browser automatically: {exc}")
+            emit_info(f"Please open the URL manually: {auth_url}")
+
+    emit_info(f"Listening for callback on {redirect_uri}")
+    emit_info(
+        "If Claude redirects you to the console callback page, copy the full URL "
+        "and paste it back into Code Puppy."
+    )
+
+    if not event.wait(timeout=timeout):
+        emit_error("OAuth callback timed out. Please try again.")
+        server.shutdown()
+        return None
+
+    server.shutdown()
+
+    if result.error:
+        emit_error(f"OAuth callback error: {result.error}")
+        return None
+
+    if result.state != context.state:
+        emit_error("State mismatch detected; aborting authentication.")
+        return None
+
+    return result.code
+
+
+def _custom_help() -> List[Tuple[str, str]]:
+    return [
+        (
+            "claude-code-auth",
+            "Authenticate with Claude Code via OAuth and import available models",
+        ),
+        (
+            "claude-code-status",
+            "Check Claude Code OAuth authentication status and configured models",
+        ),
+        ("claude-code-logout", "Remove Claude Code OAuth tokens and imported models"),
+    ]
+
+
+def _reload_current_agent() -> None:
+    """Reload the current agent so new auth tokens are picked up immediately."""
+    try:
+        from code_puppy.agents import get_current_agent
+
+        current_agent = get_current_agent()
+        if current_agent is None:
+            logger.debug("No current agent to reload")
+            return
+
+        # JSON agents may need to refresh their config before reload
+        if hasattr(current_agent, "refresh_config"):
+            try:
+                current_agent.refresh_config()
+            except Exception:
+                # Non-fatal, continue to reload
+                pass
+
+        current_agent.reload_code_generation_agent()
+        emit_info("Active agent reloaded with new authentication")
+    except Exception as e:
+        emit_warning(f"Authentication succeeded but agent reload failed: {e}")
+        logger.exception("Failed to reload agent after authentication")
+
+
+def _perform_authentication() -> None:
+    context = prepare_oauth_context()
+    code = _await_callback(context)
+    if not code:
+        return
+
+    emit_info("Exchanging authorization code for tokens…")
+    tokens = exchange_code_for_tokens(code, context)
+    if not tokens:
+        emit_error("Token exchange failed. Please retry the authentication flow.")
+        return
+
+    if not save_tokens(tokens):
+        emit_error(
+            "Tokens retrieved but failed to save locally. Check file permissions."
+        )
+        return
+
+    emit_success("Claude Code OAuth authentication successful!")
+
+    access_token = tokens.get("access_token")
+    if not access_token:
+        emit_warning("No access token returned; skipping model discovery.")
+        return
+
+    emit_info("Fetching available Claude Code models…")
+    models = fetch_claude_code_models(access_token)
+    if not models:
+        emit_warning(
+            "Claude Code authentication succeeded but no models were returned."
+        )
+        return
+
+    emit_info(f"Discovered {len(models)} models: {', '.join(models)}")
+    if add_models_to_extra_config(models):
+        emit_success(
+            "Claude Code models added to your configuration. Use the `claude-code-` prefix!"
+        )
+
+    # Reload the current agent so the new auth token is picked up immediately
+    _reload_current_agent()
+
+
+def _handle_custom_command(command: str, name: str) -> Optional[bool]:
+    if not name:
+        return None
+
+    if name == "claude-code-auth":
+        emit_info("Starting Claude Code OAuth authentication…")
+        tokens = load_stored_tokens()
+        if tokens and tokens.get("access_token"):
+            emit_warning(
+                "Existing Claude Code tokens found. Continuing will overwrite them."
+            )
+        _perform_authentication()
+        set_model_name("claude-code-claude-opus-4-5-20251101")
+        return True
+
+    if name == "claude-code-status":
+        tokens = load_stored_tokens()
+        if tokens and tokens.get("access_token"):
+            emit_success("Claude Code OAuth: Authenticated")
+            expires_at = tokens.get("expires_at")
+            if expires_at:
+                remaining = max(0, int(expires_at - time.time()))
+                hours, minutes = divmod(remaining // 60, 60)
+                emit_info(f"Token expires in ~{hours}h {minutes}m")
+
+            claude_models = [
+                name
+                for name, cfg in load_claude_models_filtered().items()
+                if cfg.get("oauth_source") == "claude-code-plugin"
+            ]
+            if claude_models:
+                emit_info(f"Configured Claude Code models: {', '.join(claude_models)}")
+            else:
+                emit_warning("No Claude Code models configured yet.")
+        else:
+            emit_warning("Claude Code OAuth: Not authenticated")
+            emit_info("Run /claude-code-auth to begin the browser sign-in flow.")
+        return True
+
+    if name == "claude-code-logout":
+        token_path = get_token_storage_path()
+        if token_path.exists():
+            token_path.unlink()
+            emit_info("Removed Claude Code OAuth tokens")
+
+        removed = remove_claude_code_models()
+        if removed:
+            emit_info(f"Removed {removed} Claude Code models from configuration")
+
+        emit_success("Claude Code logout complete")
+        return True
+
+    return None
+
+
+register_callback("custom_command_help", _custom_help)
+register_callback("custom_command", _handle_custom_command)