daniel156161
0d5c49c19a
Testing / test (push) Failing after 10m21s
- compat.py → compat/ package: auth.py (auth-field normalizers), commands.py (command-format shims), __init__.py (re-exports) - Add _auth_0_9_3 transformer: normalizes pubkey to lowercase before auth so clients < 0.9.3 sending uppercase hex are accepted - adapt_auth() now called before auth check in serve.py; command extracted after adapt_auth so future transformers can rename commands safely - serve.py: deduplicate _recv_exact (import from client), unify resp/resp_payload across Windows/Unix branches, require lowercase hex pubkey (re.fullmatch), reorganize imports, drop unused os import - client.py: move payload/framed construction inside branches (remote path no longer serializes JSON it never uses); fix _is_valid_key_spec operator precedence; import MAX_MSG_BYTES from version_manager - auth.py: narrow except clause (ValueError instead of bare Exception) - Bump version 0.9.2 → 0.9.3 Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
109 lines
3.6 KiB
Python
109 lines
3.6 KiB
Python
import click
|
|
from browser_cli.commands import _handle, _handle_multi, _multi_browser_targets
|
|
from rich.console import Console
|
|
|
|
console = Console()
|
|
|
|
|
|
@click.group("session")
|
|
def session_group():
|
|
"""Save and restore browser sessions."""
|
|
|
|
|
|
@session_group.command("save")
|
|
@click.argument("name")
|
|
def session_save(name):
|
|
"""Save all current tabs as session NAME."""
|
|
result = _handle("session.save", {"name": name})
|
|
count = result.get("tabs", 0) if isinstance(result, dict) else 0
|
|
console.print(f"[green]Session '{name}' saved[/green] ({count} tabs)")
|
|
|
|
|
|
@session_group.command("load")
|
|
@click.argument("name")
|
|
def session_load(name):
|
|
"""Restore session NAME (opens all saved tabs)."""
|
|
result = _handle("session.load", {"name": name})
|
|
count = result.get("tabs", 0) if isinstance(result, dict) else 0
|
|
console.print(f"[green]Session '{name}' loaded[/green] ({count} tabs opened)")
|
|
|
|
|
|
@session_group.command("diff")
|
|
@click.argument("name_a")
|
|
@click.argument("name_b")
|
|
def session_diff(name_a, name_b):
|
|
"""Show tabs added/removed between two saved sessions."""
|
|
diff = _handle("session.diff", {"nameA": name_a, "nameB": name_b})
|
|
if not diff:
|
|
console.print("[yellow]No diff data returned[/yellow]")
|
|
return
|
|
|
|
added = diff.get("added") or []
|
|
removed = diff.get("removed") or []
|
|
|
|
if added:
|
|
console.print(f"[green]Added in '{name_b}':[/green]")
|
|
for url in added:
|
|
console.print(f" + {url}")
|
|
|
|
if removed:
|
|
console.print(f"[red]Removed in '{name_b}':[/red]")
|
|
for url in removed:
|
|
console.print(f" - {url}")
|
|
|
|
if not added and not removed:
|
|
console.print("[green]Sessions are identical[/green]")
|
|
|
|
|
|
@session_group.command("list")
|
|
def session_list():
|
|
"""List all saved sessions."""
|
|
from rich.table import Table
|
|
targets = _multi_browser_targets()
|
|
show_browser = bool(targets)
|
|
if targets:
|
|
sessions = []
|
|
for target in targets:
|
|
result = _handle_multi("session.list", profile=target.profile, remote=target.remote)
|
|
if result is None:
|
|
continue
|
|
sessions.extend({**session, "browser": target.display_name} for session in result)
|
|
if not sessions:
|
|
console.print("[red]Error:[/red] Cannot resolve a browser socket automatically.")
|
|
raise SystemExit(1)
|
|
else:
|
|
sessions = _handle("session.list")
|
|
if not sessions:
|
|
console.print("[yellow]No saved sessions[/yellow]")
|
|
return
|
|
table = Table(show_header=True, header_style="bold cyan")
|
|
if show_browser:
|
|
table.add_column("Browser")
|
|
table.add_column("Name")
|
|
table.add_column("Tabs", width=6)
|
|
table.add_column("Saved at")
|
|
for s in sessions:
|
|
from datetime import datetime
|
|
saved = datetime.fromtimestamp(s["savedAt"] / 1000).strftime("%Y-%m-%d %H:%M") if s.get("savedAt") else ""
|
|
row = [s.get("browser", "")] if show_browser else []
|
|
row.extend([s["name"], str(s["tabs"]), saved])
|
|
table.add_row(*row)
|
|
console.print(table)
|
|
|
|
|
|
@session_group.command("remove")
|
|
@click.argument("name")
|
|
def session_remove(name):
|
|
"""Delete a saved session."""
|
|
_handle("session.remove", {"name": name})
|
|
console.print(f"[green]Session '{name}' removed[/green]")
|
|
|
|
|
|
@session_group.command("auto-save")
|
|
@click.argument("state", type=click.Choice(["on", "off"]))
|
|
def session_auto_save(state):
|
|
"""Enable or disable automatic session saving."""
|
|
enabled = state == "on"
|
|
_handle("session.auto_save", {"enabled": enabled})
|
|
console.print(f"[green]Auto-save {state}[/green]")
|