import click
from browser_cli.client import send_command, BrowserNotConnected
from rich.console import Console
from rich.table import Table

console = Console()


def _handle(command, args=None):
    try:
        return send_command(command, args or {})
    except BrowserNotConnected as e:
        console.print(f"[red]Error:[/red] {e}")
        raise SystemExit(1)
    except RuntimeError as e:
        console.print(f"[red]Browser error:[/red] {e}")
        raise SystemExit(1)


def _print_groups(groups: list[dict]) -> None:
    if not groups:
        console.print("[yellow]No groups found[/yellow]")
        return
    table = Table(show_header=True, header_style="bold cyan")
    table.add_column("ID", style="dim", no_wrap=True)
    table.add_column("Name")
    table.add_column("Color", width=10)
    table.add_column("Collapsed", width=10)
    table.add_column("Tabs", width=6)
    for g in groups:
        table.add_row(
            str(g.get("id", "")),
            g.get("title") or "(unnamed)",
            g.get("color") or "",
            "yes" if g.get("collapsed") else "no",
            str(g.get("tabCount", "")),
        )
    console.print(table)


@click.group("group")
def group_group():
    """Manage tab groups."""


@group_group.command("list")
def group_list():
    """List all tab groups."""
    groups = _handle("group.list")
    _print_groups(groups or [])


@group_group.command("tabs")
@click.argument("group_id", type=int)
def group_tabs(group_id):
    """List tabs inside a group."""
    from browser_cli.commands.tabs import _print_tabs
    tabs = _handle("group.tabs", {"groupId": group_id})
    _print_tabs(tabs or [])


@group_group.command("count")
def group_count():
    """Count all tab groups."""
    count = _handle("group.count")
    console.print(f"[bold]{count}[/bold] group(s)")


@group_group.command("query")
@click.argument("search")
def group_query(search):
    """Search groups by name."""
    groups = _handle("group.query", {"search": search})
    _print_groups(groups or [])


@group_group.command("close")
@click.argument("group_id", type=int)
def group_close(group_id):
    """Close (ungroup and optionally close) a tab group."""
    _handle("group.close", {"groupId": group_id})
    console.print(f"[green]Group {group_id} closed[/green]")


@group_group.command("create")
@click.argument("name")
def group_create(name):
    """Create a new tab group with NAME."""
    result = _handle("group.open", {"name": name})
    gid = result.get("id") if isinstance(result, dict) else result
    console.print(f"[green]Created group '{name}'[/green] (id: {gid})")


@group_group.command("add-tab")
@click.argument("group")
@click.argument("url", required=False)
def group_add_tab(group, url):
    """Open a new tab (optionally at URL) inside GROUP (name or ID)."""
    result = _handle("group.add_tab", {"group": group, "url": url})
    tab_id = result.get("tabId") if isinstance(result, dict) else result
    label = url or "new tab"
    console.print(f"[green]Opened {label}[/green] in group '{group}' (tab id: {tab_id})")