@h-rig/pi-rig 0.0.6-alpha.77 → 0.0.6-alpha.79

package/dist/src/index.js

@@ -1,4 +1,6 @@
 // @bun
+var __require = import.meta.require;
+
 // packages/pi-rig/src/client.ts
 import { existsSync, readFileSync } from "fs";
 import { homedir } from "os";
@@ -266,10 +268,11 @@ class RigBridgeClient {
     return commands.filter((entry) => Boolean(entry && typeof entry === "object" && !Array.isArray(entry)));
   }
   async workerRunCommand(command, args, runId = this.context.runId) {
+    const text = `/${command}${args.trim() ? ` ${args.trim()}` : ""}`;
     const payload = await this.piProxy("commands/run", {
       method: "POST",
       headers: { "content-type": "application/json" },
-      body: JSON.stringify({ command, args })
+      body: JSON.stringify({ text })
     }, runId);
     return payload && typeof payload === "object" && !Array.isArray(payload) ? payload : { ok: true };
   }
@@ -277,7 +280,7 @@ class RigBridgeClient {
     const payload = await this.piProxy("shell", {
       method: "POST",
       headers: { "content-type": "application/json" },
-      body: JSON.stringify({ command })
+      body: JSON.stringify({ text: command })
     }, runId);
     return payload && typeof payload === "object" && !Array.isArray(payload) ? payload : { ok: true };
   }
@@ -285,6 +288,32 @@ class RigBridgeClient {
     const payload = await this.piProxy("abort", { method: "POST" }, runId);
     return payload && typeof payload === "object" && !Array.isArray(payload) ? payload : { ok: true };
   }
+  async workerCapabilities(runId = this.context.runId) {
+    const payload = await this.piProxy("capabilities", undefined, runId);
+    return payload && typeof payload === "object" && !Array.isArray(payload) ? payload : {};
+  }
+  async workerRespondExtensionUi(requestId, response, runId = this.context.runId) {
+    const payload = await this.piProxy("extension-ui/respond", {
+      method: "POST",
+      headers: { "content-type": "application/json" },
+      body: JSON.stringify({ requestId, ...response })
+    }, runId);
+    return payload && typeof payload === "object" && !Array.isArray(payload) ? payload : { ok: true };
+  }
+  async fetchRunSessionFile(runId = this.context.runId) {
+    if (!runId)
+      return null;
+    const payload = await this.request(`/api/runs/${encodeURIComponent(runId)}/session-file`);
+    if (!payload || typeof payload !== "object" || Array.isArray(payload))
+      return null;
+    const record = payload;
+    if (record.ok !== true || typeof record.content !== "string" || !record.content.trim())
+      return null;
+    return {
+      fileName: typeof record.fileName === "string" && record.fileName.trim() ? record.fileName : `rig-run-${runId}.jsonl`,
+      content: record.content
+    };
+  }
 }
 function buildRigWebSocketUrl(serverUrl, authToken) {
   const url = new URL(serverUrl);
@@ -675,6 +704,222 @@ function createRigSlashCommands(input) {
   };
 }
 
