120 lines
3.9 KiB
Python
120 lines
3.9 KiB
Python
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})")
|
|
|
|
|
|
@group_group.command("move")
|
|
@click.argument("group")
|
|
@click.option("--forward", is_flag=True, help="Move group one position to the right")
|
|
@click.option("--backward", is_flag=True, help="Move group one position to the left")
|
|
def group_move(group, forward, backward):
|
|
"""Move a tab group forward or backward (name or ID)."""
|
|
if not forward and not backward:
|
|
console.print("[red]Specify --forward or --backward[/red]")
|
|
raise SystemExit(1)
|
|
result = _handle("group.move", {"group": 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]")
|