"""Folder download command."""

from typing import Optional

import rich_click as click

from lightning_sdk.cli.legacy.download import folder as _download_folder
from lightning_sdk.cli.utils.logging import LightningCommand


@click.command("download", cls=LightningCommand)
@click.argument("path")
@click.option(
    "--studio",
    default=None,
    help=(
        "The name of the studio to download from. "
        "Will show a menu with user's owned studios for selection if not specified. "
        "If provided, should be in the form of <TEAMSPACE-NAME>/<STUDIO-NAME> where the names are case-sensitive. "
        "The teamspace and studio names can be regular expressions to match, "
        "a menu filtered studios will be shown for final selection."
    ),
)
@click.option(
    "--teamspace",
    default=None,
    help="The teamspace the drive folder is part of. Should be of format <OWNER>/<TEAMSPACE_NAME>.",
)
@click.option(
    "--local-path",
    "--local_path",
    default=".",
    type=click.Path(file_okay=False, dir_okay=True),
    help="The path to the directory you want to download the folder to.",
)
def download_folder(
    path: str = "", studio: Optional[str] = None, teamspace: Optional[str] = None, local_path: str = "."
) -> None:
    """Download a folder from a Studio or a Teamspace drive folder."""
    callback = _download_folder.callback
    assert callback is not None
    callback(path=path, studio=studio, teamspace=teamspace, local_path=local_path)