browser-cli/browser_cli/commands/tabs.py

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

console = Console()


def _handle(command, args=None, profile=None):
    try:
        return send_command(command, args or {}, profile=profile)
    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 _handle_multi(command, args=None, profile=None):
    try:
        return send_command(command, args or {}, profile=profile)
    except (BrowserNotConnected, RuntimeError):
        return None


def _multi_browser_targets():
    root = click.get_current_context().find_root()
    if root.obj.get("browser_explicit"):
        return []
    targets = active_browser_targets()
    if len(targets) <= 1:
        return []
    return targets


def _print_tabs(tabs: list[dict], *, show_browser: bool = False) -> None:
    if not tabs:
        console.print("[yellow]No tabs found[/yellow]")
        return
    table = Table(show_header=True, header_style="bold cyan")
    if show_browser:
        table.add_column("Browser", no_wrap=True)
    table.add_column("ID", style="dim", no_wrap=True)
    table.add_column("Window", no_wrap=True)
    table.add_column("Active", width=7)
    table.add_column("Title")
    table.add_column("URL")
    for t in tabs:
        active = "[green]✓[/green]" if t.get("active") else ""
        row = [
            t.get("browser", "") if show_browser else None,
            str(t.get("id", "")),
            str(t.get("windowId", "")),
            active,
            (t.get("title") or "")[:60],
            (t.get("url") or "")[:80],
        ]
        table.add_row(*[value for value in row if value is not None])
    console.print(table)


@click.group("tabs")
def tabs_group():
    """Manage browser tabs."""


@tabs_group.command("list")
def tabs_list():
    """List all open tabs across all windows."""
    targets = _multi_browser_targets()
    if targets:
        tabs = []
        for target in targets:
            result = _handle_multi("tabs.list", profile=target.profile)
            if result is None:
                continue
            tabs.extend({**tab, "browser": target.display_name} for tab in result)
        if not tabs:
            console.print("[red]Error:[/red] Cannot resolve a browser socket automatically.")
            raise SystemExit(1)
        _print_tabs(tabs, show_browser=True)
        return
    tabs = _handle("tabs.list")
    _print_tabs(tabs or [])


@tabs_group.command("close")
@click.argument("tab_id", type=int, required=False)
@click.option("--inactive", is_flag=True, help="Close all inactive tabs")
@click.option("--duplicates", is_flag=True, help="Close duplicate tabs (keep first)")
def tabs_close(tab_id, inactive, duplicates):
    """Close a tab, all inactive tabs, or all duplicate tabs."""
    result = _handle("tabs.close", {"tabId": tab_id, "inactive": inactive, "duplicates": duplicates})
    count = result.get("closed", 0) if isinstance(result, dict) else 1
    console.print(f"[green]Closed {count} tab(s)[/green]")


@tabs_group.command("move")
@click.argument("tab_id", type=int)
@click.option("--forward", is_flag=True, help="Move one position to the right")
@click.option("--backward", is_flag=True, help="Move one position to the left")
@click.option("--group", "group_id", type=int, default=None, help="Move to tab group ID")
@click.option("--window", "window_id", type=int, default=None, help="Move to window ID")
@click.option("--index", type=int, default=None, help="Absolute position index in target")
def tabs_move(tab_id, forward, backward, group_id, window_id, index):
    """Move a tab. Use --forward/--backward for relative movement."""
    _handle("tabs.move", {
        "tabId": tab_id, "forward": forward, "backward": backward,
        "groupId": group_id, "windowId": window_id, "index": index,
    })
    console.print("[green]Tab moved[/green]")


@tabs_group.command("active")
@click.argument("tab_id", type=int)
def tabs_active(tab_id):
    """Switch browser focus to a tab."""
    _handle("tabs.active", {"tabId": tab_id})
    console.print(f"[green]Switched to tab {tab_id}[/green]")


@tabs_group.command("filter")
@click.argument("pattern")
def tabs_filter(pattern):
    """List tabs whose URL contains PATTERN."""
    tabs = _handle("tabs.filter", {"pattern": pattern})
    _print_tabs(tabs or [])


@tabs_group.command("count")
@click.argument("pattern", required=False)
def tabs_count(pattern):
    """Count open tabs, optionally filtered by URL PATTERN."""
    targets = _multi_browser_targets()
    if targets:
        table = Table(show_header=True, header_style="bold cyan")
        table.add_column("Browser")
        table.add_column("Tabs", justify="right")
        total = 0
        rows = 0
        for target in targets:
            count = _handle_multi("tabs.count", {"pattern": pattern}, profile=target.profile)
            if count is None:
                continue
            count = int(count or 0)
            total += count
            rows += 1
            table.add_row(target.display_name, str(count))
        if rows == 0:
            console.print("[red]Error:[/red] Cannot resolve a browser socket automatically.")
            raise SystemExit(1)
        table.add_row("Total", str(total))
        console.print(table)
        return
    count = _handle("tabs.count", {"pattern": pattern})
    label = f" matching '{pattern}'" if pattern else ""
    console.print(f"[bold]{count}[/bold] tab(s){label}")


@tabs_group.command("query")
@click.argument("search")
def tabs_query(search):
    """Search tabs by URL or title."""
    tabs = _handle("tabs.query", {"search": search})
    _print_tabs(tabs or [])


@tabs_group.command("html")
@click.argument("tab_id", type=int, required=False)
def tabs_html(tab_id):
    """Print the full HTML of a tab."""
    html = _handle("tabs.html", {"tabId": tab_id})
    console.print(html or "")


@tabs_group.command("dedupe")
def tabs_dedupe():
    """Close duplicate tabs (keep the first occurrence of each URL)."""
    result = _handle("tabs.dedupe")
    count = result.get("closed", 0) if isinstance(result, dict) else 0
    console.print(f"[green]Closed {count} duplicate tab(s)[/green]")


@tabs_group.command("sort")
@click.option("--by", type=click.Choice(["domain", "title", "time"]), default="domain", show_default=True)
def tabs_sort(by):
    """Sort tabs within each window."""
    _handle("tabs.sort", {"by": by})
    console.print(f"[green]Tabs sorted by {by}[/green]")


@tabs_group.command("merge-windows")
def tabs_merge_windows():
    """Move all tabs into the focused window."""
    result = _handle("tabs.merge_windows")
    count = result.get("moved", 0) if isinstance(result, dict) else 0
    console.print(f"[green]Merged — moved {count} tab(s) into current window[/green]")