+// packages/pi-rig/src/live-mirror.ts
+async function loadPiModules() {
+  const components = await import("@earendil-works/pi-coding-agent");
+  return { components };
+}
+function createRootContainer() {
+  const children = [];
+  return {
+    addChild(child) {
+      children.push(child);
+    },
+    removeChild(child) {
+      const index = children.indexOf(child);
+      if (index >= 0)
+        children.splice(index, 1);
+    },
+    clear() {
+      children.length = 0;
+    },
+    render(width) {
+      return children.flatMap((child) => child.render(width));
+    }
+  };
+}
+var DRONE_MESSAGE_TYPE = "rig-drone";
+var DRONE_USER_MESSAGE_TYPE = "rig-drone-user";
+function recordOf(value) {
+  return value && typeof value === "object" && !Array.isArray(value) ? value : null;
+}
+function messageRole(event) {
+  const message = recordOf(event.message);
+  return message && typeof message.role === "string" ? message.role : null;
+}
+function userMessageText(message) {
+  const content = message.content;
+  if (typeof content === "string")
+    return content;
+  if (Array.isArray(content)) {
+    return content.flatMap((block) => {
+      const record = recordOf(block);
+      return record && record.type === "text" && typeof record.text === "string" ? [record.text] : [];
+    }).join(`
+`);
+  }
+  return "";
+}
+async function createLiveMirror(input) {
+  const { pi } = input;
+  const modules = input.modules ?? await loadPiModules();
+  const cwd = input.workerCwd ?? process.cwd();
+  const turns = new Map;
+  const toolOwners = new Map;
+  const userComponents = new Map;
+  let streamingTurn = null;
+  let sequence = 0;
+  let tui = null;
+  let renderTimer = null;
+  const requestRender = () => {
+    if (renderTimer)
+      return;
+    renderTimer = setTimeout(() => {
+      renderTimer = null;
+      tui?.requestRender?.();
+    }, 33);
+    renderTimer.unref?.();
+  };
+  pi.registerMessageRenderer?.(DRONE_MESSAGE_TYPE, (message) => {
+    const details = recordOf(recordOf(message)?.details);
+    const key = details && typeof details.key === "string" ? details.key : null;
+    return key ? turns.get(key)?.root : undefined;
+  });
+  pi.registerMessageRenderer?.(DRONE_USER_MESSAGE_TYPE, (message) => {
+    const details = recordOf(recordOf(message)?.details);
+    const key = details && typeof details.key === "string" ? details.key : null;
+    return key ? userComponents.get(key) : undefined;
+  });
+  const ensureToolComponent = (turn, toolCallId, toolName, args) => {
+    const existing = turn.tools.get(toolCallId);
+    if (existing)
+      return existing;
+    const component = new modules.components.ToolExecutionComponent(toolName, toolCallId, args, {}, undefined, tui, cwd);
+    turn.root.addChild(component);
+    turn.tools.set(toolCallId, component);
+    toolOwners.set(toolCallId, turn);
+    return component;
+  };
+  const startAssistantTurn = (message) => {
+    const key = `turn-${++sequence}`;
+    const root = createRootContainer();
+    const assistant = new modules.components.AssistantMessageComponent;
+    root.addChild(assistant);
+    const turn = { root, assistant, tools: new Map };
+    assistant.updateContent(message);
+    turns.set(key, turn);
+    streamingTurn = turn;
+    pi.sendMessage?.({ customType: DRONE_MESSAGE_TYPE, content: "drone turn", display: true, details: { key } }, { triggerTurn: false });
+    requestRender();
+  };
+  const updateAssistantTurn = (message) => {
+    const turn = streamingTurn;
+    if (!turn)
+      return;
+    turn.assistant.updateContent(message);
+    const content = Array.isArray(message.content) ? message.content : [];
+    for (const block of content) {
+      const record = recordOf(block);
+      if (!record || record.type !== "toolCall")
+        continue;
+      const id = typeof record.id === "string" ? record.id : null;
+      const name = typeof record.name === "string" ? record.name : "tool";
+      if (!id)
+        continue;
+      const component = ensureToolComponent(turn, id, name, record.arguments);
+      component.updateArgs(record.arguments);
+    }
+    requestRender();
+  };
+  const endAssistantTurn = (message) => {
+    const turn = streamingTurn;
+    if (!turn)
+      return;
+    turn.assistant.updateContent(message);
+    const stopReason = typeof message.stopReason === "string" ? message.stopReason : "stop";
+    if (stopReason === "aborted" || stopReason === "error") {
+      const errorText = typeof message.errorMessage === "string" && message.errorMessage ? message.errorMessage : stopReason === "aborted" ? "Operation aborted" : "Error";
+      for (const component of turn.tools.values()) {
+        component.updateResult({ content: [{ type: "text", text: errorText }], isError: true });
+      }
+    } else {
+      for (const component of turn.tools.values()) {
+        component.setArgsComplete();
+      }
+    }
+    streamingTurn = null;
+    requestRender();
+  };
+  const mirrorUserMessage = (message) => {
+    const text = userMessageText(message).trim();
+    if (!text)
+      return;
+    const key = `user-${++sequence}`;
+    userComponents.set(key, new modules.components.UserMessageComponent(text));
+    pi.sendMessage?.({ customType: DRONE_USER_MESSAGE_TYPE, content: text, display: true, details: { key } }, { triggerTurn: false });
+    requestRender();
+  };
+  return {
+    captureTui(capturedTui) {
+      tui = recordOf(capturedTui);
+      return { render: () => [] };
+    },
+    handleWorkerEvent(event) {
+      const type = typeof event.type === "string" ? event.type : null;
+      switch (type) {
+        case "message_start": {
+          const message = recordOf(event.message);
+          if (!message)
+            return;
+          if (messageRole(event) === "assistant")
+            startAssistantTurn(message);
+          else if (messageRole(event) === "user")
+            mirrorUserMessage(message);
+          return;
+        }
+        case "message_update": {
+          const message = recordOf(event.message);
+          if (message && messageRole(event) === "assistant")
+            updateAssistantTurn(message);
+          return;
+        }
+        case "message_end": {
+          const message = recordOf(event.message);
+          if (message && messageRole(event) === "assistant")
+            endAssistantTurn(message);
+          return;
+        }
+        case "tool_execution_start": {
+          const id = typeof event.toolCallId === "string" ? event.toolCallId : null;
+          const name = typeof event.toolName === "string" ? event.toolName : "tool";
+          if (!id)
+            return;
+          const turn = toolOwners.get(id) ?? streamingTurn;
+          if (!turn)
+            return;
+          ensureToolComponent(turn, id, name, event.args).markExecutionStarted();
+          requestRender();
+          return;
+        }
+        case "tool_execution_update": {
+          const id = typeof event.toolCallId === "string" ? event.toolCallId : null;
+          const component = id ? toolOwners.get(id)?.tools.get(id) : undefined;
+          const partial = recordOf(event.partialResult);
+          if (component && partial) {
+            const content = Array.isArray(partial.content) ? partial.content : [];
+            component.updateResult({ ...partial, content, isError: false }, true);
+            requestRender();
+          }
+          return;
+        }
+        case "tool_execution_end": {
+          const id = typeof event.toolCallId === "string" ? event.toolCallId : null;
+          const component = id ? toolOwners.get(id)?.tools.get(id) : undefined;
+          const result = recordOf(event.result);
+          if (component && result) {
+            const content = Array.isArray(result.content) ? result.content : [];
+            component.updateResult({ ...result, content, isError: event.isError === true });
+            requestRender();
+          }
+          return;
+        }
+        default:
+          return;
+      }
+    }
+  };
+}
+
 // packages/pi-rig/src/tools.ts
 function textResult(text, details) {
   return { content: [{ type: "text", text }], ...details ? { details } : {} };
@@ -766,20 +1011,49 @@ function setStatus(ctx, id, text) {
     setStatusFn.call(ui, id, text);
   }
 }
