isa-mate 0.2.0__py3-none-any.whl

isa_mate/__init__.py

@@ -0,0 +1,27 @@
+# isA Mate Service
+
+__version__ = "0.2.0"
+
+from .pool_bridge import (
+    PoolManagerBridge,
+    PoolBridgeConfig,
+    MateVisibility,
+    MateTier,
+    MateSpecs,
+    # Backwards compatibility aliases
+    DesktopVisibility,
+    DesktopTier,
+    DesktopSpecs,
+)
+
+__all__ = [
+    "PoolManagerBridge",
+    "PoolBridgeConfig",
+    "MateVisibility",
+    "MateTier",
+    "MateSpecs",
+    # Backwards compatibility aliases
+    "DesktopVisibility",
+    "DesktopTier",
+    "DesktopSpecs",
+]

isa_mate/__main__.py

@@ -0,0 +1,5 @@
+"""Allow running as: python -m isa_mate"""
+from .cli import main
+import sys
+
+sys.exit(main() or 0)

isa_mate/accounts/__init__.py

@@ -0,0 +1,5 @@
+"""Accounts CLI — manage subscription accounts for multi-account rotation."""
+
+from .commands import cmd_accounts
+
+__all__ = ["cmd_accounts"]

isa_mate/accounts/commands.py

