Merge pull request #288 from sledtools/microvm-ship

microvm.nix agent spawning

Author: justinmoon

Cargo.lock

@@ -11,6 +11,7 @@ members = [
   "crates/pika-tls",
   "crates/rmp-cli",
   "crates/pika-server",
+  "crates/vm-spawner",
 ]
 
 [workspace.dependencies]

Cargo.toml

@@ -15,6 +15,7 @@
 import pty
 import select
 import signal
+import shlex
 import struct
 import subprocess
 import sys
@@ -102,9 +103,31 @@ def emit_pi_event(group_id: str, payload: dict) -> None:
     )
 
 
+def build_publish_keypackage_cmd():
+    raw = os.environ.get("PI_RELAYS_JSON", "").strip()
+    if not raw:
+        return {"cmd": "publish_keypackage"}
+    try:
+        parsed = json.loads(raw)
+    except Exception as err:
+        log(f"invalid PI_RELAYS_JSON, ignoring: {err}")
+        return {"cmd": "publish_keypackage"}
+    if not isinstance(parsed, list):
+        return {"cmd": "publish_keypackage"}
+    relays = [str(item).strip() for item in parsed if str(item).strip()]
+    if not relays:
+        return {"cmd": "publish_keypackage"}
+    return {"cmd": "publish_keypackage", "relays": relays}
+
+
 def spawn_pi_rpc(label: str) -> subprocess.Popen[bytes]:
     env = os.environ.copy()
-    cmd = ["pi", "--mode", "rpc", "--no-session", "--provider", "anthropic"]
+    cmd = shlex.split(
+        os.environ.get(
+            "PI_CMD",
+            "pi --mode rpc --no-session --provider anthropic",
+        )
+    )
     model = os.environ.get("PI_MODEL")
     if model:
         cmd.extend(["--model", model])
@@ -981,8 +1004,7 @@ def main() -> None:
         if msg_type == "ready":
             my_pubkey = msg.get("pubkey")
             log(f"marmotd ready, pubkey={my_pubkey}")
-            send_to_marmotd({"cmd": "publish_keypackage"})
-            continue
+            send_to_marmotd(build_publish_keypackage_cmd())
 
         if msg_type == "call_invite_received":
             call_id = str(msg.get("call_id", ""))

bots/pi-bridge.py

@@ -0,0 +1,55 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+my_pubkey=""
+
+trim() {
+  local s="$1"
+  s="${s#"${s%%[![:space:]]*}"}"
+  s="${s%"${s##*[![:space:]]}"}"
+  printf '%s' "$s"
+}
+
+emit_publish_keypackage_cmd() {
+  if [[ -n "${PI_RELAYS_JSON:-}" ]] && jq -e 'type == "array" and length > 0' >/dev/null 2>&1 <<<"${PI_RELAYS_JSON}"; then
+    jq -cn --argjson relays "${PI_RELAYS_JSON}" '{"cmd":"publish_keypackage","relays":$relays}'
+  else
+    printf '{"cmd":"publish_keypackage"}\n'
+  fi
+}
+
+while IFS= read -r line; do
+  [[ -z "${line}" ]] && continue
+
+  msg_type="$(printf '%s' "$line" | jq -r '.type // empty' 2>/dev/null || true)"
+  case "$msg_type" in
+    ready)
+      my_pubkey="$(printf '%s' "$line" | jq -r '.pubkey // ""' 2>/dev/null || true)"
+      emit_publish_keypackage_cmd
+      ;;
+    message_received)
+      from_pubkey="$(printf '%s' "$line" | jq -r '.from_pubkey // ""' 2>/dev/null || true)"
+      if [[ -n "$my_pubkey" && "$from_pubkey" == "$my_pubkey" ]]; then
+        continue
+      fi
+
+      group_id="$(printf '%s' "$line" | jq -r '.nostr_group_id // ""' 2>/dev/null || true)"
+      content="$(printf '%s' "$line" | jq -r '.content // ""' 2>/dev/null || true)"
+
+      response=""
+      if [[ "$content" =~ [Rr]eply[[:space:]]+with[[:space:]]+exactly:[[:space:]]*(.+)$ ]]; then
+        response="$(trim "${BASH_REMATCH[1]}")"
+      fi
+      if [[ -z "$response" ]]; then
+        response="ack: $content"
+      fi
+
+      jq -cn \
+        --arg gid "$group_id" \
+        --arg content "$response" \
+        '{"cmd":"send_message","nostr_group_id":$gid,"content":$content}'
+      ;;
+    *)
+      ;;
+  esac
+done