From 794d2c137417039a338b24e94e1a962b13ccd6bd Mon Sep 17 00:00:00 2001 From: ddiddi Date: Tue, 2 Dec 2025 06:06:17 -0800 Subject: [PATCH] Add authentication system with login, logout, and whoami commands MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - Add solo login command that directs users to hub.getsolo.tech to get token - Add solo logout command to remove authentication token - Add solo whoami command to display current authentication status - Include helper functions (is_authenticated, require_auth) for other commands to use - Store authentication token securely in config file šŸ¤– Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude --- solo/cli.py | 27 ++++++ solo/commands/login.py | 192 +++++++++++++++++++++++++++++++++++++++++ 2 files changed, 219 insertions(+) create mode 100644 solo/commands/login.py diff --git a/solo/cli.py b/solo/cli.py index c49161f..645b189 100644 --- a/solo/cli.py +++ b/solo/cli.py @@ -38,6 +38,33 @@ def setup(): _setup() +@app.command() +def login(): + """ + Authenticate with Solo by getting a token from hub.getsolo.tech + """ + from solo.commands.login import login as _login + _login() + + +@app.command() +def logout(): + """ + Log out from Solo by removing the authentication token + """ + from solo.commands.login import logout as _logout + _logout() + + +@app.command() +def whoami(): + """ + Display current authentication status and user information + """ + from solo.commands.login import whoami as _whoami + _whoami() + + @app.command() def serve( model: Optional[str] = typer.Option(None, "--model", "-m", help="""Model name or path. Can be: diff --git a/solo/commands/login.py b/solo/commands/login.py new file mode 100644 index 0000000..a103d4b --- /dev/null +++ b/solo/commands/login.py @@ -0,0 +1,192 @@ +import typer +import os +import json +import webbrowser +from pathlib import Path + +from rich.console import Console +from rich.panel import Panel + +from solo.config import CONFIG_PATH + +console = Console() + + +def is_authenticated(): + """ + Check if the user is authenticated + + Returns: + tuple: (bool, str) - (is_authenticated, token or None) + """ + if not os.path.exists(CONFIG_PATH): + return False, None + + try: + with open(CONFIG_PATH, 'r') as f: + config = json.load(f) + + if 'auth' in config and 'solo_token' in config['auth']: + return True, config['auth']['solo_token'] + except json.JSONDecodeError: + pass + + return False, None + + +def require_auth(): + """ + Decorator/helper function to require authentication for a command + + Returns: + str: token if authenticated, None otherwise (prints error message) + """ + authenticated, token = is_authenticated() + + if not authenticated: + typer.echo("āŒ You are not authenticated. Please run 'solo login' first.") + return None + + return token + + +def login(): + """ + Authenticate with Solo by getting a token from hub.getsolo.tech + """ + console.print(Panel.fit( + "[bold cyan]Solo Login[/bold cyan]\n\n" + "To use Solo CLI, you need to authenticate with a token.\n" + "Visit [bold yellow]hub.getsolo.tech[/bold yellow] to get your token.", + title="Authentication Required" + )) + + # Open hub.getsolo.tech in browser + try: + typer.echo("\nšŸŒ Opening hub.getsolo.tech in your browser...") + webbrowser.open("https://hub.getsolo.tech") + except Exception as e: + typer.echo(f"āš ļø Could not open browser automatically: {e}") + typer.echo("Please manually visit: https://hub.getsolo.tech") + + # Prompt user to input their token + typer.echo("\n") + token = typer.prompt("Enter your Solo token", hide_input=True) + + if not token or not token.strip(): + typer.echo("āŒ No token provided. Login cancelled.") + return + + # Load existing config or create new one + config = {} + if os.path.exists(CONFIG_PATH): + try: + with open(CONFIG_PATH, 'r') as f: + config = json.load(f) + except json.JSONDecodeError: + config = {} + + # Save token to config + if 'auth' not in config: + config['auth'] = {} + + config['auth']['solo_token'] = token.strip() + + # Create config directory if it doesn't exist + os.makedirs(os.path.dirname(CONFIG_PATH), exist_ok=True) + + # Save configuration to file + with open(CONFIG_PATH, 'w') as f: + json.dump(config, f, indent=4) + + typer.secho("\nāœ… Successfully logged in to Solo!", fg=typer.colors.GREEN) + typer.echo(f"Token saved to {CONFIG_PATH}") + + +def logout(): + """ + Log out from Solo by removing the authentication token + """ + # Check if config exists + if not os.path.exists(CONFIG_PATH): + typer.echo("āŒ No configuration found. You are not logged in.") + return + + # Load config + try: + with open(CONFIG_PATH, 'r') as f: + config = json.load(f) + except json.JSONDecodeError: + typer.echo("āŒ Error reading configuration file.") + return + + # Check if auth token exists + if 'auth' not in config or 'solo_token' not in config['auth']: + typer.echo("āŒ You are not logged in.") + return + + # Remove token + del config['auth']['solo_token'] + + # Remove auth section if empty + if not config['auth']: + del config['auth'] + + # Save updated config + with open(CONFIG_PATH, 'w') as f: + json.dump(config, f, indent=4) + + typer.secho("āœ… Successfully logged out from Solo!", fg=typer.colors.GREEN) + + +def whoami(): + """ + Display current authentication status and user information + """ + from rich.table import Table + + # Check if config exists + if not os.path.exists(CONFIG_PATH): + typer.echo("āŒ No configuration found. Please run 'solo setup' first.") + return + + # Load config + try: + with open(CONFIG_PATH, 'r') as f: + config = json.load(f) + except json.JSONDecodeError: + typer.echo("āŒ Error reading configuration file.") + return + + # Check authentication status + is_authenticated = 'auth' in config and 'solo_token' in config['auth'] + + # Create status table + table = Table(title="Solo Authentication Status") + table.add_column("Property", style="cyan") + table.add_column("Value", style="green") + + if is_authenticated: + token = config['auth']['solo_token'] + # Mask token for security (show first 8 and last 4 characters) + if len(token) > 12: + masked_token = f"{token[:8]}...{token[-4:]}" + else: + masked_token = "***" + + table.add_row("Status", "āœ… Authenticated") + table.add_row("Token", masked_token) + else: + table.add_row("Status", "āŒ Not authenticated") + table.add_row("Token", "N/A") + + # Add other user info if available + if 'user' in config: + user_config = config['user'] + if 'domain' in user_config: + table.add_row("Domain", user_config['domain']) + + console.print(table) + + if not is_authenticated: + typer.echo("\nTo authenticate, run: solo login")