-function setFooter(ctx, line) {
+function uiOf(ctx) {
   const ui = ctx && typeof ctx === "object" ? ctx.ui : null;
-  const setFooterFn = ui && typeof ui === "object" ? ui.setFooter : null;
-  if (typeof setFooterFn !== "function")
-    return;
-  setFooterFn.call(ui, () => ({
-    render(width) {
-      const max = Math.max(0, Math.trunc(width));
-      if (max === 0)
-        return [""];
-      return [line.length > max ? `${line.slice(0, Math.max(0, max - 1))}\u2026` : line];
-    },
-    invalidate() {}
-  }));
+  return ui && typeof ui === "object" ? ui : null;
+}
+function setTitle(ctx, title) {
+  const ui = uiOf(ctx);
+  const setTitleFn = ui?.setTitle;
+  if (typeof setTitleFn === "function")
+    setTitleFn.call(ui, title);
+}
+function shutdownPi(ctx) {
+  const shutdownFn = ctx && typeof ctx === "object" ? ctx.shutdown : null;
+  if (typeof shutdownFn === "function")
+    shutdownFn.call(ctx);
+}
+async function askNativeDialog(ctx, request) {
+  const ui = uiOf(ctx);
+  if (!ui)
+    return null;
+  const { method, prompt, choices } = request;
+  try {
+    if (method === "confirm" && typeof ui.confirm === "function") {
+      const confirmed = await ui.confirm("Worker request", prompt);
+      return { value: confirmed, confirmed };
+    }
+    if (choices.length > 0 && typeof ui.select === "function") {
+      const selected = await ui.select(prompt, choices);
+      return selected === undefined ? { cancelled: true } : { value: selected };
+    }
+    if (typeof ui.input === "function") {
+      const value = await ui.input("Worker request", prompt);
+      return value === undefined ? { cancelled: true } : { value };
+    }
+  } catch {
+    return { cancelled: true };
+  }
+  return null;
+}
+function shortPath(path, segments = 3) {
+  const parts = path.split("/").filter(Boolean);
+  if (parts.length <= segments)
+    return path;
+  return `\u2026/${parts.slice(-segments).join("/")}`;
 }
 function createBridgeGate(state) {
   let pending = null;
@@ -893,12 +1167,6 @@ async function handleOperatorInput(event, state, ctx, gate) {
   }
   return { action: "handled" };
 }
-function shortEntry(entry) {
-  const type = String(entry.type ?? entry.title ?? "event");
-  const text = typeof entry.text === "string" ? entry.text : typeof entry.detail === "string" ? entry.detail : typeof entry.message === "string" ? entry.message : "";
-  return `${type}: ${text}`.slice(0, 160);
-}
-var SPINNER_FRAMES = ["\u280B", "\u2819", "\u2839", "\u2838", "\u283C", "\u2834", "\u2826", "\u2827", "\u2807", "\u280F"];
 function runLocation(run) {
   const worktree = typeof run.worktreePath === "string" && run.worktreePath.trim() ? run.worktreePath.trim() : null;
   const projectRoot = typeof run.projectRoot === "string" && run.projectRoot.trim() ? run.projectRoot.trim() : null;
@@ -908,42 +1176,23 @@ function runPayload(payload) {
   return payload.run && typeof payload.run === "object" && !Array.isArray(payload.run) ? payload.run : payload;
 }
 var OPERATOR_WIDGET_WS_FALLBACK_MS = 1e4;
-function startOperatorRunWidget(state, ctx, live) {
+function startOperatorRunStatusLine(state, ctx, live) {
   if (!state.operatorSession || !state.active || !state.runId)
     return;
+  const shortId = state.runId.slice(0, 8);
   let inFlight = false;
-  let frame = 0;
   let lastRefreshAt = 0;
   const refresh = async () => {
     if (inFlight)
       return;
     inFlight = true;
     lastRefreshAt = Date.now();
-    const spinner = SPINNER_FRAMES[frame++ % SPINNER_FRAMES.length] ?? "\u2022";
     try {
-      const [runPayloadRecord, timeline] = await Promise.all([
-        state.client.attach(state.runId),
-        state.client.runTimeline(state.runId, 8).catch(() => [])
-      ]);
-      const run = runPayload(runPayloadRecord);
+      const run = runPayload(await state.client.attach(state.runId));
       const status = String(run.status ?? "unknown");
-      const header = `${spinner} Rig ${String(run.runId ?? state.runId)} \xB7 ${status}`;
-      const location = runLocation(run);
-      const detail = typeof run.statusDetail === "string" && run.statusDetail.trim() ? run.statusDetail.trim() : String(run.title ?? run.taskId ?? "");
-      const lines = [
-        header,
-        `worker: ${location}`.slice(0, 200),
-        ...detail ? [detail.slice(0, 160)] : [],
-        ...timeline.slice(-5).map(shortEntry)
-      ];
-      setStatus(ctx, "rig", `${spinner} Rig ${status}`);
-      setFooter(ctx, `${spinner} Rig ${String(run.runId ?? state.runId)} \xB7 ${status} \xB7 worker ${location}`);
-      setWidget(ctx, "rig-run", lines);
+      setStatus(ctx, "rig", `drone ${shortId} \xB7 ${status} \xB7 ${shortPath(runLocation(run))}`);
     } catch (error) {
-      const message = error instanceof Error ? error.message : String(error);
-      setStatus(ctx, "rig", `${spinner} Rig unavailable`);
-      setFooter(ctx, `${spinner} Rig ${state.runId} \xB7 unavailable \xB7 ${message}`);
-      setWidget(ctx, "rig-run", [`${spinner} Rig ${state.runId} \xB7 unavailable`, message]);
+      setStatus(ctx, "rig", `drone ${shortId} \xB7 server unreachable: ${error instanceof Error ? error.message : String(error)}`);
     } finally {
       inFlight = false;
     }
@@ -955,7 +1204,7 @@ function startOperatorRunWidget(state, ctx, live) {
       return;
     }
     refresh();
-  }, 1000);
+  }, 5000);
   unrefTimer(timer);
 }
 function operatorInboxNotification(event) {
@@ -1012,47 +1261,6 @@ function startOperatorBridge(state, ctx) {
     }
   };
 }
