import click
from browser_cli.commands import client_from_ctx, gentle_mode_option, handle_errors, print_counts
from rich.console import Console
from rich.table import Table

console = Console()

def _print_groups(groups, *, show_browser: bool = False) -> None:
    if not groups:
        console.print("[yellow]No groups found[/yellow]")
        return
    table = Table(show_header=True, header_style="bold cyan")
    if show_browser:
        table.add_column("Browser")
    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:
        row = [
            (g.browser or "") if show_browser else None,
            str(g.id),
            g.title or "",
            g.color or "",
            "yes" if g.collapsed else "no",
            str(g.tab_count),
        ]
        table.add_row(*[value for value in row if value is not None])
    console.print(table)

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

@group_group.command("list")
@handle_errors
def group_list():
    """List all tab groups."""
    groups = client_from_ctx().groups.list()
    _print_groups(groups, show_browser=any(g.browser for g in groups))

@group_group.command("tabs")
@click.argument("group_id", type=int)
@handle_errors
def group_tabs(group_id):
    """List tabs inside a group."""
    from browser_cli.commands.tabs import _print_tabs
    _print_tabs(client_from_ctx().groups.tabs(group_id))

@group_group.command("count")
@handle_errors
def group_count():
    """Count all tab groups."""
    print_counts(client_from_ctx().groups.count(), "group")

@group_group.command("query")
@click.argument("search")
@handle_errors
def group_query(search):
    """Search groups by name."""
    _print_groups(client_from_ctx().groups.query(search))

@group_group.command("close")
@click.argument("group_id", type=int)
@gentle_mode_option("Throttle mode for large group operations.")
@handle_errors
def group_close(group_id, gentle_mode):
    """Close (ungroup and optionally close) a tab group."""
    client_from_ctx().groups.close(group_id, gentle_mode=gentle_mode)
    console.print(f"[green]Group {group_id} closed[/green]")

@group_group.command("create")
@click.argument("name")
@handle_errors
def group_create(name):
    """Create a new tab group with NAME."""
    group = client_from_ctx().groups.create(name)
    console.print(f"[green]Created group '{name}'[/green] (id: {group.id})")

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

@group_group.command("move")
@click.argument("group")
@click.option("-f", "--forward", "forward", is_flag=True, help="Move group one position to the right")
@click.option("-b", "--backward", "backward", is_flag=True, help="Move group one position to the left")
@click.option("-r", "--right", "forward", is_flag=True, help="Move group one position to the right")
@click.option("-l", "--left", "backward", is_flag=True, help="Move group one position to the left")
@handle_errors
def group_move(group, forward, backward):
    """Move a tab group forward/backward or right/left (name or ID)."""
    if not forward and not backward:
        console.print("[red]Specify --forward/--right or --backward/--left[/red]")
        raise SystemExit(1)
    result = client_from_ctx().groups.move(group, forward=forward, backward=backward)
    if isinstance(result, dict) and not result.get("moved"):
        console.print(f"[yellow]Group '{group}' is already at the {'end' if forward else 'start'}[/yellow]")
    else:
        direction = "forward" if forward else "backward"
        console.print(f"[green]Group '{group}' moved {direction}[/green]")