@h-rig/cli 0.0.6-alpha.40 → 0.0.6-alpha.42

package/dist/src/commands/_pi-worker-bridge-extension.js

@@ -248,433 +248,272 @@ async function buildRunPiEventsWebSocketUrl(context, runId) {
   return url.toString();
 }
 
-// packages/cli/src/commands/_pi-worker-bridge-extension.ts
+// packages/cli/src/commands/_pi-remote-session.ts
+import { AgentSession as PiAgentSession } from "@earendil-works/pi-coding-agent";
 var TERMINAL_RUN_STATUSES = new Set(["completed", "failed", "stopped", "cancelled", "canceled", "closed", "merged", "needs_attention", "needs-attention"]);
-var MAX_TRANSCRIPT_LINES = 120;
+function defaultTransport() {
+  return {
+    getSession: getRunPiSessionViaServer,
+    getMessages: getRunPiMessagesViaServer,
+    getStatus: getRunPiStatusViaServer,
+    getCommands: getRunPiCommandsViaServer,
+    sendPrompt: sendRunPiPromptViaServer,
+    sendShell: sendRunPiShellViaServer,
+    runCommand: runRunPiCommandViaServer,
+    abort: abortRunPiViaServer,
+    buildEventsWebSocketUrl: buildRunPiEventsWebSocketUrl
+  };
+}
 function recordOf(value) {
   return value && typeof value === "object" && !Array.isArray(value) ? value : null;
 }
