runtime-sdk 0.4.11__tar.gz → 0.4.12__tar.gz

{runtime_sdk-0.4.11 → runtime_sdk-0.4.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: runtime-sdk
-Version: 0.4.11
+Version: 0.4.12
 Summary: Runtime Python SDK and CLI
 Project-URL: Repository, https://github.com/The-Money-Company-Limited/runtimevm
 Project-URL: Issues, https://github.com/The-Money-Company-Limited/runtimevm/issues
@@ -52,6 +52,9 @@ runtime verify 123456
 runtime whoami
 runtime logout
 runtime login --api-key rt_live_...
+runtime api-keys list
+runtime api-keys create "my laptop"
+runtime api-keys revoke 123
 
 # Computers
 runtime create                       # creates a computer with the starter app already published

{runtime_sdk-0.4.11 → runtime_sdk-0.4.12}/README.md

@@ -33,6 +33,9 @@ runtime verify 123456
 runtime whoami
 runtime logout
 runtime login --api-key rt_live_...
+runtime api-keys list
+runtime api-keys create "my laptop"
+runtime api-keys revoke 123
 
 # Computers
 runtime create                       # creates a computer with the starter app already published

{runtime_sdk-0.4.11 → runtime_sdk-0.4.12}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "runtime-sdk"
-version = "0.4.11"
+version = "0.4.12"
 description = "Runtime Python SDK and CLI"
 readme = "README.md"
 requires-python = ">=3.11"

{runtime_sdk-0.4.11 → runtime_sdk-0.4.12}/scripts/runtime_sdk/cli.py

@@ -165,6 +165,14 @@ def build_parser() -> argparse.ArgumentParser:
     subparsers.add_parser("whoami")
     subparsers.add_parser("logout")
 
+    api_keys_cmd = subparsers.add_parser("api-keys", help="Manage Runtime API keys")
+    api_keys_subparsers = api_keys_cmd.add_subparsers(dest="api_keys_action", required=True)
+    api_keys_subparsers.add_parser("list", aliases=["ls"], help="List API keys")
+    api_keys_create = api_keys_subparsers.add_parser("create", help="Create a new API key")
+    api_keys_create.add_argument("label", nargs="?", default=None, help="Human-friendly label for the new API key")
+    api_keys_revoke = api_keys_subparsers.add_parser("revoke", help="Revoke an API key")
+    api_keys_revoke.add_argument("id", nargs="?", default=None, help="API key id")
+
     create_cmd = subparsers.add_parser("create", help="Create a new computer with the starter app published by default")
     create_cmd.add_argument(
         "name",
@@ -280,6 +288,14 @@ def main(argv: list[str] | None = None) -> int:
             return handle_whoami(config)
         if args.command == "logout":
             return handle_logout(config)
+        if args.command == "api-keys":
+            if args.api_keys_action in ("list", "ls"):
+                return handle_api_keys_list(config)
+            if args.api_keys_action == "create":
+                return handle_api_keys_create(config, args.label)
+            if args.api_keys_action == "revoke":
+                return handle_api_keys_revoke(config, args.id)
+            return report_error("unknown api-keys command")
         if args.command == "create":
             return handle_create(
                 config,
@@ -543,7 +559,7 @@ def _spawn_proxy_agent(base_url: str, api_key: str, computer_id: str, display_na
     finally:
         log_file.close()
 
-    deadline = time.time() + 4.0
+    deadline = time.time() + max(4.0, DEFAULT_WARMUP_TIMEOUT + 2.0)
     state_path = _proxy_state_path(local_port)
     while time.time() < deadline:
         if state_path.exists():
@@ -715,8 +731,8 @@ def _computer_url_label(c: dict[str, Any]) -> str:
 
 
 
-def _computer_created_label(c: dict[str, Any]) -> str:
-    raw = str(c.get("created_at") or "").strip()
+def _relative_time_label(value: Any) -> str:
+    raw = str(value or "").strip()
     if not raw:
         return ""
     try:
@@ -738,6 +754,11 @@ def _computer_created_label(c: dict[str, Any]) -> str:
 
 
 
+def _computer_created_label(c: dict[str, Any]) -> str:
+    return _relative_time_label(c.get("created_at"))
+
+
+
 def _computer_status_style(status: str) -> str:
     if "cold" in status:
         return "bold cyan"
@@ -1072,6 +1093,84 @@ def _render_whoami_panel(payload: dict[str, Any]) -> None:
     _UI.console().print(Panel(body, title="whoami", border_style="cyan", expand=False))
 
 
+
+def _api_key_status_label(key: dict[str, Any]) -> str:
+    if key.get("revoked_at"):
+        return "revoked"
+    if key.get("last_used_at"):
+        return "used"
+    return "unused"
+
+
+
+def _api_key_status_style(status: str) -> str:
+    if status == "revoked":
+        return "bold red"
+    if status == "used":
+        return "bold green"
+    return "yellow"
+
+
+
+def _render_api_keys_table(payload: dict[str, Any]) -> None:
+    mods = _UI.rich()
+    Table = mods["Table"]
+    api_keys = payload.get("api_keys") or []
+    if not isinstance(api_keys, list) or not api_keys:
+        _UI.console().print("[yellow]no api keys found[/yellow]")
+        return
+
+    table = Table(title="API keys", expand=True, box=mods["box"].SIMPLE_HEAVY)
+    table.add_column("ID", style="dim", no_wrap=True)
+    table.add_column("Label", style="bold white")
+    table.add_column("Prefix", style="white", no_wrap=True)
+    table.add_column("Created", style="dim", no_wrap=True, justify="right")
+    table.add_column("Last used", style="dim", no_wrap=True, justify="right")
+    table.add_column("Status", no_wrap=True)
+
+    for entry in api_keys:
+        if not isinstance(entry, dict):
+            continue
+        status = _api_key_status_label(entry)
+        style = _api_key_status_style(status)
+        table.add_row(
+            str(entry.get("id") or "—"),
+            str(entry.get("label") or "—"),
+            str(entry.get("key_prefix") or "—"),
+            _relative_time_label(entry.get("created_at")) or "—",
+            _relative_time_label(entry.get("last_used_at")) or "—",
+            f"[{style}]{status}[/{style}]",
+        )
+
+    _UI.console().print(table)
+
+
+
+def _render_api_key_created(payload: dict[str, Any]) -> None:
+    mods = _UI.rich()
+    Panel, Text = mods["Panel"], mods["Text"]
+    key = payload.get("key") or {}
+    secret = str(payload.get("api_key") or "")
+    body = Text()
+    rows = [
+        ("id", key.get("id")),
+        ("label", key.get("label")),
+        ("prefix", key.get("key_prefix")),
+    ]
+    for label, value in rows:
+        if value is None:
+            continue
+        body.append(f"{label:<12}", style="dim")
+        body.append(f"{value}\n")
+    if secret:
+        body.append("save now     ", style="bold yellow")
+        body.append("this secret is only shown once\n", style="yellow")
+        body.append("api key      ", style="dim")
+        body.append(secret, style="bold white")
+        body.append("\n")
+    _UI.console().print(Panel(body, title="api key created", border_style="cyan", expand=False))
+
+
 # --------------------------------------------------------------------------- #
 # Command handlers                                                            #
 # --------------------------------------------------------------------------- #
@@ -1220,6 +1319,56 @@ def handle_logout(config: RuntimeConfig) -> int:
     return report_success({"message": message}, render)
 
 
+
+def handle_api_keys_list(config: RuntimeConfig) -> int:
+    if (err := _require_api_key(config)) is not None:
+        return err
+    client = RuntimeClient(base_url=config.base_url, api_key=config.api_key)
+    return report_success(client.list_api_keys(), _render_api_keys_table)
+
+
+
+def handle_api_keys_create(config: RuntimeConfig, label: str | None) -> int:
+    if (err := _require_api_key(config)) is not None:
+        return err
+    if label is None:
+        if not _interactive():
+            return report_error("label is required")
+        label = _prompt_text("API key label")
+        if not label:
+            return report_error("label is required")
+
+    client = RuntimeClient(base_url=config.base_url, api_key=config.api_key)
+    result = client.create_api_key(label)
+    return report_success(result, _render_api_key_created)
+
+
+
+def handle_api_keys_revoke(config: RuntimeConfig, key_id: str | None) -> int:
+    if (err := _require_api_key(config)) is not None:
+        return err
+    if key_id is None:
+        if not _interactive():
+            return report_error("api key id is required")
+        key_id = _prompt_text("API key id")
+        if not key_id:
+            return report_error("api key id is required")
+
+    try:
+        resolved_key_id = int(str(key_id).strip())
+    except ValueError:
+        return report_error("api key id must be a number")
+
+    client = RuntimeClient(base_url=config.base_url, api_key=config.api_key)
+    client.revoke_api_key(resolved_key_id)
+
+    def render(_: dict[str, Any]) -> None:
+        _UI.console().print(f"[green]✓[/green] api key {resolved_key_id} revoked")
+
+    return report_success({"id": resolved_key_id, "message": f"api key {resolved_key_id} revoked"}, render)
+
+
+
 def handle_create(
     config: RuntimeConfig,
     name: str | None,
@@ -1633,6 +1782,10 @@ def handle_proxy_start(config: RuntimeConfig, computer_id: str | None, specs: li
             if started_ports:
                 _stop_proxy_ports(started_ports, stop_all=False)
             return report_error(str(exc), status_code=exc.status_code)
+        except Exception as exc:
+            if started_ports:
+                _stop_proxy_ports(started_ports, stop_all=False)
+            return report_error(str(exc))
         started_ports.append(local_port)
         started.append({"computer_id": resolved_id, "computer": computer_id, "local_port": local_port, "remote_port": remote_port})
 

{runtime_sdk-0.4.11 → runtime_sdk-0.4.12}/scripts/runtime_sdk/client.py

@@ -43,6 +43,15 @@ class RuntimeClient:
     def logout(self) -> dict[str, Any]:
         return self._request("POST", "/api/auth/logout", auth_required=True)
 
+    def list_api_keys(self) -> dict[str, Any]:
+        return self._request("GET", "/api/api-keys", auth_required=True)
+
+    def create_api_key(self, label: str) -> dict[str, Any]:
+        return self._request("POST", "/api/api-keys", json={"label": label}, auth_required=True)
+
+    def revoke_api_key(self, key_id: int) -> dict[str, Any]:
+        return self._request("POST", f"/api/api-keys/{key_id}/revoke", auth_required=True)
+
     # --- Computers ---
 
     def create_computer(

{runtime_sdk-0.4.11 → runtime_sdk-0.4.12}/scripts/runtime_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: runtime-sdk
-Version: 0.4.11
+Version: 0.4.12
 Summary: Runtime Python SDK and CLI
 Project-URL: Repository, https://github.com/The-Money-Company-Limited/runtimevm
 Project-URL: Issues, https://github.com/The-Money-Company-Limited/runtimevm/issues
@@ -52,6 +52,9 @@ runtime verify 123456
 runtime whoami
 runtime logout
 runtime login --api-key rt_live_...
+runtime api-keys list
+runtime api-keys create "my laptop"
+runtime api-keys revoke 123
 
 # Computers
 runtime create                       # creates a computer with the starter app already published