connectonion 0.5.8__py3-none-any.whl

connectonion/cli/commands/reset_commands.py

@@ -0,0 +1,149 @@
+"""
+Purpose: Delete global ConnectOnion configuration and create fresh account with new Ed25519 keypair
+LLM-Note:
+  Dependencies: imports from [sys, shutil, toml, pathlib, rich.console, rich.prompt, rich.panel, address, auth_commands.authenticate, __version__, datetime] | imported by [cli/main.py via handle_reset()] | tested by [tests/cli/test_cli_reset.py]
+  Data flow: receives no args → checks ~/.co/ exists → prompts user for 'Y' confirmation with clear warnings → deletes ~/.co/keys/, ~/.co/config.toml, ~/.co/keys.env → recreates directory structure → address.generate() creates new Ed25519 keypair with seed phrase → address.save() saves to ~/.co/keys/ → creates new config.toml with fresh agent identity → calls authenticate(global_dir, save_to_project=False) to register new account and get bonus credits → displays seed phrase in Rich panel → warns user to update project .env files
+  State/Effects: DESTRUCTIVE OPERATION | deletes entire ~/.co/ directory contents (keys, config, keys.env) | creates fresh ~/.co/ with new keypair | calls authenticate() which creates new backend account and writes OPENONION_API_KEY to keys.env | writes to stdout via rich.Console with warnings and confirmations | existing projects still have old API key (requires manual 'co init' to update)
+  Integration: exposes handle_reset() for CLI | similar to ensure_global_config() in init.py but deletes first | relies on address.generate() for new Ed25519 keypair | calls authenticate() to register new account | displays seed phrase via Rich panel | requires explicit 'Y' confirmation to proceed
+  Performance: file deletion is fast (<100ms) | address.generate() is fast (<100ms) | authenticate() makes network call (2-5s) | config file writes are I/O bound
+  Errors: gracefully handles missing ~/.co/ (nothing to reset) | requires uppercase 'Y' for confirmation (case-sensitive) | cancels if user types anything else | authenticate() may fail but reset still completes | warns user about data loss before proceeding
+"""
+
+import sys
+import shutil
+import toml
+from pathlib import Path
+from rich.console import Console
+from rich.prompt import Prompt
+from rich.panel import Panel
+
+from ... import address
+from .auth_commands import authenticate
+
+console = Console()
+
+
+def handle_reset():
+    """Reset ConnectOnion global configuration and create new account.
+
+    WARNING: This will delete all your data including:
+    - Your Ed25519 keypair and account access
+    - Your balance and transaction history
+    - All configuration and credentials
+
+    You will get a fresh new account.
+    """
+    global_dir = Path.home() / ".co"
+
+    # Check if global config exists
+    if not global_dir.exists():
+        console.print("\n❌ [bold red]No global configuration found[/bold red]")
+        console.print("[yellow]Nothing to reset. Run 'co init' to set up ConnectOnion.[/yellow]\n")
+        return
+
+    # Show clear warning
+    console.print("\n[bold yellow]⚠️  WARNING: This will DELETE ALL your ConnectOnion data[/bold yellow]\n")
+    console.print("[red]You will lose:[/red]")
+    console.print("  • Your account and balance")
+    console.print("  • All transaction history")
+    console.print("  • Your Ed25519 keypair")
+    console.print("  • All configurations and credentials\n")
+
+    console.print("[green]You will get:[/green]")
+    console.print("  • Fresh new account")
+    console.print("  • New Ed25519 keypair with seed phrase\n")
+
+    console.print("[yellow]💡 Save your 12-word seed phrase if you want to recover your current account![/yellow]\n")
+
+    confirmation = Prompt.ask("[bold red]Type 'Y' to confirm reset[/bold red]")
+
+    if confirmation.upper() != "Y":
+        console.print("\n[yellow]Cancelled.[/yellow]\n")
+        return
+
+    # Delete everything
+    keys_dir = global_dir / "keys"
+    if keys_dir.exists():
+        shutil.rmtree(keys_dir)
+        console.print("✓ Deleted ~/.co/keys/")
+
+    config_path = global_dir / "config.toml"
+    if config_path.exists():
+        config_path.unlink()
+        console.print("✓ Deleted ~/.co/config.toml")
+
+    keys_env = global_dir / "keys.env"
+    if keys_env.exists():
+        keys_env.unlink()
+        console.print("✓ Deleted ~/.co/keys.env")
+
+    # Recreate directory structure
+    global_dir.mkdir(exist_ok=True)
+    keys_dir.mkdir(exist_ok=True)
+    (global_dir / "logs").mkdir(exist_ok=True)
+
+    # Generate new keypair
+    console.print("\n🔑 Generating new Ed25519 keypair...")
+    addr_data = address.generate()
+    address.save(addr_data, global_dir)
+
+    console.print(f"✓ Generated new keypair")
+    console.print(f"✓ Your new address: [bold]{addr_data['short_address']}[/bold]")
+
+    # Show seed phrase
+    console.print(Panel.fit(
+        f"[bold yellow]{addr_data['seed_phrase']}[/bold yellow]",
+        title="🔐 Your 12-Word Seed Phrase (SAVE THIS!)",
+        border_style="yellow"
+    ))
+
+    # Create new config
+    from ... import __version__
+    from datetime import datetime
+
+    config = {
+        "connectonion": {
+            "framework_version": __version__,
+            "created": datetime.now().isoformat(),
+        },
+        "cli": {
+            "version": "1.0.0",
+        },
+        "agent": {
+            "address": addr_data["address"],
+            "short_address": addr_data["short_address"],
+            "email": f"{addr_data['address'][:10]}@mail.openonion.ai",
+            "email_active": False,
+            "created_at": datetime.now().isoformat(),
+            "algorithm": "ed25519",
+            "default_model": "co/gemini-2.5-pro",
+            "max_iterations": 10,
+        },
+    }
+
+    with open(config_path, 'w', encoding='utf-8') as f:
+        toml.dump(config, f)
+    console.print("✓ Created ~/.co/config.toml")
+
+    keys_env.touch()
+    if sys.platform != 'win32':
+        keys_env.chmod(0o600)
+    console.print("✓ Created ~/.co/keys.env")
+
+    # Authenticate to get fresh bonus
+    console.print("\n🔐 Authenticating with OpenOnion...")
+    success = authenticate(global_dir, save_to_project=False)
+
+    if success:
+        console.print("\n[bold green]✅ Reset complete! Your new account is ready.[/bold green]")
+        console.print("\n[yellow]💡 Save your 12-word seed phrase somewhere safe![/yellow]")
+
+        console.print("\n[bold yellow]⚠️  IMPORTANT: Update your project .env files![/bold yellow]")
+        console.print("\n[yellow]Your existing projects still have the old API key.[/yellow]")
+        console.print("[yellow]To use your new account in each project:[/yellow]")
+        console.print("  [bold cyan]1.[/bold cyan] cd into the project directory")
+        console.print("  [bold cyan]2.[/bold cyan] Run: [bold]co init[/bold]")
+        console.print("\n[dim]This will update the project's .env file with your new account.[/dim]\n")
+    else:
+        console.print("\n[yellow]⚠️  Reset complete, but authentication failed.[/yellow]")
+        console.print("[yellow]Run 'co auth' to authenticate manually.[/yellow]\n")

