"""Studio start command."""

from typing import Optional

import rich_click as click

from lightning_sdk.cli.utils.get_base_studio import get_base_studio_id
from lightning_sdk.cli.utils.json_output import echo_json
from lightning_sdk.cli.utils.logging import LightningCommand
from lightning_sdk.cli.utils.resource_resolution import resolve_teamspace
from lightning_sdk.cli.utils.richt_print import studio_name_link
from lightning_sdk.cli.utils.save_to_config import save_teamspace_to_config
from lightning_sdk.lightning_cloud.openapi.rest import ApiException
from lightning_sdk.studio import Studio


@click.command("create", cls=LightningCommand)
@click.option("--name", help="The name of the studio to create. If not provided, a random name will be generated.")
@click.option("--teamspace", help="Override default teamspace (format: owner/teamspace)")
@click.option("--cloud", help="Cloud provider or cloud account to create the studio on. Defaults to teamspace default.")
@click.option(
    "--studio-type",
    help="The base studio template name to use for creating the studio. "
    "Must be lowercase and hyphenated (use '-' instead of spaces). "
    "Run 'lightning base-studio list' to see all available templates. "
    "Defaults to the first available template.",
    type=click.STRING,
)
@click.option("--json", "as_json", is_flag=True, default=False, help="Output the created Studio as JSON.")
def create_studio(
    name: Optional[str] = None,
    teamspace: Optional[str] = None,
    cloud: Optional[str] = None,
    studio_type: Optional[str] = None,
    as_json: bool = False,
) -> None:
    """Create a new Studio.

    Example:
        lightning studio create
    """
    create_impl(
        name=name,
        teamspace=teamspace,
        cloud=cloud,
        studio_type=studio_type,
        as_json=as_json,
    )


def create_impl(
    name: Optional[str],
    teamspace: Optional[str],
    cloud: Optional[str] = None,
    studio_type: Optional[str] = None,
    as_json: bool = False,
) -> None:
    resolved_teamspace = resolve_teamspace(teamspace)
    save_teamspace_to_config(resolved_teamspace, overwrite=False)

    cls_name = Studio.__qualname__

    # check for available base studios
    template_id = get_base_studio_id(studio_type, teamspace=resolved_teamspace)

    try:
        studio = Studio(
            name=name,
            teamspace=resolved_teamspace,
            create_ok=True,
            cloud=cloud,
            studio_type=template_id,
        )
    except (RuntimeError, ValueError, ApiException):
        if name:
            raise ValueError(f"Could not create {cls_name}: '{name}'. Does the {cls_name} exist?") from None
        raise ValueError(f"Could not create {cls_name}: '{name}'. Please provide a {cls_name} name") from None

    if as_json:
        echo_json({"name": studio.name, "teamspace": f"{resolved_teamspace.owner.name}/{resolved_teamspace.name}"})
        return

    click.echo(f"{cls_name} {studio_name_link(studio)} created successfully")