from typing import Annotated, Any

import typer
from pydantic import BaseModel
from rich.table import Table
from rich.text import Text
from rich_toolkit import RichToolkit

from fastapi_cloud_cli.api import APIClient
from fastapi_cloud_cli.commands._auth import UserCommand, get_user_command_context
from fastapi_cloud_cli.commands.tokens._app import tokens_app
from fastapi_cloud_cli.utils.apps import resolve_app_id_or_fail
from fastapi_cloud_cli.utils.execution import JsonOutputOption


class DeployToken(BaseModel):
    id: str
    name: str
    created_at: str
    expired_at: str


class DeployTokensListAPIResponse(BaseModel):
    data: list[DeployToken]


class DeployTokensListOutput(BaseModel):
    app_id: str
    tokens: list[DeployToken]


def _get_deploy_tokens(client: APIClient, app_id: str) -> DeployTokensListAPIResponse:
    response = client.get(f"/apps/{app_id}/tokens")
    response.raise_for_status()

    return DeployTokensListAPIResponse.model_validate(response.json())


def _get_deploy_tokens_table(tokens: list[DeployToken]) -> Table:
    table = Table.grid(padding=(0, 2), pad_edge=False)
    table.add_column("Name", no_wrap=True)
    table.add_column("Expiration", no_wrap=True)
    table.add_column("ID", no_wrap=True, overflow="ignore")
    table.add_row(
        Text("Name", style="bold"),
        Text("Expiration", style="bold"),
        Text("ID", style="bold"),
    )
    table.add_row("", "", "")

    for token in tokens:
        table.add_row(
            Text(token.name),
            Text(token.expired_at[:10], style="dim"),
            Text(token.id),
        )

    return table


def _render_deploy_tokens_list_output(
    data: DeployTokensListOutput, toolkit: RichToolkit
) -> None:
    toolkit.print_title("deploy tokens")
    toolkit.print_line()

    if not data.tokens:
        toolkit.print("No deploy tokens found.", bullet=False)
        return

    toolkit.print(_get_deploy_tokens_table(data.tokens), bullet=False)


@tokens_app.command("list", cls=UserCommand)
def list_tokens(
    ctx: typer.Context,
    app_id: Annotated[
        str | None,
        typer.Option(
            "--app-id",
            help="ID of the app whose deploy tokens should be listed.",
        ),
    ] = None,
    json_output: JsonOutputOption = False,
) -> Any:
    """
    List deploy tokens for an app.
    """

    toolkit = get_user_command_context(ctx).toolkit

    target_app_id = resolve_app_id_or_fail(toolkit, app_id=app_id)

    with APIClient() as client:
        with toolkit.progress(
            title="Fetching deploy tokens",
            transient=True,
        ) as progress:
            with client.handle_http_errors(
                progress,
                default_message="Error fetching deploy tokens. Please try again later.",
                not_found_message="App not found.",
                toolkit=toolkit,
            ):
                tokens = _get_deploy_tokens(client=client, app_id=target_app_id)

    toolkit.success(
        DeployTokensListOutput(app_id=target_app_id, tokens=tokens.data),
        render_output=_render_deploy_tokens_list_output,
    )