-var MIRROR_TEXT_LIMIT = 2000;
-function clipMirrorText(text) {
-  const trimmed = text.trim();
-  return trimmed.length > MIRROR_TEXT_LIMIT ? `${trimmed.slice(0, MIRROR_TEXT_LIMIT)}
-\u2026 [clipped; full output in the worker session]` : trimmed;
-}
-function workerMirrorText(message) {
-  const role = typeof message.role === "string" ? message.role : "assistant";
-  const content = message.content;
-  const parts = [];
-  if (typeof content === "string") {
-    parts.push(content);
-  } else if (Array.isArray(content)) {
-    for (const block of content) {
-      if (!block || typeof block !== "object" || Array.isArray(block))
-        continue;
-      const record = block;
-      if (record.type === "text" && typeof record.text === "string" && record.text.trim()) {
-        parts.push(record.text);
-        continue;
-      }
-      if (record.type === "toolCall") {
-        const name = typeof record.name === "string" ? record.name : "tool";
-        let args = "";
-        try {
-          args = record.arguments === undefined ? "" : JSON.stringify(record.arguments);
-        } catch {
-          args = "";
-        }
-        parts.push(`\u2699 ${name}(${args.length > 160 ? `${args.slice(0, 160)}\u2026` : args})`);
-      }
-    }
-  }
-  const body = parts.join(`
-`).trim();
-  if (!body)
-    return null;
-  const label = role === "assistant" ? "worker" : role === "user" ? "worker \u21D0 input" : role === "toolResult" ? `worker \u2937 ${typeof message.toolName === "string" ? message.toolName : "tool"} result` : `worker (${role})`;
-  return `[${label}]
-${clipMirrorText(body)}`;
-}
 function workerStatusLine(status) {
   const model = status.model && typeof status.model === "object" && !Array.isArray(status.model) ? status.model : null;
   const modelId = model && typeof model.id === "string" ? model.id : null;
@@ -1062,23 +1270,129 @@ function workerStatusLine(status) {
   const parts = [modelId, percent, streaming].filter((value) => Boolean(value));
   return parts.length > 0 ? `worker ${parts.join(" \xB7 ")}` : "worker session connected";
 }
+function applyRigTheme(ctx) {
+  const ui = uiOf(ctx);
+  if (!ui || typeof ui.getTheme !== "function" || typeof ui.setTheme !== "function")
+    return;
+  try {
+    const theme = ui.getTheme("rig");
+    if (theme)
+      ui.setTheme(theme);
+  } catch {}
+}
+var MICRO_DRONE_BLADES = ["---", "\\\\\\", "|||", "///"];
+var MICRO_DRONE_EYES = ["@", "o", "."];
+function applyDroneWorkingIndicator(ctx) {
+  const ui = uiOf(ctx);
+  if (!ui || typeof ui.setWorkingIndicator !== "function")
+    return;
+  const frames = Array.from({ length: 12 }, (_, index) => {
+    const blade = MICRO_DRONE_BLADES[index % MICRO_DRONE_BLADES.length];
+    const eye = MICRO_DRONE_EYES[Math.floor(index / 2) % MICRO_DRONE_EYES.length];
+    return `(${blade})${eye}(${blade})`;
+  });
+  try {
+    ui.setWorkingIndicator({ frames, intervalMs: 120 });
+  } catch {}
+}
+function renderWorkerCapabilities(capabilities) {
+  const names = (value, key = "name") => Array.isArray(value) ? value.flatMap((entry) => {
+    if (typeof entry === "string")
+      return [entry];
+    if (entry && typeof entry === "object" && !Array.isArray(entry)) {
+      const name = entry[key];
+      return typeof name === "string" ? [name] : [];
+    }
+    return [];
+  }) : [];
+  const lines = ["Drone capabilities (in effect for this run)"];
+  const tools = Array.isArray(capabilities.tools) ? capabilities.tools : [];
+  const activeTools = tools.flatMap((tool) => {
+    const record = tool && typeof tool === "object" && !Array.isArray(tool) ? tool : null;
+    return record && record.active === true && typeof record.name === "string" ? [record.name] : [];
+  });
+  const inactiveCount = tools.length - activeTools.length;
+  lines.push(`  tools       ${activeTools.join(", ") || "(none)"}${inactiveCount > 0 ? `  (+${inactiveCount} inactive)` : ""}`);
+  const extensions = Array.isArray(capabilities.extensions) ? capabilities.extensions : [];
+  const extensionLabels = extensions.flatMap((entry) => {
+    const record = entry && typeof entry === "object" && !Array.isArray(entry) ? entry : null;
+    if (!record)
+      return [];
+    const path = typeof record.path === "string" ? record.path : "";
+    const short = path.split("/").filter(Boolean).slice(-2).join("/") || path || "extension";
+    return [short];
+  });
+  lines.push(`  extensions  ${extensionLabels.join(", ") || "(none)"}`);
+  lines.push(`  hooks       ${names(capabilities.hookEvents).join(", ") || "(none)"}`);
+  lines.push(`  skills      ${names(capabilities.skills).join(", ") || "(none)"}`);
+  lines.push(`  prompts     ${names(capabilities.prompts).join(", ") || "(none)"}`);
+  const model = typeof capabilities.model === "string" ? capabilities.model : "(unknown)";
+  const thinking = typeof capabilities.thinkingLevel === "string" ? capabilities.thinkingLevel : "";
+  lines.push(`  model       ${model}${thinking ? ` \xB7 ${thinking}` : ""}`);
+  if (typeof capabilities.cwd === "string" && capabilities.cwd)
+    lines.push(`  cwd         ${capabilities.cwd}`);
+  lines.push("  (drone commands are in your palette \xB7 /worker toggles this panel)");
+  return lines;
+}
+function registerOperatorConsoleCommands(pi, state, tryRegister) {
+  if (typeof pi.registerCommand !== "function")
+    return;
+  let capabilitiesShown = false;
+  tryRegister("command:detach", () => pi.registerCommand?.("detach", {
+    description: "Detach from this run; the drone keeps flying",
+    handler: async (_args, ctx) => {
+      notify(ctx, "Detached. The drone continues on the worker.");
+      shutdownPi(ctx);
+    }
+  }));
+  tryRegister("command:stop", () => pi.registerCommand?.("stop", {
+    description: "Abort the drone's current turn and detach",
+    handler: async (_args, ctx) => {
+      try {
+        await state.client.workerAbort(state.runId);
+        notify(ctx, "Abort requested; detaching.");
+      } catch (error) {
+        notify(ctx, `Abort failed: ${error instanceof Error ? error.message : String(error)}`, "error");
+      }
+      shutdownPi(ctx);
+    }
+  }));
+  tryRegister("command:worker", () => pi.registerCommand?.("worker", {
+    description: "Toggle the drone's capability panel (tools, extensions, hooks, skills, model)",
+    handler: async (_args, ctx) => {
+      if (capabilitiesShown) {
+        const ui = uiOf(ctx);
+        if (ui && typeof ui.setWidget === "function")
+          ui.setWidget("rig-worker-capabilities", undefined);
+        capabilitiesShown = false;
+        return;
+      }
+      try {
+        const capabilities = await state.client.workerCapabilities(state.runId);
+        setWidget(ctx, "rig-worker-capabilities", renderWorkerCapabilities(capabilities));
+        capabilitiesShown = true;
+      } catch (error) {
+        notify(ctx, `Drone capabilities unavailable: ${error instanceof Error ? error.message : String(error)}`, "error");
+      }
+    }
+  }));
+}
 function startWorkerSessionMirror(pi, state, ctx) {
   if (!state.operatorSession || !state.active || !state.runId)
     return;
-  if (typeof pi.sendMessage !== "function")
-    return;
-  const mirrored = new Set;
-  const rememberMirrored = (key) => {
-    if (mirrored.has(key))
-      return false;
-    mirrored.add(key);
-    if (mirrored.size > 1000) {
-      const oldest = mirrored.values().next().value;
-      if (typeof oldest === "string")
-        mirrored.delete(oldest);
+  let mirror = null;
+  const pendingEvents = [];
+  createLiveMirror({ pi }).then((created) => {
+    mirror = created;
+    const ui = uiOf(ctx);
+    if (ui && typeof ui.setWidget === "function") {
+      ui.setWidget("rig-tui-capture", (tui) => created.captureTui(tui));
     }
-    return true;
-  };
+    for (const event of pendingEvents.splice(0))
+      created.handleWorkerEvent(event);
+  }).catch((error) => {
+    notify(ctx, `Live drone transcript unavailable: ${error instanceof Error ? error.message : String(error)}`, "error");
+  });
   const socket = new RigWorkerEventsSocket({
     context: state,
     webSocketFactory: state.webSocketFactory,
@@ -1093,44 +1407,66 @@ function startWorkerSessionMirror(pi, state, ctx) {
         if (frame.type !== "pi.event")
           return;
         const event = frame.event && typeof frame.event === "object" && !Array.isArray(frame.event) ? frame.event : null;
-        if (!event || event.type !== "message_end")
+        if (!event)
           return;
-        const message = event.message && typeof event.message === "object" && !Array.isArray(event.message) ? event.message : null;
-        if (!message)
+        if (event.type === "extension_ui_request") {
+          forwardWorkerUiRequest(state, ctx, event);
           return;
-        const key = typeof message.id === "string" && message.id.trim() ? message.id : `${String(message.role ?? "message")}:${String(message.timestamp ?? "")}`;
-        if (!rememberMirrored(key))
-          return;
-        const text = workerMirrorText(message);
-        if (!text)
-          return;
-        pi.sendMessage?.({ customType: "rig-worker-mirror", content: text, display: true }, { triggerTurn: false });
+        }
+        if (mirror)
+          mirror.handleWorkerEvent(event);
+        else
+          pendingEvents.push(event);
       },
       onConnect: () => {
-        setStatus(ctx, "rig-worker", "worker session connected");
+        setStatus(ctx, "rig-worker", "drone link live");
       },
       onDisconnect: () => {
-        setStatus(ctx, "rig-worker", "worker session disconnected (reconnecting\u2026)");
+        setStatus(ctx, "rig-worker", "drone link down (reconnecting\u2026)");
       }
     }
   });
   socket.start();
 }
-async function registerWorkerCommands(pi, state, ctx) {
-  if (!state.operatorSession || !state.active || !state.runId)
+async function forwardWorkerUiRequest(state, ctx, event) {
+  const request = event.request && typeof event.request === "object" && !Array.isArray(event.request) ? event.request : event;
+  const requestId = String(request.requestId ?? request.id ?? `ui-${state.runId}-${event.timestamp ?? ""}`);
+  const method = String(request.method ?? request.type ?? "input");
+  const prompt = typeof request.prompt === "string" && request.prompt.trim() ? request.prompt : typeof request.message === "string" && request.message.trim() ? request.message : typeof request.title === "string" && request.title.trim() ? request.title : method;
+  const rawChoices = Array.isArray(request.options) ? request.options : Array.isArray(request.items) ? request.items : [];
+  const choices = rawChoices.map((option) => {
+    if (typeof option === "string")
+      return option;
+    if (option && typeof option === "object" && !Array.isArray(option)) {
+      const record = option;
+      return String(record.label ?? record.value ?? record.name ?? "");
+    }
+    return "";
+  }).filter(Boolean);
+  const answer = await askNativeDialog(ctx, { method, prompt, choices });
+  if (!answer)
     return;
+  try {
+    await state.client.workerRespondExtensionUi(requestId, answer, state.runId);
+  } catch (error) {
+    notify(ctx, `Failed to answer the drone's question: ${error instanceof Error ? error.message : String(error)}`, "error");
+  }
+}
+async function registerWorkerCommands(pi, state, ctx, registeredNames = new Set) {
+  if (!state.operatorSession || !state.active || !state.runId)
+    return false;
   if (typeof pi.registerCommand !== "function")
-    return;
+    return false;
   let commands = [];
   try {
     commands = await state.client.workerCommands(state.runId);
   } catch {
-    return;
+    return false;
   }
   let registered = 0;
   for (const command of commands) {
     const name = typeof command.name === "string" && command.name.trim() ? command.name.trim() : null;
-    if (!name || name === "rig")
+    if (!name || name === "rig" || registeredNames.has(name))
       continue;
     const description = typeof command.description === "string" && command.description.trim() ? `[worker] ${command.description.trim()}` : "[worker] remote session command";
     try {
@@ -1145,12 +1481,47 @@ async function registerWorkerCommands(pi, state, ctx) {
           }
         }
       });
+      registeredNames.add(name);
       registered += 1;
-    } catch {}
+    } catch {
+      registeredNames.add(name);
+    }
   }
   if (registered > 0) {
     notify(ctx, `Worker session commands available: ${registered} registered from the run's Pi session.`);
   }
