import subprocess
import sys
from typing import Optional

import click

from lightning_sdk.cli.legacy.configure import _configure_ssh_internal
from lightning_sdk.cli.utils.resource_resolution import resolve_studio, resolve_teamspace


@click.group(name="connect")
def connect() -> None:
    """Connect to lightning products."""


@connect.command(name="studio")
@click.option("--name", default=None, help="The name of the studio to connect to.")
@click.option(
    "--teamspace",
    default=None,
    help="The teamspace the studio is part of. Should be of format <OWNER>/<TEAMSPACE_NAME>.",
)
def studio(name: Optional[str], teamspace: Optional[str]) -> None:
    """Connect to a studio via SSH."""
    _configure_ssh_internal(name=name, teamspace=teamspace, overwrite=False)

    studio = resolve_studio(name, resolve_teamspace(teamspace))

    try:
        subprocess.run(["ssh", studio.name])
    except Exception as ex:
        print(f"Failed to establish SSH connection: {ex}")
        sys.exit(1)