init commit

browser_cli/commands/extract.py

@@ -0,0 +1,68 @@
+import click
+import json
+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)
+
+
+@click.group("extract")
+def extract_group():
+    """Extract content from the active tab."""
+
+
+@extract_group.command("links")
+def extract_links():
+    """Extract all links from the active tab."""
+    links = _handle("extract.links")
+    if not links:
+        console.print("[yellow]No links found[/yellow]")
+        return
+    table = Table(show_header=True, header_style="bold cyan")
+    table.add_column("Text", width=40)
+    table.add_column("URL")
+    for lnk in links:
+        table.add_row((lnk.get("text") or "")[:60], lnk.get("href") or "")
+    console.print(table)
+
+
+@extract_group.command("images")
+def extract_images():
+    """Extract all images from the active tab."""
+    images = _handle("extract.images")
+    if not images:
+        console.print("[yellow]No images found[/yellow]")
+        return
+    table = Table(show_header=True, header_style="bold cyan")
+    table.add_column("Alt", width=30)
+    table.add_column("Src")
+    for img in images:
+        table.add_row((img.get("alt") or "")[:40], img.get("src") or "")
+    console.print(table)
+
+
+@extract_group.command("text")
+def extract_text():
+    """Extract all visible text from the active tab."""
+    text = _handle("extract.text")
+    console.print(text or "")
+
+
+@extract_group.command("json")
+@click.argument("selector")
+def extract_json(selector):
+    """Parse and pretty-print JSON content inside SELECTOR."""
+    data = _handle("extract.json", {"selector": selector})
+    console.print_json(json.dumps(data))