"""Job inspect command."""

from typing import Optional

import rich_click as click

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_job, resolve_job_machine, resolve_teamspace


@click.command("inspect", cls=LightningCommand)
@click.argument("name", required=False, help="The job name. Required.")
@click.option(
    "--teamspace",
    default=None,
    help=(
        "the name of the teamspace the job lives in. "
        "Should be specified as {teamspace_owner}/{teamspace_name} (e.g my-org/my-teamspace). "
        "If not specified, uses the configured default teamspace."
    ),
)
@click.option("--rank", type=int, default=None, help="Inspect one machine in a multi-machine job.")
@click.option("--json", "as_json", is_flag=True, default=False, help="Output as JSON (inspect always emits JSON).")
def inspect_job(
    name: Optional[str] = None,
    teamspace: Optional[str] = None,
    rank: Optional[int] = None,
    as_json: bool = False,
) -> None:
    """Inspect a job for further details as JSON."""
    resolved_teamspace = resolve_teamspace(teamspace)
    job = resolve_job(name, resolved_teamspace)
    if job.is_multi_machine is True and rank is not None:
        job = resolve_job_machine(job, rank)
    elif rank is not None:
        raise click.UsageError("--rank is only supported for multi-machine jobs.")
    echo_json(job.dict())