+  return true;
+}
+function startWorkerCommandRegistration(pi, state, ctx) {
+  const registeredNames = new Set;
+  let attempts = 0;
+  let inFlight = false;
+  const attempt = async () => {
+    if (inFlight)
+      return false;
+    inFlight = true;
+    attempts += 1;
+    try {
+      return await registerWorkerCommands(pi, state, ctx, registeredNames);
+    } finally {
+      inFlight = false;
+    }
+  };
+  attempt().then((ready) => {
+    if (ready)
+      return;
+    const timer = setInterval(() => {
+      if (attempts >= 60) {
+        clearInterval(timer);
+        return;
+      }
+      attempt().then((nextReady) => {
+        if (nextReady)
+          clearInterval(timer);
+      });
+    }, 2000);
+    unrefTimer(timer);
+  });
 }
 function startSteeringBridge(pi, state, ctx, gate, deliveredIds) {
   if (state.operatorSession || !state.active || !state.runId || typeof pi.sendUserMessage !== "function")
@@ -1243,6 +1614,9 @@ function createPiRigExtension(pi, options = {}) {
       }
     }));
   }
+  if (state.operatorSession && state.active && state.runId) {
+    registerOperatorConsoleCommands(pi, state, tryRegister);
+  }
   if (state.active && state.runId) {
     for (const tool of createRigTools({ context: state, client: state.client })) {
       tryRegister(`tool:${String(tool.name ?? "rig-tool")}`, () => pi.registerTool?.({
@@ -1262,26 +1636,24 @@ function createPiRigExtension(pi, options = {}) {
   pi.on?.("session_start", async (_event, ctx) => {
     if (!state.active || !state.runId)
       return;
-    setStatus(ctx, "rig", `Rig ${state.runId} \xB7 connecting\u2026`);
+    const shortId = state.runId.slice(0, 8);
+    setStatus(ctx, "rig", `drone ${shortId} \xB7 waiting for worker daemon\u2026`);
     if (state.operatorSession) {
-      setFooter(ctx, `Rig ${state.runId} \xB7 connecting to ${state.serverUrl ?? "Rig server"}\u2026`);
-      setWidget(ctx, "rig-run", [`Rig ${state.runId} \xB7 connecting\u2026`]);
+      setTitle(ctx, `Rig \xB7 run ${shortId}`);
+      applyRigTheme(ctx);
+      applyDroneWorkingIndicator(ctx);
     }
     const gateResult = await gate(ctx);
     if (!gateResult.allowed) {
-      if (state.operatorSession) {
-        const message = gateResult.message ?? "Rig bridge disabled (protocol mismatch).";
-        setFooter(ctx, `Rig ${state.runId} \xB7 bridge disabled (protocol mismatch)`);
-        setWidget(ctx, "rig-run", [message]);
-      }
+      setStatus(ctx, "rig", `drone ${shortId} \xB7 bridge disabled (protocol mismatch)`);
       return;
     }
-    setStatus(ctx, "rig", `Rig ${state.runId}`);
+    setStatus(ctx, "rig", `drone ${shortId} \xB7 connecting\u2026`);
     const live = gateResult.status === "compatible" ? startOperatorBridge(state, ctx) : undefined;
-    startOperatorRunWidget(state, ctx, live);
+    startOperatorRunStatusLine(state, ctx, live);
     if (state.operatorSession && gateResult.status === "compatible") {
       startWorkerSessionMirror(pi, state, ctx);
-      registerWorkerCommands(pi, state, ctx);
+      startWorkerCommandRegistration(pi, state, ctx);
     }
     await consumeQueuedSteering(pi, state, ctx, gate, deliveredSteeringIds);
   });