browser-cli/browser_cli/commands/session.py

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")
@click.option("--gentle-mode", type=click.Choice(["auto", "normal", "gentle", "ultra"]), default="auto", show_default=True, help="Throttle mode for large restores.")
@click.option("--discard-background-tabs", is_flag=True, help="Discard restored background tabs after opening to reduce load.")
@click.option("--lazy", is_flag=True, help="Create lightweight placeholder tabs after --eager-tabs; placeholders load when selected.")
@click.option("--eager-tabs", type=int, default=10, show_default=True, help="Number of real tabs to open before lazy placeholders.")
@click.option("--background", "background_job", is_flag=True, help="Start restore as a background job and return immediately.")
def session_load(name, gentle_mode, discard_background_tabs, lazy, eager_tabs, background_job):
    """Restore session NAME (opens all saved tabs)."""
    result = _handle("session.load", {
        "name": name,
        "gentleMode": gentle_mode,
        "discardBackgroundTabs": discard_background_tabs,
        "lazy": lazy,
        "eagerTabs": eager_tabs,
        "__background": background_job,
    })
    if background_job and isinstance(result, dict) and result.get("jobId"):
        console.print(f"[green]Session restore started[/green] job={result['jobId']}")
        return
    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("job-status")
@click.argument("job_id")
def session_job_status(job_id):
    """Show status for a background session job."""
    result = _handle("jobs.status", {"jobId": job_id}) or {}
    status = result.get("status", "unknown")
    console.print(f"[bold]{job_id}[/bold]: {status}")
    if result.get("error"):
        console.print(f"[red]{result['error']}[/red]")
    elif result.get("result"):
        console.print(result["result"])

@session_group.command("job-cancel")
@click.argument("job_id")
def session_job_cancel(job_id):
    """Cancel a running background job."""
    _handle("jobs.cancel", {"jobId": job_id})
    console.print(f"[green]Cancel requested for {job_id}[/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]")