import logging
from typing import Annotated, Any

import typer
from pydantic import BaseModel
from rich.markup import escape
from rich.table import Table
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.teams._app import teams_app
from fastapi_cloud_cli.commands.teams.get import Team, _get_team_dashboard_url
from fastapi_cloud_cli.config import Settings
from fastapi_cloud_cli.utils.execution import JsonOutputOption

logger = logging.getLogger(__name__)

DEFAULT_LIMIT = 100
DEFAULT_OFFSET = 0


class TeamsListAPIResponse(BaseModel):
    data: list[Team]
    count: int


class TeamsListOutput(BaseModel):
    teams: list[Team]
    total_count: int
    limit: int
    offset: int


def _get_teams(client: APIClient, *, limit: int, offset: int) -> TeamsListOutput:
    response = client.get(
        "/teams/",
        params={
            "limit": limit,
            "skip": offset,
        },
    )
    response.raise_for_status()

    data = TeamsListAPIResponse.model_validate(response.json())

    return TeamsListOutput(
        teams=data.data,
        total_count=data.count,
        limit=limit,
        offset=offset,
    )


def _render_teams_list_output(data: TeamsListOutput, toolkit: RichToolkit) -> None:
    toolkit.print_title("teams")
    toolkit.print_line()

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

    settings = Settings.get()

    table = Table.grid(padding=(0, 2), pad_edge=False)
    table.add_column("Name")
    table.add_column("ID")
    table.add_row("[bold]Name[/bold]", "[bold]ID[/bold]")
    table.add_row("", "")

    for team in data.teams:
        table.add_row(
            f"[link={_get_team_dashboard_url(team, settings=settings)}]{escape(team.name)}[/link]",
            team.id,
        )

    toolkit.print(table, bullet=False)


@teams_app.command("list", cls=UserCommand)
def list_teams(
    ctx: typer.Context,
    limit: Annotated[
        int,
        typer.Option(
            "--limit",
            help="Maximum number of teams to return.",
            min=1,
        ),
    ] = DEFAULT_LIMIT,
    offset: Annotated[
        int,
        typer.Option(
            "--offset",
            help="Offset into the team result set.",
            min=0,
        ),
    ] = DEFAULT_OFFSET,
    json_output: JsonOutputOption = False,
) -> Any:
    """
    List FastAPI Cloud teams.
    """

    toolkit = get_user_command_context(ctx).toolkit

    with APIClient() as client:
        with toolkit.progress(
            title="Fetching teams",
            transient=True,
        ) as progress:
            with client.handle_http_errors(
                progress,
                default_message="Error fetching teams. Please try again later.",
                toolkit=toolkit,
            ):
                result = _get_teams(client, limit=limit, offset=offset)

    toolkit.success(result, render_output=_render_teams_list_output)