"""Shared registration for resource deletion commands."""

from __future__ import annotations

from functools import partial
from typing import Any, Callable

import rich_click as click

from lightning_sdk.cli.utils.logging import LightningCommand

DeleteAction = Callable[[], None]
DeleteResolver = Callable[[str, str | None], DeleteAction]


def _default_delete_resolver(
    resource_cls: type[Any],
    identifier: str,
    context: str | None,
    *,
    context_option: str,
    resource_kwargs: dict[str, Any],
) -> DeleteAction:
    resource = resource_cls(
        name=identifier,
        **{context_option: context},
        **resource_kwargs,
    )
    return resource.delete


def register_delete_command(
    group: click.Group,
    resource_cls: type[Any] | None = None,
    *,
    label: str,
    help: str,  # noqa: A002
    identifier: str = "name",
    context_option: str = "teamspace",
    context_help: str | None = None,
    resolve_delete: DeleteResolver | None = None,
    resource_kwargs: dict[str, Any] | None = None,
) -> click.Command:
    """Create and directly attach a resource delete command."""
    if resolve_delete is not None:
        resolver = resolve_delete
    elif resource_cls is not None:
        resolver = partial(
            _default_delete_resolver,
            resource_cls,
            context_option=context_option,
            resource_kwargs=dict(resource_kwargs or {}),
        )
    else:
        raise TypeError("resource_cls is required without resolve_delete")

    def callback(**params: Any) -> None:
        delete = resolver(params[identifier], params[context_option])
        if not params["yes"]:
            click.confirm(
                "Are you sure you want to delete?",
                default=False,
                abort=True,
            )
        delete()
        click.echo(f"{label} deleted")

    callback = click.option(
        "--yes",
        "-y",
        is_flag=True,
        default=False,
        help="Delete without prompting for confirmation.",
    )(callback)
    callback = click.option(f"--{context_option.replace('_', '-')}", help=context_help)(callback)
    callback = click.argument(identifier)(callback)
    command = click.command("delete", cls=LightningCommand, help=help)(callback)
    group.add_command(command)
    return command