init commit

browser_cli/commands/groups.py

@@ -0,0 +1,102 @@
+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})")