import click
from browser_cli.commands import client_from_ctx, handle_errors
from rich.console import Console
from rich.table import Table
from rich.tree import Tree

console = Console()

def _print_windows(windows: list[dict], *, show_browser: bool = False) -> None:
    if not windows:
        console.print("[yellow]No windows 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("Alias", width=20)
    table.add_column("Tabs", width=6)
    table.add_column("State", width=12)
    for w in windows:
        row = [
            w.get("browser", "") if show_browser else None,
            str(w.get("id", "")),
            w.get("alias") or "",
            str(w.get("tabCount", "")),
            w.get("state") or "",
        ]
        table.add_row(*[value for value in row if value is not None])
    console.print(table)

@click.group("windows")
def windows_group():
    """Manage browser windows."""

@windows_group.command("list")
@handle_errors
def windows_list():
    """List all browser windows."""
    windows = client_from_ctx().windows.list()
    _print_windows(windows, show_browser=any("browser" in w for w in windows))

@windows_group.command("tree")
@handle_errors
def windows_tree():
    """Show windows and their tabs as a tree."""
    client = client_from_ctx()
    windows = client.windows.list()
    tabs = client.tabs.list()
    root = Tree("[bold]Windows[/bold]")
    for w in sorted(windows, key=lambda item: (item.get("browser", ""), item.get("id", 0))):
        wid = w.get("id")
        label = f"Window {wid}"
        if w.get("alias"):
            label += f" ({w['alias']})"
        if w.get("browser"):
            label = f"{w['browser']}: " + label
        node = root.add(label)
        for tab in sorted([t for t in tabs if t.window_id == wid and (not w.get("browser") or t.browser == w.get("browser"))], key=lambda t: t.index):
            active = " [green]*[/green]" if tab.active else ""
            node.add(f"[{tab.id}] {tab.title or '(untitled)'}{active} — [dim]{tab.url or ''}[/dim]")
    console.print(root)

@windows_group.command("rename")
@click.argument("window_id", type=int)
@click.argument("name")
@handle_errors
def windows_rename(window_id, name):
    """Give a window a local alias NAME (stored in native host)."""
    client_from_ctx().windows.rename(window_id, name)
    console.print(f"[green]Window {window_id} aliased as '{name}'[/green]")

@windows_group.command("close")
@click.argument("window_id", type=int)
@handle_errors
def windows_close(window_id):
    """Close a browser window."""
    client_from_ctx().windows.close(window_id)
    console.print(f"[green]Window {window_id} closed[/green]")

@windows_group.command("open")
@click.argument("url", required=False)
@handle_errors
def windows_open(url):
    """Open a new browser window."""
    result = client_from_ctx().windows.open(url)
    wid = result.get("id") if isinstance(result, dict) else result
    console.print(f"[green]Opened new window[/green] (id: {wid})" + (f" with {url}" if url else ""))