dell_ai/cli/utils.py

"""Utility functions for the Dell AI CLI."""

import json
from typing import Any, Optional

import typer
from rich.console import Console
from rich.panel import Panel
from dell_ai import DellAIClient
from dell_ai.exceptions import AuthenticationError

# Create console for rich output
console = Console(stderr=True)


def print_json(data: Any) -> None:
    """
    Print data as JSON to stdout.

    Args:
        data: Data to print as JSON
    """
    if hasattr(data, "dict"):
        data = data.dict()
    print(json.dumps(data, indent=2, default=str))


def print_error(message: str) -> None:
    """
    Print an error message to stderr and exit with status code 1.
    Uses Rich formatting for better readability.

    Args:
        message: Error message to print
    """
    console.print(
        Panel.fit(
            f"[bold red]Error:[/bold red] {message}",
            border_style="red",
            title="Dell AI CLI",
        )
    )
    raise typer.Exit(code=1)


def get_client(token: Optional[str] = None) -> DellAIClient:
    """
    Create and return a DellAIClient instance.

    Args:
        token: Optional Hugging Face API token. If not provided, will attempt to load
              from the Hugging Face token cache.

    Returns:
        A configured DellAIClient instance

    Raises:
        SystemExit: If authentication fails
    """
    try:
        return DellAIClient(token=token)
    except AuthenticationError as e:
        print_error(str(e))


def confirm_action(message: str, default: bool = False) -> bool:
    """
    Ask for user confirmation before proceeding with an action.

    Args:
        message: The confirmation message to display
        default: The default value if the user just presses Enter

    Returns:
        True if the user confirms, False otherwise
    """
    return typer.confirm(message, default=default)