vantage-cli 0.1.1__py3-none-any.whl

vantage_cli/commands/profile/crud.py

@@ -0,0 +1,529 @@
+# © 2025 Vantage Compute, Inc. All rights reserved.
+# Confidential and proprietary. Unauthorized use prohibited.
+"""Profile management CRUD operations for Vantage CLI."""
+
+import json
+from typing import Any, Dict
+
+import typer
+from loguru import logger
+from rich import print_json
+from rich.console import Console
+from rich.panel import Panel
+from rich.table import Table
+from typing_extensions import Annotated
+
+from vantage_cli.command_base import JsonOption, get_effective_json_output
+from vantage_cli.config import (
+    Settings,
+    dump_settings,
+    get_active_profile,
+    init_user_filesystem,
+    set_active_profile,
+)
+from vantage_cli.constants import USER_CONFIG_FILE, USER_TOKEN_CACHE_DIR
+from vantage_cli.exceptions import Abort
+
+
+def create_profile(
+    ctx: typer.Context,
+    profile_name: Annotated[str, typer.Argument(help="Name of the profile to create")],
+    api_base_url: Annotated[
+        str, typer.Option("--api-url", help="API base URL")
+    ] = "https://apis.vantagecompute.ai",
+    tunnel_api_url: Annotated[
+        str, typer.Option("--tunnel-url", help="Tunnel API URL")
+    ] = "https://tunnel.vantagecompute.ai",
+    oidc_base_url: Annotated[
+        str, typer.Option("--oidc-url", help="OIDC base URL")
+    ] = "https://auth.vantagecompute.ai",
+    oidc_client_id: Annotated[str, typer.Option("--client-id", help="OIDC client ID")] = "default",
+    oidc_max_poll_time: Annotated[
+        int, typer.Option("--max-poll-time", help="OIDC max poll time in seconds")
+    ] = 300,
+    force: Annotated[
+        bool, typer.Option("--force", "-f", help="Overwrite existing profile")
+    ] = False,
+    activate: Annotated[
+        bool, typer.Option("--activate", help="Activate this profile after creation")
+    ] = False,
+    json_output: JsonOption = False,
+):
+    """Create a new Vantage CLI profile."""
+    # Get the effective JSON output preference
+    effective_json = get_effective_json_output(ctx, json_output)
+
+    # Check if profile already exists
+    existing_profiles = _get_all_profiles()
+
+    if profile_name in existing_profiles and not force:
+        message = f"Profile '{profile_name}' already exists. Use --force to overwrite."
+        if effective_json:
+            result = {"success": False, "profile_name": profile_name, "message": message}
+            print_json(data=result)
+            return
+        else:
+            raise Abort(
+                message,
+                subject="Profile Exists",
+                log_message=f"Profile '{profile_name}' already exists",
+            )
+
+    # Create the settings
+    try:
+        settings = Settings(
+            api_base_url=api_base_url,
+            oidc_base_url=oidc_base_url,
+            tunnel_api_url=tunnel_api_url,
+            oidc_client_id=oidc_client_id,
+            oidc_max_poll_time=oidc_max_poll_time,
+        )
+
+        # Initialize filesystem for this profile
+        init_user_filesystem(profile_name)
+
+        # Save the settings
+        dump_settings(profile_name, settings)
+
+        # Set as active profile if requested
+        if activate:
+            set_active_profile(profile_name)
+            logger.info(f"Set '{profile_name}' as active profile")
+
+        logger.info(f"Created profile '{profile_name}'")
+
+        if effective_json:
+            result = {
+                "success": True,
+                "profile_name": profile_name,
+                "settings": settings.model_dump(),
+                "is_active": activate,
+                "message": f"Profile '{profile_name}' created successfully",
+            }
+            if activate:
+                result["message"] += " and set as active"
+            print_json(data=result)
+        else:
+            console = Console()
+            console.print()
+            message = f"✅ Profile '[bold cyan]{profile_name}[/bold cyan]' created successfully!"
+            if activate:
+                message += "\n🎯 Set as active profile!"
+            console.print(
+                Panel(message, title="[green]Profile Created[/green]", border_style="green")
+            )
+
+            # Show profile details
+            _render_profile_details(profile_name, settings)
+
+    except Exception as e:
+        logger.error(f"Failed to create profile '{profile_name}': {str(e)}")
+        if effective_json:
+            result = {
+                "success": False,
+                "profile_name": profile_name,
+                "message": f"Failed to create profile: {str(e)}",
+            }
+            print_json(data=result)
+        else:
+            raise Abort(
+                f"Failed to create profile '{profile_name}': {str(e)}",
+                subject="Profile Creation Failed",
+                log_message=f"Profile creation error: {str(e)}",
+            )
+
+
+def delete_profile(
+    ctx: typer.Context,
+    profile_name: Annotated[str, typer.Argument(help="Name of the profile to delete")],
+    force: Annotated[bool, typer.Option("--force", "-f", help="Skip confirmation prompt")] = False,
+    json_output: JsonOption = False,
+):
+    """Delete a Vantage CLI profile."""
+    # Get the effective JSON output preference
+    effective_json = get_effective_json_output(ctx, json_output)
+
+    # Check if profile exists
+    existing_profiles = _get_all_profiles()
+
+    if profile_name not in existing_profiles:
+        message = f"Profile '{profile_name}' does not exist."
+        if effective_json:
+            result = {"success": False, "profile_name": profile_name, "message": message}
+            print_json(data=result)
+            return
+        else:
+            raise Abort(
+                message,
+                subject="Profile Not Found",
+                log_message=f"Profile '{profile_name}' not found for deletion",
+            )
+
+    # Don't allow deletion of 'default' profile without force
+    if profile_name == "default" and not force:
+        message = "Cannot delete 'default' profile without --force flag."
+        if effective_json:
+            result = {"success": False, "profile_name": profile_name, "message": message}
+            print_json(data=result)
+            return
+        else:
+            raise Abort(
+                message,
+                subject="Cannot Delete Default Profile",
+                log_message="Attempted to delete default profile without force",
+            )
+
+    # Confirmation prompt unless force is used
+    if not force and not effective_json:
+        from rich.prompt import Confirm
+
+        console = Console()
+        console.print(
+            f"\n[yellow]⚠️  You are about to delete profile '[bold red]{profile_name}[/bold red]'[/yellow]"
+        )
+        console.print(
+            "[yellow]This will remove all settings and cached tokens for this profile![/yellow]"
+        )
+
+        if not Confirm.ask("Are you sure you want to proceed?"):
+            console.print("[dim]Deletion cancelled.[/dim]")
+            return
+
+    try:
+        # Remove from config file
+        if USER_CONFIG_FILE.exists():
+            all_profiles = json.loads(USER_CONFIG_FILE.read_text())
+            if profile_name in all_profiles:
+                del all_profiles[profile_name]
+                USER_CONFIG_FILE.write_text(json.dumps(all_profiles, indent=2))
+
+        # Clear token cache for this profile
+        _clear_profile_token_cache(profile_name)
+
+        # Remove profile token cache directory if it exists
+        profile_cache_dir = USER_TOKEN_CACHE_DIR / profile_name
+        if profile_cache_dir.exists():
+            import shutil
+
+            shutil.rmtree(profile_cache_dir)
+
+        logger.info(f"Deleted profile '{profile_name}'")
+
+        if effective_json:
+            result = {
+                "success": True,
+                "profile_name": profile_name,
+                "message": f"Profile '{profile_name}' deleted successfully",
+            }
+            print_json(data=result)
+        else:
+            console = Console()
+            console.print()
+            console.print(
+                Panel(
+                    f"✅ Profile '[bold cyan]{profile_name}[/bold cyan]' deleted successfully!",
+                    title="[green]Profile Deleted[/green]",
+                    border_style="green",
+                )
+            )
+            console.print()
+
+    except Exception as e:
+        logger.error(f"Failed to delete profile '{profile_name}': {str(e)}")
+        if effective_json:
+            result = {
+                "success": False,
+                "profile_name": profile_name,
+                "message": f"Failed to delete profile: {str(e)}",
+            }
+            print_json(data=result)
+        else:
+            raise Abort(
+                f"Failed to delete profile '{profile_name}': {str(e)}",
+                subject="Profile Deletion Failed",
+                log_message=f"Profile deletion error: {str(e)}",
+            )
+
+
+def get_profile(
+    ctx: typer.Context,
+    profile_name: Annotated[str, typer.Argument(help="Name of the profile to get details for")],
+    json_output: JsonOption = False,
+):
+    """Get details of a specific Vantage CLI profile."""
+    # Get the effective JSON output preference
+    effective_json = get_effective_json_output(ctx, json_output)
+
+    # Check if profile exists
+    existing_profiles = _get_all_profiles()
+
+    if profile_name not in existing_profiles:
+        message = f"Profile '{profile_name}' does not exist."
+        if effective_json:
+            result = {"success": False, "profile_name": profile_name, "message": message}
+            print_json(data=result)
+            return
+        else:
+            raise Abort(
+                message,
+                subject="Profile Not Found",
+                log_message=f"Profile '{profile_name}' not found",
+            )
+
+    try:
+        # Load the profile settings
+        profile_data = existing_profiles[profile_name]
+        settings = Settings(**profile_data)
+
+        if effective_json:
+            result = {
+                "success": True,
+                "profile_name": profile_name,
+                "settings": settings.model_dump(),
+            }
+            print_json(data=result)
+        else:
+            _render_profile_details(profile_name, settings)
+
+    except Exception as e:
+        logger.error(f"Failed to get profile '{profile_name}': {str(e)}")
+        if effective_json:
+            result = {
+                "success": False,
+                "profile_name": profile_name,
+                "message": f"Failed to get profile: {str(e)}",
+            }
+            print_json(data=result)
+        else:
+            raise Abort(
+                f"Failed to get profile '{profile_name}': {str(e)}",
+                subject="Profile Retrieval Failed",
+                log_message=f"Profile retrieval error: {str(e)}",
+            )
+
+
+def list_profiles(ctx: typer.Context, json_output: JsonOption = False):
+    """List all Vantage CLI profiles."""
+    # Get the effective JSON output preference
+    effective_json = get_effective_json_output(ctx, json_output)
+
+    try:
+        # Get active profile from file system
+        active_profile = get_active_profile()
+
+        # Get all profiles
+        all_profiles = _get_all_profiles()
+
+        if not all_profiles:
+            if effective_json:
+                result = {"profiles": [], "total": 0, "current_profile": active_profile}
+                print_json(data=result)
+            else:
+                console = Console()
+                console.print()
+                console.print(Panel("No profiles found.", title="[yellow]No Profiles"))
+                console.print()
+            return
+
+        if effective_json:
+            # Prepare profiles with additional metadata
+            profiles_list = []
+            for name, settings_data in all_profiles.items():
+                profile_info = {
+                    "name": name,
+                    "settings": settings_data,
+                    "is_current": name == active_profile,
+                }
+                profiles_list.append(profile_info)
+
+            result = {
+                "profiles": profiles_list,
+                "total": len(profiles_list),
+                "current_profile": active_profile,
+            }
+            print_json(data=result)
+        else:
+            _render_profiles_table(all_profiles, active_profile)
+
+    except Exception as e:
+        logger.error(f"Failed to list profiles: {str(e)}")
+        if effective_json:
+            result = {"success": False, "message": f"Failed to list profiles: {str(e)}"}
+            print_json(data=result)
+        else:
+            raise Abort(
+                f"Failed to list profiles: {str(e)}",
+                subject="Profile Listing Failed",
+                log_message=f"Profile listing error: {str(e)}",
+            )
+
+
+def use_profile(
+    ctx: typer.Context,
+    profile_name: Annotated[str, typer.Argument(help="Name of the profile to activate")],
+    json_output: JsonOption = False,
+):
+    """Activate a profile for use in the current session."""
+    # Get the effective JSON output preference
+    effective_json = get_effective_json_output(ctx, json_output)
+
+    # Check if profile exists
+    existing_profiles = _get_all_profiles()
+
+    if profile_name not in existing_profiles:
+        message = f"Profile '{profile_name}' does not exist."
+        if effective_json:
+            result = {
+                "success": False,
+                "profile_name": profile_name,
+                "message": message,
+                "available_profiles": list(existing_profiles.keys()),
+            }
+            print_json(data=result)
+            return
+        else:
+            console = Console()
+            console.print()
+            console.print(
+                Panel(
+                    f"❌ Profile '[bold red]{profile_name}[/bold red]' does not exist.\n\n"
+                    f"Available profiles: {', '.join(existing_profiles.keys()) if existing_profiles else 'None'}",
+                    title="[red]Profile Not Found[/red]",
+                    border_style="red",
+                )
+            )
+            raise typer.Exit(1)
+
+    # Set as active profile
+    try:
+        set_active_profile(profile_name)
+        logger.info(f"Set '{profile_name}' as active profile")
+
+        if effective_json:
+            result = {
+                "success": True,
+                "profile_name": profile_name,
+                "message": f"Profile '{profile_name}' is now active",
+                "note": "This profile will be used for all future commands until changed",
+            }
+            print_json(data=result)
+        else:
+            console = Console()
+            console.print()
+            console.print(
+                Panel(
+                    f"🎯 Profile '[bold cyan]{profile_name}[/bold cyan]' is now active!\n\n"
+                    f"This profile will be used for all future commands until changed.",
+                    title="[green]Profile Activated[/green]",
+                    border_style="green",
+                )
+            )
+
+    except Exception as e:
+        logger.error(f"Failed to set '{profile_name}' as active: {str(e)}")
+        if effective_json:
+            result = {
+                "success": False,
+                "profile_name": profile_name,
+                "message": f"Failed to activate profile: {str(e)}",
+            }
+            print_json(data=result)
+        else:
+            raise Abort(
+                f"Failed to activate profile '{profile_name}': {str(e)}",
+                subject="Profile Activation Failed",
+                log_message=f"Profile activation error: {str(e)}",
+            )
+
+
+def _get_all_profiles() -> Dict[str, Any]:
+    """Get all profiles from the config file."""
+    if not USER_CONFIG_FILE.exists():
+        return {}
+
+    try:
+        return json.loads(USER_CONFIG_FILE.read_text())
+    except (json.JSONDecodeError, FileNotFoundError):
+        return {}
+
+
+def _render_profiles_table(profiles: Dict[str, Any], current_profile: str) -> None:
+    """Render a table of all profiles."""
+    console = Console()
+
+    # Create the table
+    table = Table(
+        title="Vantage CLI Profiles",
+        caption=f"Items: {len(profiles)} • Current profile: [bold]{current_profile}[/bold]",
+        show_header=True,
+        header_style="bold white",
+    )
+
+    # Add columns
+    table.add_column("Name", style="bold cyan")
+    table.add_column("Current", style="green", justify="center")
+    table.add_column("API Base URL", style="blue")
+    table.add_column("OIDC Base URL", style="yellow")
+    table.add_column("Tunnel API URL", style="yellow")
+    table.add_column("Client ID", style="white")
+
+    # Add rows
+    for name, settings_data in profiles.items():
+        current_marker = "✓" if name == current_profile else ""
+
+        # Handle both old and new settings format
+        api_url = settings_data.get("api_base_url", "N/A")
+        oidc_url = settings_data.get("oidc_base_url", "N/A")
+        tunnel_url = settings_data.get("tunnel_api_url", "N/A")
+        client_id = settings_data.get("oidc_client_id", "N/A")
+
+        table.add_row(
+            name, current_marker, str(api_url), str(oidc_url), str(tunnel_url), str(client_id)
+        )
+
+    console.print()
+    console.print(table)
+    console.print()
+
+
+def _render_profile_details(profile_name: str, settings: Settings) -> None:
+    """Render detailed information for a single profile."""
+    console = Console()
+
+    # Create a table matching the whoami command style
+    table = Table(
+        title=f"Profile Details: {profile_name}", show_header=True, header_style="bold white"
+    )
+
+    table.add_column("Property", style="bold cyan")
+    table.add_column("Value", style="white")
+
+    # Add profile information
+    table.add_row("Profile Name", profile_name)
+    table.add_row("API Base URL", settings.api_base_url)
+    table.add_row("OIDC Base URL", settings.oidc_base_url)
+    table.add_row("Tunnel Base URL", settings.tunnel_api_url)
+    table.add_row("OIDC Domain", settings.oidc_domain)
+    table.add_row("OIDC Client ID", settings.oidc_client_id)
+    table.add_row("OIDC Max Poll Time", f"{settings.oidc_max_poll_time} seconds")
+    table.add_row("Supported Clouds", ", ".join(settings.supported_clouds))
+
+    console.print()
+    console.print(table)
+    console.print()
+
+
+def _clear_profile_token_cache(profile: str) -> None:
+    """Clear token cache for a specific profile."""
+    token_dir = USER_TOKEN_CACHE_DIR / profile
+    if token_dir.exists():
+        access_token_path = token_dir / "access.token"
+        refresh_token_path = token_dir / "refresh.token"
+
+        logger.debug(f"Removing access token at {access_token_path}")
+        if access_token_path.exists():
+            access_token_path.unlink()
+
+        logger.debug(f"Removing refresh token at {refresh_token_path}")
+        if refresh_token_path.exists():
+            refresh_token_path.unlink()

