alexveley
83 lines
2.6 KiB
Python
83 lines
2.6 KiB
Python
#!/usr/bin/env python3
|
|
"""Call Firefox Agent Bridge with a registered Ed25519 client key."""
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import json
|
|
import os
|
|
import sys
|
|
import urllib.error
|
|
import urllib.request
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
sys.path.insert(0, str(ROOT))
|
|
|
|
from fab_auth import load_key_bundle, sign_headers # noqa: E402
|
|
|
|
DEFAULT_BASE = os.environ.get("FAB_URL", "http://127.0.0.1:17634")
|
|
|
|
|
|
def key_path() -> Path:
|
|
env = os.environ.get("FAB_KEY_FILE")
|
|
if env:
|
|
return Path(env)
|
|
raise SystemExit(
|
|
"no client key: set FAB_KEY_FILE or pass --key "
|
|
"(python tools/register_client.py add --name NAME --write-key PATH)"
|
|
)
|
|
|
|
|
|
def call(
|
|
method: str,
|
|
args: list[Any] | None = None,
|
|
base: str = DEFAULT_BASE,
|
|
key_file: Path | None = None,
|
|
) -> Any:
|
|
payload = json.dumps({"method": method, "args": args or []}).encode("utf-8")
|
|
path = "/v1/call"
|
|
bundle = load_key_bundle(key_file or key_path())
|
|
headers = {
|
|
"Content-Type": "application/json",
|
|
**sign_headers(bundle, "POST", path, payload),
|
|
}
|
|
req = urllib.request.Request(
|
|
base.rstrip("/") + path,
|
|
data=payload,
|
|
method="POST",
|
|
headers=headers,
|
|
)
|
|
try:
|
|
with urllib.request.urlopen(req, timeout=20) as resp:
|
|
body = json.loads(resp.read().decode("utf-8"))
|
|
except urllib.error.HTTPError as exc:
|
|
detail = exc.read().decode("utf-8", errors="replace")
|
|
raise SystemExit(f"bridge HTTP {exc.code}: {detail}") from exc
|
|
except urllib.error.URLError as exc:
|
|
raise SystemExit(
|
|
f"bridge unreachable at {base} ({exc}). "
|
|
"Firefox must be open with the extension loaded."
|
|
) from exc
|
|
if not body.get("ok"):
|
|
raise SystemExit(body.get("error") or body)
|
|
return body.get("result")
|
|
|
|
|
|
def main() -> int:
|
|
parser = argparse.ArgumentParser(description="Call Firefox Agent Bridge")
|
|
parser.add_argument("method", help="e.g. bookmarks.search or meta.methods")
|
|
parser.add_argument("args_json", nargs="?", default="[]", help="JSON array of arguments")
|
|
parser.add_argument("--url", default=DEFAULT_BASE)
|
|
parser.add_argument("--key", type=Path, help="client key bundle (or FAB_KEY_FILE)")
|
|
ns = parser.parse_args()
|
|
args = json.loads(ns.args_json)
|
|
if not isinstance(args, list):
|
|
raise SystemExit("args_json must be a JSON array")
|
|
print(json.dumps(call(ns.method, args, base=ns.url, key_file=ns.key), indent=2))
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|