init commit

browser_cli/commands/tabs.py

@@ -0,0 +1,138 @@
+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_tabs(tabs: list[dict]) -> None:
+    if not tabs:
+        console.print("[yellow]No tabs found[/yellow]")
+        return
+    table = Table(show_header=True, header_style="bold cyan")
+    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 ""
+        table.add_row(
+            str(t.get("id", "")),
+            str(t.get("windowId", "")),
+            active,
+            (t.get("title") or "")[:60],
+            (t.get("url") or "")[:80],
+        )
+    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."""
+    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("--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="Position index in target")
+def tabs_move(tab_id, group_id, window_id, index):
+    """Move a tab to a different window or group."""
+    _handle("tabs.move", {"tabId": tab_id, "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."""
+    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]")