vantage_cli/commands/profile/render.py

@@ -0,0 +1,55 @@
+# © 2025 Vantage Compute, Inc. All rights reserved.
+# Confidential and proprietary. Unauthorized use prohibited.
+"""Render helpers for profile command output formatting."""
+
+from typing import Any, Dict, Optional
+
+from rich.console import Console
+from rich.panel import Panel
+from rich.table import Table
+
+
+def render_profile_operation_result(
+    operation: str,
+    profile_name: str,
+    success: bool = True,
+    details: Optional[Dict[str, Any]] = None,
+) -> None:
+    """Render the result of a profile operation.
+
+    Args:
+        operation: The operation performed (create, delete, update)
+        profile_name: Name of the profile
+        success: Whether the operation was successful
+        details: Additional details about the operation
+    """
+    console = Console()
+    console.print()
+
+    status_icon = "✅" if success else "❌"
+    status_color = "green" if success else "red"
+    action_text = f"{operation.title()}d" if success else f"{operation.title()} Failed"
+
+    console.print(
+        Panel(
+            f"{status_icon} Profile '[bold cyan]{profile_name}[/bold cyan]' {operation} {'successful' if success else 'failed'}!",
+            title=f"[{status_color}]{action_text}[/{status_color}]",
+            border_style=status_color,
+        )
+    )
+
+    if details:
+        # Show additional details if provided
+        details_table = Table(title="Profile Details", show_header=False, box=None, padding=(0, 1))
+        details_table.add_column("Field", style="bold blue", width=20)
+        details_table.add_column("Value", style="white")
+
+        for key, value in details.items():
+            if value is not None:
+                display_key = key.replace("_", " ").title()
+                details_table.add_row(display_key, str(value))
+
+        console.print()
+        console.print(details_table)
+
+    console.print()