connectonion/cli/commands/status_commands.py

@@ -0,0 +1,168 @@
+"""
+Purpose: Display account status including balance, usage, and email configuration without re-authenticating
+LLM-Note:
+  Dependencies: imports from [os, toml, requests, pathlib, rich.console, rich.panel, dotenv.load_dotenv, jwt, address] | imported by [cli/main.py via handle_status()] | calls backend at [https://oo.openonion.ai/api/v1/auth] | tested by [tests/cli/test_cli_status.py]
+  Data flow: receives no args → _load_api_key() checks OPENONION_API_KEY from env/local .env/global ~/.co/keys.env → _load_config() loads agent info from .co/config.toml or ~/.co/config.toml → address.load() reads Ed25519 keypair → creates fresh auth message with timestamp → address.sign() creates signature → POST to /api/v1/auth to get current user data → displays balance, credits, total spent, email, agent ID → warns if balance <= 0
+  State/Effects: no state modifications | makes network GET request to oo.openonion.ai | reads from env vars, .env, ~/.co/keys.env, config.toml | writes to stdout via rich.Console and rich.Panel | does NOT update any files
+  Integration: exposes handle_status() for CLI | similar to authenticate() but read-only | relies on address module for signature generation | uses requests for HTTP calls | displays Rich panel with account info | checks OPENONION_API_KEY in 3 locations (priority: env var > local .env > global ~/.co/keys.env)
+  Performance: network call to backend (1-2s) | signature generation is fast (<10ms) | file I/O for config and .env files
+  Errors: fails gracefully if OPENONION_API_KEY not found (prints message to run 'co auth') | fails if keys missing in .co/keys/ | fails if backend unreachable (prints HTTP error) | handles response errors with status code display
+"""
+
+import os
+import toml
+import requests
+from pathlib import Path
+from rich.console import Console
+from rich.panel import Panel
+from dotenv import load_dotenv
+
+console = Console()
+
+
+def _load_api_key() -> str:
+    """Load OPENONION_API_KEY from environment.
+
+    Checks in order:
+    1. Environment variable
+    2. Local .env file
+    3. Global ~/.co/keys.env file
+
+    Returns:
+        API key if found, None otherwise
+    """
+    # Check environment variable first
+    api_key = os.getenv("OPENONION_API_KEY")
+    if api_key:
+        return api_key
+
+    # Check local .env
+    local_env = Path(".env")
+    if local_env.exists():
+        load_dotenv(local_env)
+        api_key = os.getenv("OPENONION_API_KEY")
+        if api_key:
+            return api_key
+
+    # Check global ~/.co/keys.env
+    global_env = Path.home() / ".co" / "keys.env"
+    if global_env.exists():
+        load_dotenv(global_env)
+        api_key = os.getenv("OPENONION_API_KEY")
+        if api_key:
+            return api_key
+
+    return None
+
+
+def _load_config() -> dict:
+    """Load config from .co/config.toml or ~/.co/config.toml.
+
+    Returns:
+        Config dict if found, empty dict otherwise
+    """
+    # Check local .co/config.toml first
+    local_config = Path(".co") / "config.toml"
+    if local_config.exists():
+        return toml.load(local_config)
+
+    # Check global ~/.co/config.toml
+    global_config = Path.home() / ".co" / "config.toml"
+    if global_config.exists():
+        return toml.load(global_config)
+
+    return {}
+
+
+def handle_status():
+    """Check account status without re-authenticating.
+
+    Shows:
+    - Agent ID
+    - Email address
+    - Balance (remaining credits)
+    - Total spent
+    - Last seen
+    - Warnings if balance is low
+    """
+    # Load API key
+    api_key = _load_api_key()
+    if not api_key:
+        console.print("\n❌ [bold red]No API key found[/bold red]")
+        console.print("\n[cyan]Authenticate first:[/cyan]")
+        console.print("  [bold]co auth[/bold]     Authenticate with OpenOnion\n")
+        return
+
+    # Load config for agent info
+    config = _load_config()
+    agent_info = config.get("agent", {})
+
+    # Decode JWT to extract public_key and re-authenticate to get fresh data
+    import jwt
+    import time
+    from ... import address
+
+    # Load keys to re-sign
+    co_dir = Path(".co")
+    if not (co_dir.exists() and (co_dir / "keys" / "agent.key").exists()):
+        co_dir = Path.home() / ".co"
+
+    addr_data = address.load(co_dir)
+    if not addr_data:
+        console.print("\n❌ [bold red]No keys found[/bold red]")
+        console.print("[yellow]Run 'co auth' first.[/yellow]\n")
+        return
+
+    public_key = addr_data["address"]
+    timestamp = int(time.time())
+    message = f"ConnectOnion-Auth-{public_key}-{timestamp}"
+    signature = address.sign(addr_data, message.encode()).hex()
+
+    # Call auth endpoint to get fresh user data
+    response = requests.post(
+        "https://oo.openonion.ai/api/v1/auth",
+        json={
+            "public_key": public_key,
+            "signature": signature,
+            "message": message
+        }
+    )
+
+    if response.status_code != 200:
+        console.print(f"\n❌ [bold red]Error {response.status_code}[/bold red]")
+        console.print(f"[yellow]{response.text}[/yellow]\n")
+        return
+
+    data = response.json()
+    user = data.get("user", {})
+    email_info = user.get("email") or {}
+
+    # Build info display
+    api_key_display = f"{api_key[:20]}..." if len(api_key) > 20 else api_key
+
+    # Compute short address from full address (first 6 chars + ... + last 4 chars)
+    short_address = f"{public_key[:6]}...{public_key[-4:]}"
+
+    info_lines = [
+        f"[cyan]Agent Address:[/cyan] {public_key}",
+        f"[cyan]Agent ID:[/cyan] {short_address}",
+        f"[cyan]Email:[/cyan] {email_info.get('address') or os.getenv('AGENT_EMAIL', 'Not configured')}",
+        f"[cyan]API Key:[/cyan] {api_key_display}",
+        f"[cyan]Balance:[/cyan] ${user.get('balance_usd', 0.0):.4f}",
+        f"[cyan]Total Spent:[/cyan] ${user.get('total_cost_usd', 0.0):.4f}",
+        f"[cyan]Credits:[/cyan] ${user.get('credits_usd', 0.0):.4f}",
+    ]
+
+    console.print("\n")
+    console.print(Panel.fit(
+        "\n".join(info_lines),
+        title="📊 Account Status",
+        border_style="cyan"
+    ))
+
+    if user.get('balance_usd', 0) <= 0:
+        console.print("\n[yellow]⚠️  Low balance! Purchase tokens at https://oo.openonion.ai[/yellow]")
+
+    console.print("\n[yellow]💡 Tips:[/yellow]")
+    console.print("   • Use 'co auth' to refresh your token")
+    console.print("   • Visit https://oo.openonion.ai to manage your account\n")