-function asText(value) {
-  if (typeof value === "string")
-    return value;
-  if (value === null || value === undefined)
-    return "";
-  if (typeof value === "number" || typeof value === "boolean")
-    return String(value);
-  try {
-    return JSON.stringify(value);
-  } catch {
-    return String(value);
-  }
-}
-function textFromContent(content) {
-  if (typeof content === "string")
-    return content;
-  if (!Array.isArray(content))
-    return asText(content);
-  return content.flatMap((part) => {
-    const item = recordOf(part);
-    if (!item)
-      return [];
-    if (typeof item.text === "string")
-      return [item.text];
-    if (typeof item.content === "string")
-      return [item.content];
-    if (item.type === "toolCall")
-      return [`\u23FA ${String(item.name ?? "tool")} ${asText(item.arguments ?? "")}`.trim()];
-    if (item.type === "toolResult")
-      return [`\u21B3 ${asText(item.content ?? item.result ?? "")}`.trim()];
-    return [];
-  }).join(`
-`);
-}
-function appendTranscript(state, label, text) {
-  const trimmed = text.trimEnd();
-  if (!trimmed)
-    return;
-  const lines = trimmed.split(/\r?\n/);
-  state.transcript.push(`${label}: ${lines[0] ?? ""}`);
-  for (const line of lines.slice(1))
-    state.transcript.push(`  ${line}`);
-  if (state.transcript.length > MAX_TRANSCRIPT_LINES) {
-    state.transcript.splice(0, state.transcript.length - MAX_TRANSCRIPT_LINES);
-  }
-}
-function nativePiUi(ctx) {
-  const ui = ctx.ui;
-  return typeof ui.emitSessionEvent === "function" && typeof ui.appendSessionMessages === "function" ? ui : null;
-}
-function syncNativeDisplayCwd(ctx, state) {
-  const ui = nativePiUi(ctx);
-  if (ui?.setDisplayCwd && state.cwd)
-    ui.setDisplayCwd(state.cwd);
-}
-function parseExtensionUiRequest(value) {
-  const request = recordOf(value) ?? {};
-  const requestId = String(request.requestId ?? request.id ?? `ui-${Date.now()}`);
-  const method = String(request.method ?? request.type ?? "input");
-  const prompt = asText(request.prompt ?? request.message ?? request.title ?? method);
-  const rawOptions = Array.isArray(request.options) ? request.options : Array.isArray(request.items) ? request.items : [];
-  const options = rawOptions.map((option) => {
-    const record = recordOf(option);
-    return record ? asText(record.label ?? record.value ?? record.name ?? option) : asText(option);
-  }).filter(Boolean);
-  return { requestId, method, prompt, options };
-}
-function renderBridgeWidget(state) {
-  const statusParts = [
-    state.wsConnected ? "live" : "connecting",
-    state.status,
-    state.model,
-    state.cwd
-  ].filter(Boolean);
-  const lines = [`Rig worker session \xB7 ${statusParts.join(" \xB7 ")}`];
-  if (state.activity)
-    lines.push(state.activity);
-  if (state.commands.length > 0) {
-    lines.push(`Worker commands: ${state.commands.slice(0, 10).join(", ")}${state.commands.length > 10 ? ", \u2026" : ""}`);
-  }
-  lines.push("");
-  if (state.transcript.length > 0) {
-    lines.push(...state.transcript.slice(-MAX_TRANSCRIPT_LINES));
-  } else {
-    lines.push("Waiting for the worker session transcript\u2026 (/detach exits and leaves the worker running)");
-  }
-  if (state.pendingUi) {
-    lines.push("");
-    lines.push(`Worker needs input \xB7 ${state.pendingUi.method}`);
-    lines.push(state.pendingUi.prompt);
-    state.pendingUi.options.forEach((option, index) => lines.push(`${index + 1}. ${option}`));
-    lines.push("Reply in the editor below. /cancel dismisses this request.");
-  }
-  return lines;
-}
-function reportBridgeError(ctx, state, message) {
-  appendTranscript(state, "Error", message);
-  state.status = message;
-  try {
-    ctx.ui.notify(message, "error");
-  } catch {}
-  updatePiUi(ctx, state);
-}
-function updatePiUi(ctx, state) {
-  ctx.ui.setTitle("Rig \xB7 worker session");
-  ctx.ui.setStatus("rig-worker-pi", state.wsConnected ? "worker session live" : state.status);
-  syncNativeDisplayCwd(ctx, state);
-  if (state.nativeStream && nativePiUi(ctx)) {
-    ctx.ui.setWidget("rig-worker-pi-transcript", undefined);
-    return;
-  }
-  ctx.ui.setWorkingVisible(false);
-  ctx.ui.setWidget("rig-worker-pi-transcript", renderBridgeWidget(state), { placement: "aboveEditor" });
-}
-function applyStatus(state, payload) {
-  const status = recordOf(payload.status) ?? payload;
-  state.streaming = status.isStreaming === true || status.isCompacting === true || status.isBashRunning === true;
-  state.cwd = typeof status.cwd === "string" ? status.cwd : state.cwd;
-  state.model = typeof status.model === "string" ? status.model : state.model;
-  const pending = typeof status.pendingMessageCount === "number" ? status.pendingMessageCount : 0;
-  state.status = `${state.streaming ? "streaming" : "idle"}${pending ? ` \xB7 ${pending} queued` : ""}`;
-}
-function applyMessage(state, message) {
-  const record = recordOf(message);
-  if (!record)
-    return;
-  const role = String(record.role ?? "system");
-  const label = role === "assistant" ? "Pi" : role === "user" ? "You" : role === "tool" || role === "toolResult" ? "Tool" : "System";
-  appendTranscript(state, label, textFromContent(record.content ?? record.message ?? record.text ?? ""));
-}
-function applyPiEvent(ctx, state, eventValue) {
-  const event = recordOf(eventValue);
-  if (!event)
-    return;
-  const type = String(event.type ?? "event");
-  if (type === "agent_start") {
-    state.streaming = true;
-    state.status = "streaming";
-  } else if (type === "agent_end") {
-    state.streaming = false;
-    state.status = "idle";
-  } else if (type === "queue_update") {
-    const steering = Array.isArray(event.steering) ? event.steering.length : 0;
-    const followUp = Array.isArray(event.followUp) ? event.followUp.length : 0;
-    state.status = `queued \xB7 steer ${steering} \xB7 follow-up ${followUp}`;
-  }
-  const native = nativePiUi(ctx);
-  if (state.nativeStream && native?.emitSessionEvent) {
-    native.emitSessionEvent(eventValue);
-    return;
-  }
-  if (type === "agent_end") {
-    appendTranscript(state, "System", "Agent turn complete.");
-    return;
-  }
-  if (type === "message_start" || type === "message_end" || type === "turn_end") {
-    applyMessage(state, event.message);
-    return;
-  }
-  if (type === "message_update") {
-    const assistantEvent = recordOf(event.assistantMessageEvent);
-    const delta = typeof assistantEvent?.delta === "string" ? assistantEvent.delta : typeof assistantEvent?.text === "string" ? assistantEvent.text : "";
-    if (delta)
-      appendTranscript(state, assistantEvent?.type === "thinking_delta" ? "Thinking" : "Pi", delta);
-    return;
-  }
-  if (type === "tool_execution_start") {
-    appendTranscript(state, "Tool", `${String(event.toolName ?? "tool")} ${asText(event.args ?? "")}`.trim());
-    return;
-  }
-  if (type === "tool_execution_update") {
-    appendTranscript(state, "Tool", asText(event.partialResult ?? ""));
-    return;
-  }
-  if (type === "tool_execution_end") {
-    appendTranscript(state, event.isError === true ? "Error" : "Tool", asText(event.result ?? `${String(event.toolName ?? "tool")} complete`));
-  }
-}
-function firstPendingShell(state) {
-  return state.pendingShells[0];
-}
-function finishPendingShell(state, shell, result) {
-  const index = state.pendingShells.indexOf(shell);
-  if (index !== -1)
-    state.pendingShells.splice(index, 1);
-  shell.resolve(result);
-}
-function failPendingShell(state, shell, error) {
-  const index = state.pendingShells.indexOf(shell);
-  if (index !== -1)
-    state.pendingShells.splice(index, 1);
-  shell.reject(error);
-}
-function applyUiEvent(state, value) {
-  const event = recordOf(value);
-  if (!event)
-    return;
-  const type = String(event.type ?? "ui");
-  if (type === "shell.chunk") {
-    const pending = firstPendingShell(state);
-    const chunk = asText(event.chunk);
-    if (pending) {
-      pending.sawChunk = true;
-      pending.onData(Buffer.from(chunk));
-    } else {
-      appendTranscript(state, "Tool", chunk);
-    }
-    return;
-  }
-  if (type === "shell.end") {
-    const pending = firstPendingShell(state);
-    const output = asText(event.output ?? "");
-    const exitCode = typeof event.exitCode === "number" ? event.exitCode : event.isError === true ? 1 : 0;
-    if (pending) {
-      if (output && !pending.sawChunk)
-        pending.onData(Buffer.from(output));
-      finishPendingShell(state, pending, { exitCode });
-    } else {
-      appendTranscript(state, event.isError === true ? "Error" : "Tool", output || `exit ${String(exitCode)}`);
-    }
-    return;
-  }
-  if (type === "shell.start") {
-    if (!firstPendingShell(state))
-      appendTranscript(state, "Tool", `$ ${asText(event.command)}`);
-    return;
-  }
-  appendTranscript(state, "System", `${type}: ${asText(event)}`);
-}
-function applyEnvelope(ctx, state, envelopeValue) {
-  const envelope = recordOf(envelopeValue);
-  if (!envelope)
-    return;
-  const type = String(envelope.type ?? "");
-  if (type === "ready") {
-    const metadata = recordOf(envelope.metadata);
-    state.cwd = typeof metadata?.cwd === "string" ? metadata.cwd : state.cwd;
-    state.status = "worker Pi daemon ready";
-    if (!state.nativeStream)
-      appendTranscript(state, "System", "Connected to worker Pi daemon.");
-  } else if (type === "status.update") {
-    applyStatus(state, envelope);
-  } else if (type === "activity.update") {
-    const activity = recordOf(envelope.activity);
-    state.activity = [activity?.label, activity?.detail].map(asText).filter(Boolean).join(" \u2014 ");
-  } else if (type === "extension_ui_request") {
-    state.pendingUi = parseExtensionUiRequest(envelope.request);
-    appendTranscript(state, "System", `Extension UI request: ${state.pendingUi.prompt}`);
-  } else if (type === "pi.ui_event") {
-    applyUiEvent(state, envelope.event);
-  } else if (type === "pi.event") {
-    applyPiEvent(ctx, state, envelope.event);
-  } else if (type === "error") {
-    appendTranscript(state, "Error", asText(envelope.message ?? envelope.detail ?? "unknown error"));
-  }
-  syncNativeDisplayCwd(ctx, state);
+function emptyRemoteStatus() {
+  return {
+    isStreaming: false,
+    isCompacting: false,
+    isBashRunning: false,
+    pendingMessageCount: 0,
+    steeringMessages: [],
+    followUpMessages: [],
+    model: null,
+    thinkingLevel: null,
+    sessionName: null,
+    cwd: null,
+    stats: null,
+    contextUsage: null
+  };
 }
 function resolveAttachReadyTimeoutMs() {
   const raw = Number.parseInt(process.env.RIG_PI_ATTACH_TIMEOUT_MS ?? "", 10);
   return Number.isFinite(raw) && raw > 0 ? raw : 10 * 60000;
 }
-function formatElapsed(sinceMs) {
-  const totalSeconds = Math.floor((Date.now() - sinceMs) / 1000);
-  const minutes = Math.floor(totalSeconds / 60);
-  const seconds = totalSeconds % 60;
-  return minutes > 0 ? `${minutes}m${String(seconds).padStart(2, "0")}s` : `${seconds}s`;
-}
-async function waitForWorkerReady(options, ctx, state) {
-  const startedAt = Date.now();
-  const deadline = startedAt + resolveAttachReadyTimeoutMs();
-  let consecutiveFailures = 0;
-  while (true) {
-    let requestFailed = false;
-    const session = await getRunPiSessionViaServer(options.context, options.runId).catch((error) => {
-      requestFailed = true;
-      return {
-        ready: false,
-        status: error instanceof Error ? error.message : String(error),
-        retryAfterMs: 1000
-      };
-    });
-    if (session.ready === false) {
-      consecutiveFailures = requestFailed ? consecutiveFailures + 1 : 0;
-      const status = String(session.status ?? "starting");
-      if (TERMINAL_RUN_STATUSES.has(status.toLowerCase())) {
-        reportBridgeError(ctx, state, `Run ended before worker Pi daemon became ready: ${status}. Inspect with \`rig run show ${options.runId}\`; restart with \`rig task run --task <id>\`.`);
-        return false;
-      }
-      if (consecutiveFailures >= 5) {
-        state.status = `Rig server unreachable \xB7 retrying (${formatElapsed(startedAt)}) \xB7 /detach to exit`;
-      } else {
-        state.status = `waiting for worker Pi daemon \xB7 ${status} \xB7 ${formatElapsed(startedAt)} \xB7 /detach to exit`;
-      }
-      updatePiUi(ctx, state);
-      if (Date.now() >= deadline) {
-        reportBridgeError(ctx, state, `Worker Pi daemon did not become ready within ${formatElapsed(startedAt)} (last status: ${status}). ` + `Check \`rig run show ${options.runId}\` and \`rig doctor\`; the run may have been restarted by a server deploy. ` + "Set RIG_PI_ATTACH_TIMEOUT_MS to wait longer.");
-        return false;
-      }
-      await Bun.sleep(typeof session.retryAfterMs === "number" ? session.retryAfterMs : 750);
-      continue;
-    }
-    const sessionRecord = recordOf(session) ?? {};
-    applyEnvelope(ctx, state, { type: "ready", metadata: sessionRecord.metadata ?? sessionRecord });
-    updatePiUi(ctx, state);
-    return true;
-  }
-}
-function parseWsPayload(message) {
-  if (typeof message.data === "string")
-    return JSON.parse(message.data);
-  return JSON.parse(Buffer.from(message.data).toString("utf8"));
-}
-var BRIDGE_LOCAL_COMMANDS = new Set(["detach", "quit", "q", "stop"]);
-function registerDaemonCommandsNatively(pi, options, ctx, state, commands, registered) {
-  for (const command of commands) {
-    const record = recordOf(command);
-    const name = typeof record?.name === "string" ? record.name : "";
-    if (!name || registered.has(name) || BRIDGE_LOCAL_COMMANDS.has(name))
-      continue;
-    registered.add(name);
-    const description = typeof record?.description === "string" ? record.description : undefined;
-    const source = typeof record?.source === "string" ? record.source : "worker";
-    try {
-      pi.registerCommand(name, {
-        description: `[worker ${source}] ${description ?? ""}`.trim(),
-        handler: async (args) => {
-          const text = `/${name}${args ? ` ${args}` : ""}`;
-          appendTranscript(state, "You", text);
-          try {
-            const result = await runRunPiCommandViaServer(options.context, options.runId, text);
-            const message = typeof result.message === "string" ? result.message : "worker command accepted";
-            appendTranscript(state, "System", message);
-            if (state.nativeStream)
-              ctx.ui.notify(message, "info");
-          } catch (error) {
-            reportBridgeError(ctx, state, error instanceof Error ? error.message : String(error));
-          }
-          updatePiUi(ctx, state);
-        }
-      });
-    } catch {}
-  }
-}
-async function connectWorkerStream(options, pi, ctx, state, registeredDaemonCommands) {
-  const ready = await waitForWorkerReady(options, ctx, state);
-  if (!ready)
-    return;
-  let catchupDone = false;
-  const buffered = [];
-  const wsUrl = await buildRunPiEventsWebSocketUrl(options.context, options.runId);
-  const socket = new WebSocket(wsUrl);
-  const closePromise = new Promise((resolve3) => {
+
+class RigRemoteSessionController {
+  context;
+  runId;
+  status = emptyRemoteStatus();
+  transport;
+  hooks = {};
+  session = null;
+  socket = null;
+  closed = false;
+  pendingShells = [];
+  pendingCompactions = [];
+  constructor(input) {
+    this.context = input.context;
+    this.runId = input.runId;
+    this.transport = input.transport ?? defaultTransport();
+  }
+  ingestEnvelope(envelopeValue) {
+    this.applyEnvelope(envelopeValue);
+  }
+  bindSession(session) {
+    this.session = session;
+  }
+  setUiHooks(hooks) {
+    this.hooks = hooks;
+  }
+  async connect() {
+    const ready = await this.waitForReady();
+    if (!ready || this.closed)
+      return;
+    let catchupDone = false;
+    const buffered = [];
+    const wsUrl = await this.transport.buildEventsWebSocketUrl(this.context, this.runId);
+    const socket = new WebSocket(wsUrl);
+    this.socket = socket;
     socket.onopen = () => {
-      state.wsConnected = true;
-      state.status = "live worker Pi WebSocket connected";
-      updatePiUi(ctx, state);
+      this.hooks.onConnectionChange?.(true);
+      this.hooks.onStatusText?.("worker session live");
     };
     socket.onmessage = (message) => {
       try {
-        const payload = parseWsPayload(message);
+        const payload = typeof message.data === "string" ? JSON.parse(message.data) : JSON.parse(Buffer.from(message.data).toString("utf8"));
         if (!catchupDone)
           buffered.push(payload);
-        else {
-          applyEnvelope(ctx, state, payload);
-          updatePiUi(ctx, state);
-        }
+        else
+          this.applyEnvelope(payload);
       } catch (error) {
-        appendTranscript(state, "Error", `Unparseable worker Pi event: ${error instanceof Error ? error.message : String(error)}`);
-        updatePiUi(ctx, state);
+        this.hooks.onError?.(`Unparseable worker Pi event: ${error instanceof Error ? error.message : String(error)}`);
       }
     };
     socket.onerror = () => socket.close();
     socket.onclose = () => {
-      state.wsConnected = false;
-      state.status = "worker Pi WebSocket disconnected";
-      updatePiUi(ctx, state);
-      resolve3();
+      this.hooks.onConnectionChange?.(false);
+      if (!this.closed)
+        this.hooks.onStatusText?.("worker Pi WebSocket disconnected");
     };
-  });
-  try {
-    const [messagesPayload, statusPayload, commandsPayload] = await Promise.all([
-      getRunPiMessagesViaServer(options.context, options.runId),
-      getRunPiStatusViaServer(options.context, options.runId),
-      getRunPiCommandsViaServer(options.context, options.runId)
-    ]);
-    const messages = Array.isArray(messagesPayload.messages) ? messagesPayload.messages : [];
-    const native = nativePiUi(ctx);
-    if (state.nativeStream && native?.appendSessionMessages)
-      native.appendSessionMessages(messages);
+    try {
+      const [messagesPayload, statusPayload, commandsPayload] = await Promise.all([
+        this.transport.getMessages(this.context, this.runId),
+        this.transport.getStatus(this.context, this.runId),
+        this.transport.getCommands(this.context, this.runId)
+      ]);
+      const messages = Array.isArray(messagesPayload.messages) ? messagesPayload.messages : [];
+      this.applyStatusPayload(statusPayload);
+      this.session?.replaceRemoteMessages(messages);
+      const commands = Array.isArray(commandsPayload.commands) ? commandsPayload.commands : [];
+      this.hooks.onCatchUp?.(messages, commands);
+      catchupDone = true;
+      for (const payload of buffered.splice(0))
+        this.applyEnvelope(payload);
+    } catch (error) {
+      catchupDone = true;
+      this.hooks.onError?.(`Worker Pi catch-up failed: ${error instanceof Error ? error.message : String(error)}`);
+    }
+  }
+  close() {
+    this.closed = true;
+    this.socket?.close();
+    for (const shell of this.pendingShells.splice(0))
+      shell.reject(new Error("Remote session closed."));
+    for (const compaction of this.pendingCompactions.splice(0))
+      compaction.reject(new Error("Remote session closed."));
+  }
+  async sendPrompt(text, streamingBehavior) {
+    await this.transport.sendPrompt(this.context, this.runId, text, streamingBehavior);
+  }
+  async sendCommand(text) {
+    const result = await this.transport.runCommand(this.context, this.runId, text);
+    return typeof result.message === "string" ? result.message : "worker command accepted";
+  }
+  async sendShell(text) {
+    await this.transport.sendShell(this.context, this.runId, text);
+  }
+  async abort() {
+    await this.transport.abort(this.context, this.runId);
+  }
+  registerPendingShell(shell) {
+    this.pendingShells.push(shell);
+  }
+  failPendingShell(shell, error) {
+    const index = this.pendingShells.indexOf(shell);
+    if (index !== -1)
+      this.pendingShells.splice(index, 1);
+    shell.reject(error);
+  }
+  registerPendingCompaction(pending) {
+    this.pendingCompactions.push(pending);
+  }
+  async waitForReady() {
+    const startedAt = Date.now();
+    const deadline = startedAt + resolveAttachReadyTimeoutMs();
+    let consecutiveFailures = 0;
+    while (!this.closed) {
+      let requestFailed = false;
+      const session = await this.transport.getSession(this.context, this.runId).catch((error) => {
+        requestFailed = true;
+        return { ready: false, status: error instanceof Error ? error.message : String(error), retryAfterMs: 1000 };
+      });
+      if (session.ready !== false)
+        return true;
+      consecutiveFailures = requestFailed ? consecutiveFailures + 1 : 0;
+      const status = String(session.status ?? "starting");
+      if (TERMINAL_RUN_STATUSES.has(status.toLowerCase())) {
+        this.hooks.onError?.(`Run ended before the worker Pi daemon became ready: ${status}. Inspect with \`rig run show ${this.runId}\`.`);
+        return false;
+      }
+      this.hooks.onStatusText?.(consecutiveFailures >= 5 ? "Rig server unreachable \xB7 retrying \xB7 /detach to exit" : `waiting for worker Pi daemon \xB7 ${status} \xB7 /detach to exit`);
+      if (Date.now() >= deadline) {
+        this.hooks.onError?.(`Worker Pi daemon did not become ready (last status: ${status}). Set RIG_PI_ATTACH_TIMEOUT_MS to wait longer.`);
+        return false;
+      }
+      await Bun.sleep(typeof session.retryAfterMs === "number" ? session.retryAfterMs : 750);
+    }
+    return false;
+  }
+  applyEnvelope(envelopeValue) {
+    const envelope = recordOf(envelopeValue);
+    if (!envelope)
+      return;
+    const type = String(envelope.type ?? "");
+    if (type === "status.update") {
+      this.applyStatusPayload(envelope);
+      return;
+    }
+    if (type === "activity.update") {
+      const activity = recordOf(envelope.activity);
+      this.hooks.onActivity?.(String(activity?.label ?? ""), activity?.detail ? String(activity.detail) : undefined);
+      return;
+    }
+    if (type === "extension_ui_request") {
+      const request = recordOf(envelope.request);
+      if (request)
+        this.hooks.onExtensionUiRequest?.(request);
+      return;
+    }
+    if (type === "pi.ui_event") {
+      this.applyShellUiEvent(envelope.event);
+      return;
+    }
+    if (type === "pi.event") {
+      this.session?.handleRemoteSessionEvent(envelope.event);
+      this.settlePendingCompaction(envelope.event);
+      return;
+    }
+    if (type === "error") {
+      this.hooks.onError?.(String(envelope.message ?? envelope.detail ?? "unknown worker error"));
+    }
+  }
+  applyStatusPayload(payload) {
+    const status = recordOf(payload.status) ?? payload;
+    const next = this.status;
+    next.isStreaming = status.isStreaming === true;
+    next.isCompacting = status.isCompacting === true;
+    next.isBashRunning = status.isBashRunning === true;
+    next.pendingMessageCount = typeof status.pendingMessageCount === "number" ? status.pendingMessageCount : 0;
+    next.steeringMessages = Array.isArray(status.steeringMessages) ? status.steeringMessages.map(String) : [];
+    next.followUpMessages = Array.isArray(status.followUpMessages) ? status.followUpMessages.map(String) : [];
+    next.model = recordOf(status.model) ?? next.model;
+    next.thinkingLevel = typeof status.thinkingLevel === "string" ? status.thinkingLevel : next.thinkingLevel;
+    next.sessionName = typeof status.sessionName === "string" ? status.sessionName : next.sessionName;
+    next.cwd = typeof status.cwd === "string" ? status.cwd : next.cwd;
+    next.stats = recordOf(status.stats) ?? next.stats;
+    next.contextUsage = recordOf(status.contextUsage) ?? next.contextUsage;
+  }
+  applyShellUiEvent(value) {
+    const event = recordOf(value);
+    if (!event)
+      return;
+    const type = String(event.type ?? "");
+    const pending = this.pendingShells[0];
+    if (type === "shell.chunk" && pending) {
+      pending.sawChunk = true;
+      pending.onData(Buffer.from(String(event.chunk ?? "")));
+      return;
+    }
+    if (type === "shell.end" && pending) {
+      const output = String(event.output ?? "");
+      if (output && !pending.sawChunk)
+        pending.onData(Buffer.from(output));
+      const index = this.pendingShells.indexOf(pending);
+      if (index !== -1)
+        this.pendingShells.splice(index, 1);
+      pending.resolve({ exitCode: typeof event.exitCode === "number" ? event.exitCode : event.isError === true ? 1 : 0 });
+    }
+  }
+  settlePendingCompaction(eventValue) {
+    const event = recordOf(eventValue);
+    if (!event)
+      return;
+    const type = String(event.type ?? "");
+    if (type !== "compaction_end" || this.pendingCompactions.length === 0)
+      return;
+    const pending = this.pendingCompactions.shift();
+    const result = recordOf(event.result);
+    if (result)
+      pending.resolve(result);
+    else if (event.aborted === true)
+      pending.reject(new Error("Compaction aborted on the worker."));
     else
-      for (const message of messages)
-        applyMessage(state, message);
-    applyStatus(state, statusPayload);
-    const commands = Array.isArray(commandsPayload.commands) ? commandsPayload.commands : [];
-    state.commands = commands.flatMap((command) => {
-      const record = recordOf(command);
-      return typeof record?.name === "string" ? [`/${record.name}`] : [];
-    });
-    registerDaemonCommandsNatively(pi, options, ctx, state, commands, registeredDaemonCommands);
-    catchupDone = true;
-    for (const payload of buffered.splice(0))
-      applyEnvelope(ctx, state, payload);
-    updatePiUi(ctx, state);
-  } catch (error) {
-    appendTranscript(state, "Error", `Worker Pi catch-up failed: ${error instanceof Error ? error.message : String(error)}`);
-    catchupDone = true;
-    updatePiUi(ctx, state);
+      pending.resolve({});
   }
-  await closePromise;
 }
-function createRemoteBashOperations(options, state, excludeFromContext) {
+function createRemoteBashOperations(controller, excludeFromContext) {
   return {
     exec(command, _cwd, execOptions) {
       return new Promise((resolve3, reject) => {
-        const pending = {
-          command,
-          onData: execOptions.onData,
-          resolve: resolve3,
-          reject,
-          sawChunk: false
-        };
+        const pending = { onData: execOptions.onData, resolve: resolve3, reject, sawChunk: false };
         const cleanup = () => {
           execOptions.signal?.removeEventListener("abort", onAbort);
           if (timer)
@@ -682,12 +521,12 @@ function createRemoteBashOperations(options, state, excludeFromContext) {
         };
         const onAbort = () => {
           cleanup();
-          failPendingShell(state, pending, new Error("Remote worker shell command aborted locally."));
+          controller.failPendingShell(pending, new Error("Remote worker shell command aborted locally."));
         };
         const timeoutMs = typeof execOptions.timeout === "number" && execOptions.timeout > 0 ? execOptions.timeout : 0;
         const timer = timeoutMs > 0 ? setTimeout(() => {
           cleanup();
-          failPendingShell(state, pending, new Error(`Remote worker shell command timed out after ${timeoutMs}ms.`));
+          controller.failPendingShell(pending, new Error(`Remote worker shell command timed out after ${timeoutMs}ms.`));
         }, timeoutMs) : null;
         const wrappedResolve = pending.resolve;
         const wrappedReject = pending.reject;
@@ -700,125 +539,136 @@ function createRemoteBashOperations(options, state, excludeFromContext) {
           wrappedReject(error);
         };
         execOptions.signal?.addEventListener("abort", onAbort, { once: true });
-        state.pendingShells.push(pending);
-        sendRunPiShellViaServer(options.context, options.runId, `${excludeFromContext ? "!!" : "!"}${command}`).catch((error) => {
+        controller.registerPendingShell(pending);
+        controller.sendShell(`${excludeFromContext ? "!!" : "!"}${command}`).catch((error) => {
           cleanup();
-          failPendingShell(state, pending, error instanceof Error ? error : new Error(String(error)));
+          controller.failPendingShell(pending, error instanceof Error ? error : new Error(String(error)));
         });
       });
     }
   };
 }
-async function answerPendingUi(options, state, line) {
-  const pending = state.pendingUi;
-  if (!pending)
-    return false;
-  if (line === "/cancel") {
-    await respondRunPiExtensionUiViaServer(options.context, options.runId, pending.requestId, { cancelled: true });
-  } else if (pending.method === "confirm") {
-    const confirmed = /^(y|yes|true|1)$/i.test(line);
-    await respondRunPiExtensionUiViaServer(options.context, options.runId, pending.requestId, { value: confirmed, confirmed });
-  } else if (pending.options.length > 0 && /^\d+$/.test(line)) {
-    const selected = pending.options[Math.max(0, Number(line) - 1)] ?? line;
-    await respondRunPiExtensionUiViaServer(options.context, options.runId, pending.requestId, { value: selected });
-  } else {
-    await respondRunPiExtensionUiViaServer(options.context, options.runId, pending.requestId, { value: line });
-  }
-  appendTranscript(state, "System", `Responded to extension UI request ${pending.requestId}.`);
-  state.pendingUi = null;
-  return true;
-}
-async function routeInput(options, ctx, state, line) {
-  const text = line.trim();
-  if (!text)
-    return;
-  if (await answerPendingUi(options, state, text)) {
-    updatePiUi(ctx, state);
-    return;
-  }
-  if (text === "/detach" || text === "/quit" || text === "/q") {
-    appendTranscript(state, "System", "Detached locally; worker Pi daemon continues.");
-    updatePiUi(ctx, state);
-    ctx.shutdown();
-    return;
+
+// packages/cli/src/commands/_pi-worker-bridge-extension.ts
+function recordOf2(value) {
+  return value && typeof value === "object" && !Array.isArray(value) ? value : null;
+}
+function asText(value) {
+  if (typeof value === "string")
+    return value;
+  if (value === null || value === undefined)
+    return "";
+  if (typeof value === "number" || typeof value === "boolean")
+    return String(value);
+  try {
+    return JSON.stringify(value);
+  } catch {
+    return String(value);
   }
-  if (text === "/stop") {
-    await abortRunPiViaServer(options.context, options.runId);
-    appendTranscript(state, "System", "Stop requested for worker Pi daemon.");
-    updatePiUi(ctx, state);
-    ctx.shutdown();
-    return;
+}
+async function answerExtensionUiRequest(options, ctx, request) {
+  const requestId = String(request.requestId ?? request.id ?? `ui-${Date.now()}`);
+  const method = String(request.method ?? request.type ?? "input");
+  const prompt = asText(request.prompt ?? request.message ?? request.title ?? method);
+  const rawOptions = Array.isArray(request.options) ? request.options : Array.isArray(request.items) ? request.items : [];
+  const choices = rawOptions.map((option) => {
+    const record = recordOf2(option);
+    return record ? asText(record.label ?? record.value ?? record.name ?? option) : asText(option);
+  }).filter(Boolean);
+  try {
+    if (method === "confirm") {
+      const confirmed = await ctx.ui.confirm("Worker request", prompt);
+      await respondRunPiExtensionUiViaServer(options.context, options.runId, requestId, { value: confirmed, confirmed });
+      return;
+    }
+    if (choices.length > 0) {
+      const selected = await ctx.ui.select(prompt, choices);
+      await respondRunPiExtensionUiViaServer(options.context, options.runId, requestId, selected === undefined ? { cancelled: true } : { value: selected });
+      return;
+    }
+    const value = await ctx.ui.input("Worker request", prompt);
+    await respondRunPiExtensionUiViaServer(options.context, options.runId, requestId, value === undefined ? { cancelled: true } : { value });
+  } catch (error) {
+    ctx.ui.notify(`Failed to answer worker UI request: ${error instanceof Error ? error.message : String(error)}`, "error");
   }
-  if (text.startsWith("!")) {
-    appendTranscript(state, "You", text);
-    await sendRunPiShellViaServer(options.context, options.runId, text);
-  } else if (text.startsWith("/")) {
-    appendTranscript(state, "You", text);
-    const result = await runRunPiCommandViaServer(options.context, options.runId, text);
-    const message = typeof result.message === "string" ? result.message : "worker command accepted";
-    appendTranscript(state, "System", message);
-  } else {
-    appendTranscript(state, "You", text);
-    await sendRunPiPromptViaServer(options.context, options.runId, text, state.streaming ? "steer" : undefined);
+}
+var LOCALLY_OWNED_COMMANDS = new Set(["detach", "quit", "q", "stop", "settings", "model", "thinking", "compact", "session", "name", "reload"]);
+function registerDaemonCommands(pi, options, ctx, commands, registered) {
+  for (const command of commands) {
+    const record = recordOf2(command);
+    const name = typeof record?.name === "string" ? record.name : "";
+    const source = typeof record?.source === "string" ? record.source : "worker";
+    if (!name || source === "builtin" || registered.has(name) || LOCALLY_OWNED_COMMANDS.has(name))
+      continue;
+    registered.add(name);
+    const description = typeof record?.description === "string" ? record.description : undefined;
+    try {
+      pi.registerCommand(name, {
+        description: `[worker ${source}] ${description ?? ""}`.trim(),
+        handler: async (args) => {
+          try {
+            const message = await options.controller.sendCommand(`/${name}${args ? ` ${args}` : ""}`);
+            ctx.ui.notify(message, "info");
+          } catch (error) {
+            ctx.ui.notify(error instanceof Error ? error.message : String(error), "error");
+          }
+        }
+      });
+    } catch {}
   }
-  updatePiUi(ctx, state);
 }
 function createRigWorkerPiBridgeExtension(options) {
   return (pi) => {
-    const state = {
-      transcript: [],
-      status: "starting worker Pi daemon bridge",
-      activity: "",
-      cwd: "",
-      model: "",
-      commands: [],
-      streaming: false,
-      pendingUi: null,
-      pendingShells: [],
-      wsConnected: false,
-      nativeStream: false
-    };
-    if (options.initialMessageSent)
-      appendTranscript(state, "System", "Initial message sent to worker Pi daemon.");
     const registeredDaemonCommands = new Set;
-    let nativePiUiContextAvailable = false;
-    pi.on("user_bash", (event) => {
-      state.nativeStream = Boolean(state.nativeStream || nativePiUiContextAvailable);
-      return { operations: createRemoteBashOperations(options, state, event.excludeFromContext === true) };
+    pi.registerCommand("detach", {
+      description: "Detach from this run; the worker keeps going",
+      handler: async (_args, ctx) => {
+        ctx.ui.notify("Detached locally; the worker Pi daemon continues.", "info");
+        ctx.shutdown();
+      }
+    });
+    pi.registerCommand("stop", {
+      description: "Stop the worker Pi run and detach",
+      handler: async (_args, ctx) => {
+        await options.controller.abort();
+        ctx.ui.notify("Stop requested for the worker Pi daemon.", "info");
+        ctx.shutdown();
+      }
     });
+    pi.on("user_bash", (event) => ({
+      operations: createRemoteBashOperations(options.controller, event.excludeFromContext === true)
+    }));
     pi.on("session_start", async (_event, ctx) => {
-      nativePiUiContextAvailable = Boolean(nativePiUi(ctx));
-      state.nativeStream = nativePiUiContextAvailable;
-      updatePiUi(ctx, state);
-      ctx.ui.notify(nativePiUiContextAvailable ? "Connected to the Rig worker session (native stream). /detach exits, /stop cancels the run." : "Connected to the Rig worker session (transcript view). /detach exits, /stop cancels the run.", "info");
-      ctx.ui.onTerminalInput((data) => {
-        if (data.includes("\x04")) {
-          ctx.shutdown();
-          return { consume: true };
+      ctx.ui.setTitle("Rig \xB7 worker session");
+      ctx.ui.setStatus("rig-worker-pi", "connecting to worker session");
+      ctx.ui.notify(`Attached to Rig run ${options.runId}. Native Pi, remote brain \u2014 /detach exits, /stop cancels the run.`, "info");
+      if (options.initialMessageSent)
+        ctx.ui.notify("Initial message sent to the worker Pi daemon.", "info");
+      const nativeUi = ctx.ui;
+      options.controller.setUiHooks({
+        onStatusText: (text) => ctx.ui.setStatus("rig-worker-pi", text),
+        onActivity: (label, detail) => ctx.ui.setStatus("rig-worker-pi", [label, detail].filter(Boolean).join(" \u2014 ")),
+        onConnectionChange: (connected) => ctx.ui.setStatus("rig-worker-pi", connected ? "worker session live" : "worker session disconnected"),
+        onError: (message) => ctx.ui.notify(message, "error"),
+        onExtensionUiRequest: (request) => {
+          answerExtensionUiRequest(options, ctx, request);
+        },
+        onCatchUp: (messages, commands) => {
+          if (nativeUi.appendSessionMessages)
+            nativeUi.appendSessionMessages(messages);
+          const cwd = options.controller.status.cwd;
+          if (nativeUi.setDisplayCwd && cwd)
+            nativeUi.setDisplayCwd(cwd);
+          registerDaemonCommands(pi, options, ctx, commands, registeredDaemonCommands);
         }
-        if (!data.includes("\r") && !data.includes(`
-`))
-          return;
-        const inlineText = data.replace(/[\r\n]+/g, "").trim();
-        const editorText = ctx.ui.getEditorText().trim();
-        const text = [editorText, inlineText].filter(Boolean).join(" ").trim();
-        if (!text)
-          return;
-        if (text.startsWith("!"))
-          return;
-        ctx.ui.setEditorText("");
-        routeInput(options, ctx, state, text).catch((error) => {
-          appendTranscript(state, "Error", error instanceof Error ? error.message : String(error));
-          updatePiUi(ctx, state);
-        });
-        return { consume: true };
       });
-      connectWorkerStream(options, pi, ctx, state, registeredDaemonCommands).catch((error) => {
-        appendTranscript(state, "Error", error instanceof Error ? error.message : String(error));
-        updatePiUi(ctx, state);
+      options.controller.connect().catch((error) => {
+        ctx.ui.notify(error instanceof Error ? error.message : String(error), "error");
       });
     });
-    pi.on("session_shutdown", () => {});
+    pi.on("session_shutdown", () => {
+      options.controller.close();
+    });
   };
 }
 export {