daniel156161
1c5fd0ffee
Navigation: open-wait (open + block until loaded) DOM: key, hover, check/uncheck, clear, focus, submit, poll, scroll, select, eval, wait-for Tabs: pin/unpin, screenshot, watch-url (block until URL matches regex) New command groups: page info, storage get/set, cookies list/get/set Extension: add cookies permission
87 lines
3.0 KiB
Python
87 lines
3.0 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)
|
|
|
|
|
|
@click.group("cookies")
|
|
def cookies_group():
|
|
"""Manage browser cookies."""
|
|
|
|
|
|
@cookies_group.command("list")
|
|
@click.option("--url", default=None, help="Filter by URL")
|
|
@click.option("--domain", default=None, help="Filter by domain")
|
|
@click.option("--name", default=None, help="Filter by cookie name")
|
|
def cookies_list(url, domain, name):
|
|
"""List cookies, optionally filtered by URL, domain, or name."""
|
|
cookies = _handle("cookies.list", {"url": url, "domain": domain, "name": name}) or []
|
|
if not cookies:
|
|
console.print("[yellow]No cookies found[/yellow]")
|
|
return
|
|
table = Table(show_header=True, header_style="bold cyan")
|
|
table.add_column("Name")
|
|
table.add_column("Value")
|
|
table.add_column("Domain")
|
|
table.add_column("Path")
|
|
table.add_column("Secure", width=7)
|
|
table.add_column("HttpOnly", width=9)
|
|
for c in cookies:
|
|
table.add_row(
|
|
c.get("name", ""),
|
|
(c.get("value") or "")[:60],
|
|
c.get("domain", ""),
|
|
c.get("path", ""),
|
|
"[green]✓[/green]" if c.get("secure") else "",
|
|
"[green]✓[/green]" if c.get("httpOnly") else "",
|
|
)
|
|
console.print(table)
|
|
|
|
|
|
@cookies_group.command("get")
|
|
@click.argument("url")
|
|
@click.argument("name")
|
|
def cookies_get(url, name):
|
|
"""Get the value of a single cookie by URL and NAME."""
|
|
cookie = _handle("cookies.get", {"url": url, "name": name})
|
|
if cookie is None:
|
|
console.print(f"[yellow]Cookie '{name}' not found for {url}[/yellow]")
|
|
raise SystemExit(1)
|
|
console.print(cookie.get("value", ""))
|
|
|
|
|
|
@cookies_group.command("set")
|
|
@click.argument("url")
|
|
@click.argument("name")
|
|
@click.argument("value")
|
|
@click.option("--domain", default=None)
|
|
@click.option("--path", default=None)
|
|
@click.option("--secure", is_flag=True)
|
|
@click.option("--http-only", "http_only", is_flag=True)
|
|
@click.option("--expires", "expiration_date", type=float, default=None, help="Unix timestamp")
|
|
@click.option("--same-site", type=click.Choice(["no_restriction", "lax", "strict"]), default=None)
|
|
def cookies_set(url, name, value, domain, path, secure, http_only, expiration_date, same_site):
|
|
"""Set a cookie on URL."""
|
|
_handle("cookies.set", {
|
|
"url": url, "name": name, "value": value,
|
|
"domain": domain, "path": path,
|
|
"secure": secure or None,
|
|
"httpOnly": http_only or None,
|
|
"expirationDate": expiration_date,
|
|
"sameSite": same_site,
|
|
})
|
|
console.print(f"[green]Set cookie:[/green] {name}={value!r} on {url}")
|