@@ -0,0 +1,394 @@
+"""
+Subscription account management CLI commands.
+
+Issue #25, Epic #112
+
+Commands:
+    accounts list     — Show all accounts with status
+    accounts add      — Register a new account
+    accounts remove   — Remove an account from config
+    accounts status   — Detailed per-account usage status
+"""
+
+import json
+import logging
+import shutil
+import subprocess
+from pathlib import Path
+from typing import Any, Dict, List, Optional
+
+logger = logging.getLogger(__name__)
+
+VALID_PROVIDERS = ("anthropic-sub", "codex-sub")
+
+# Status color indicators for terminal output
+_STATUS_INDICATORS = {
+    "active": "\033[32m● active\033[0m",
+    "standby": "\033[34m○ standby\033[0m",
+    "exhausted": "\033[31m✗ exhausted\033[0m",
+    "cooldown": "\033[33m◌ cooldown\033[0m",
+    "auth_failed": "\033[31m✗ auth_failed\033[0m",
+    "configured": "\033[34m○ configured\033[0m",
+}
+
+
+def _load_config(config_path: str):
+    """Load MateConfig from path."""
+    from ..agent.config import MateConfig
+    return MateConfig.load(config_path)
+
+
+def _save_config(config, config_path: str) -> None:
+    """Patch only the agent.accounts and agent.rotation keys in the YAML file.
+
+    Loads the original YAML, updates the accounts/rotation sections, and
+    writes it back — preserving all other config sections untouched.
+    """
+    import yaml
+
+    path = Path(config_path)
+
+    # Load existing YAML to preserve all sections
+    if path.exists():
+        with open(path) as f:
+            data = yaml.safe_load(f) or {}
+    else:
+        data = {}
+
+    # Ensure agent section exists
+    if "agent" not in data:
+        data["agent"] = {}
+
+    # Patch accounts
+    if config.agent.accounts:
+        accounts_data = {}
+        for provider, accts in config.agent.accounts.items():
+            accounts_data[provider] = [
+                {
+                    "name": a.name,
+                    "priority": a.priority,
+                    "cli_profile": a.cli_profile,
+                }
+                for a in accts
+            ]
+        data["agent"]["accounts"] = accounts_data
+    else:
+        data["agent"].pop("accounts", None)
+
+    # Patch rotation
+    if hasattr(config.agent, "rotation") and config.agent.rotation:
+        data["agent"]["rotation"] = {
+            "strategy": config.agent.rotation.strategy,
+            "cooldown_minutes": config.agent.rotation.cooldown_minutes,
+            "alert_thresholds": list(config.agent.rotation.alert_thresholds),
+        }
+
+    # Patch provider (may change when first account is added)
+    data["agent"]["provider"] = config.agent.provider
+
+    path.parent.mkdir(parents=True, exist_ok=True)
+    with open(path, "w") as f:
+        yaml.safe_dump(data, f, default_flow_style=False, sort_keys=False)
+
+
+def _check_cli_profile(provider: str, cli_profile: str) -> tuple:
+    """Validate a CLI profile by checking for CLI binary and stored credentials.
+
+    This performs a fast, offline check (no API calls).  Use
+    ``_check_cli_profile_live`` for a full network-based auth probe.
+
+    Note: This validates CLI availability, not per-profile authentication.
+    Claude CLI does not expose profile-specific credential files, and Codex
+    CLI ``login status`` checks the active session regardless of profile.
+
+    Returns (ok: bool, message: str).
+    """
+    if provider == "anthropic-sub":
+        cli_name = "claude"
+        if not shutil.which(cli_name):
+            return False, f"'{cli_name}' CLI not found in PATH"
+        # Check for stored credentials in known Claude CLI config locations.
+        # Claude CLI does not expose per-profile credential files, so we
+        # can only verify that the CLI has been configured at least once.
+        home = Path.home()
+        cred_dirs = [
+            home / ".claude",
+            home / ".config" / "claude",
+        ]
+        has_config = any(d.is_dir() for d in cred_dirs)
+        if not has_config:
+            return False, "no Claude CLI config directory found"
+        return True, f"CLI installed, config present (profile '{cli_profile}')"
+
+    elif provider == "codex-sub":
+        cli_name = "codex"
+        if not shutil.which(cli_name):
+            return False, f"'{cli_name}' CLI not found in PATH"
+        # codex login status is a lightweight local check (no API call).
+        # Codex CLI does not support per-profile login status.
+        try:
+            result = subprocess.run(
+                [cli_name, "login", "status"],
+                capture_output=True, text=True, timeout=8,
+            )
+            if result.returncode == 0:
+                return True, "authenticated"
+            return False, result.stderr.strip() or "not logged in"
+        except subprocess.TimeoutExpired:
+            return False, "login check timed out"
+        except Exception as e:
+            return False, str(e)
+
+    return False, f"Unknown provider: {provider}"
+
+
+def _check_cli_profile_live(provider: str, cli_profile: str) -> tuple:
+    """Validate a CLI profile with a live network auth probe.
+
+    WARNING: This sends a real API request that costs tokens.
+    Use only when explicitly requested (e.g. ``accounts status --check-auth``).
+
+    Returns (ok: bool, message: str).
+    """
+    if provider == "anthropic-sub":
+        cli_name = "claude"
+        if not shutil.which(cli_name):
+            return False, f"'{cli_name}' CLI not found in PATH"
+        try:
+            cmd = [cli_name, "-p", "Reply OK", "--output-format", "text",
+                   "--model", "sonnet"]
+            if cli_profile and cli_profile != "default":
+                cmd.extend(["--profile", cli_profile])
+            result = subprocess.run(
+                cmd, capture_output=True, text=True, timeout=25,
+            )
+            if result.returncode == 0 and "OK" in (result.stdout or ""):
+                return True, "authenticated (live)"
+            return False, result.stderr.strip() or "auth probe failed"
+        except subprocess.TimeoutExpired:
+            return False, "auth probe timed out"
+        except Exception as e:
+            return False, str(e)
+
+    elif provider == "codex-sub":
+        cli_name = "codex"
+        if not shutil.which(cli_name):
+            return False, f"'{cli_name}' CLI not found in PATH"
+        try:
+            # Codex CLI does not support per-profile checks, but
+            # `codex login status` verifies active session auth.
+            result = subprocess.run(
+                [cli_name, "login", "status"],
+                capture_output=True, text=True, timeout=8,
+            )
+            if result.returncode == 0:
+                return True, "authenticated (live)"
+            return False, result.stderr.strip() or "not logged in"
+        except subprocess.TimeoutExpired:
+            return False, "login check timed out"
+        except Exception as e:
+            return False, str(e)
+
+    return False, f"Unknown provider: {provider}"
+
+
+# =============================================================================
+# Subcommand handlers
+# =============================================================================
+
+
+def cmd_accounts_list(config, args) -> int:
+    """Show all configured accounts with status."""
+    accounts = config.agent.accounts
+
+    if not accounts:
+        print("No subscription accounts configured.")
+        print(f"  Provider: {config.agent.provider or '(none)'}")
+        print("\nAdd accounts with: isa-mate accounts add <provider> --name <name> --profile <profile>")
+        return 0
+
+    print("Subscription Accounts")
+    print("=" * 60)
+
+    for provider, acct_list in accounts.items():
+        is_active_provider = provider == config.agent.provider
+        provider_marker = " (active)" if is_active_provider else ""
+        print(f"\n  Provider: {provider}{provider_marker}")
+        print(f"  Strategy: {config.agent.rotation.strategy}")
+        print(f"  Cooldown: {config.agent.rotation.cooldown_minutes} min")
+        print()
+
+        for acct in sorted(acct_list, key=lambda a: a.priority):
+            status = _STATUS_INDICATORS.get("configured", "?")
+            if acct.priority == 1 and is_active_provider:
+                status = _STATUS_INDICATORS.get("active", "active")
+            print(f"    {acct.priority}. {acct.name:<20} profile={acct.cli_profile:<15} {status}")
+
+    print()
+    return 0
+
+
+def cmd_accounts_add(config, args) -> int:
+    """Add a new subscription account."""
+    from ..agent.config import AccountConfig
+
+    provider = args.provider
+    name = args.name
+    profile = args.profile
+    priority = args.priority
+
+    if provider not in VALID_PROVIDERS:
+        print(f"Error: Invalid provider '{provider}'. Must be one of: {', '.join(VALID_PROVIDERS)}")
+        return 1
+
+    # Check for duplicate name
+    existing = config.agent.accounts.get(provider, [])
+    if any(a.name == name for a in existing):
+        print(f"Error: Account '{name}' already exists for provider '{provider}'")
+        return 1
+
+    # Auto-assign priority if not specified
+    if priority is None:
+        priority = max((a.priority for a in existing), default=0) + 1
+
+    # Validate CLI profile (skip if --no-validate)
+    if not args.no_validate:
+        print(f"Validating CLI profile '{profile}' for {provider}...")
+        ok, msg = _check_cli_profile(provider, profile)
+        if ok:
+            print(f"  OK: {msg}")
+        else:
+            print(f"  FAILED: {msg}")
+            print(f"\nFix: Log in with the '{profile}' profile first, then retry.")
+            if provider == "anthropic-sub":
+                print(f"  claude login --profile {profile}")
+            elif provider == "codex-sub":
+                print(f"  codex login --device-auth")
+            return 1
+
+    # Add the account
+    new_acct = AccountConfig(name=name, priority=priority, cli_profile=profile)
+    if provider not in config.agent.accounts:
+        config.agent.accounts[provider] = []
+    config.agent.accounts[provider].append(new_acct)
+
+    # If no provider set, set it
+    if not config.agent.provider:
+        config.agent.provider = provider
+
+    # Save
+    _save_config(config, args.config)
+    print(f"\nAdded account '{name}' (priority={priority}, profile={profile}) to {provider}")
+    print(f"Config saved to: {args.config}")
+    return 0
+
+
+def cmd_accounts_remove(config, args) -> int:
+    """Remove a subscription account."""
+    provider = args.provider
+    name = args.name
+
+    acct_list = config.agent.accounts.get(provider, [])
+    if not any(a.name == name for a in acct_list):
+        print(f"Error: Account '{name}' not found for provider '{provider}'")
+        return 1
+
+    if len(acct_list) == 1:
+        print(f"Warning: This is the only account for '{provider}'.")
+        if not args.force:
+            print("Use --force to remove the last account.")
+            return 1
+
+    # Remove the account
+    config.agent.accounts[provider] = [a for a in acct_list if a.name != name]
+
+    # Clean up empty provider
+    if not config.agent.accounts[provider]:
+        del config.agent.accounts[provider]
+
+    # Save
+    _save_config(config, args.config)
+    print(f"Removed account '{name}' from {provider}")
+    print(f"Config saved to: {args.config}")
+    return 0
+
+
+def cmd_accounts_status(config, args) -> int:
+    """Show detailed usage status per account."""
+    accounts = config.agent.accounts
+
+    if not accounts:
+        print("No subscription accounts configured.")
+        return 0
+
+    print("Account Usage Status")
+    print("=" * 70)
+
+    for provider, acct_list in accounts.items():
+        print(f"\n  Provider: {provider}")
+        print(f"  {'Name':<20} {'Priority':<10} {'Profile':<15} {'CLI Status'}")
+        print(f"  {'-'*20} {'-'*10} {'-'*15} {'-'*20}")
+
+        check_fn = _check_cli_profile_live if getattr(args, "check_auth", False) else _check_cli_profile
+        for acct in sorted(acct_list, key=lambda a: a.priority):
+            ok, msg = check_fn(provider, acct.cli_profile)
+            cli_status = f"\033[32mOK\033[0m" if ok else f"\033[31m{msg}\033[0m"
+            print(f"  {acct.name:<20} {acct.priority:<10} {acct.cli_profile:<15} {cli_status}")
+
+    # Try to load runtime usage tracker state if available
+    db_path = Path.home() / ".isa" / "mate" / "subscription_state.db"
+    if db_path.exists():
+        try:
+            import sqlite3
+            conn = sqlite3.connect(str(db_path))
+            cursor = conn.execute(
+                "SELECT account_name, provider_id, total_requests, "
+                "exhaustion_count, cooldown_until "
+                "FROM account_usage"
+            )
+            rows = cursor.fetchall()
+            conn.close()
+
+            if rows:
+                import time
+                print(f"\n  Runtime Usage (from {db_path})")
+                print(f"  {'Account':<20} {'Provider':<15} {'Requests':<10} {'Exhaustions':<12} {'Cooldown'}")
+                print(f"  {'-'*20} {'-'*15} {'-'*10} {'-'*12} {'-'*20}")
+                for name, prov, reqs, exhaustions, cooldown in rows:
+                    cd_str = "none"
+                    if cooldown:
+                        remaining = cooldown - time.time()
+                        if remaining > 0:
+                            cd_str = f"{remaining/60:.1f} min remaining"
+                        else:
+                            cd_str = "expired"
+                    print(f"  {name:<20} {prov:<15} {reqs:<10} {exhaustions:<12} {cd_str}")
+        except Exception as e:
+            logger.debug("Could not read usage tracker DB: %s", e)
+
+    print()
+    return 0
+
+
+# =============================================================================
+# Main dispatcher
+# =============================================================================
+
+
+def cmd_accounts(args) -> int:
+    """Dispatch accounts subcommand."""
+    config = _load_config(args.config)
+
+    subcmd = args.accounts_command
+
+    if subcmd == "list":
+        return cmd_accounts_list(config, args)
+    elif subcmd == "add":
+        return cmd_accounts_add(config, args)
+    elif subcmd == "remove":
+        return cmd_accounts_remove(config, args)
+    elif subcmd == "status":
+        return cmd_accounts_status(config, args)
+    else:
+        # Default to list if no subcommand
+        return cmd_accounts_list(config, args)

isa_mate/agent/__init__.py

@@ -0,0 +1,12 @@
+"""isA Mate agent runtime and configuration."""
+
+from .config import MateConfig, DesktopConfig
+from .runtime import MateRuntime, DesktopAgentRuntime
+
+__all__ = [
+    "MateConfig",
+    "MateRuntime",
+    # Backwards compatibility aliases
+    "DesktopConfig",
+    "DesktopAgentRuntime",
+]