@h-rig/cli 0.0.6-alpha.63 → 0.0.6-alpha.65

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

@@ -1,834 +0,0 @@
-// @bun
-// packages/cli/src/commands/_server-client.ts
-import { existsSync as existsSync2, readFileSync as readFileSync2 } from "fs";
-import { resolve as resolve2 } from "path";
-
-// packages/cli/src/runner.ts
-import { EventBus } from "@rig/runtime/control-plane/runtime/events";
-import { CliError } from "@rig/runtime/control-plane/errors";
-import { evaluate, loadPolicy, resolveAction } from "@rig/runtime/control-plane/runtime/guard";
-import { buildBinary } from "@rig/runtime/control-plane/runtime/isolation";
-import { CliError as CliError2 } from "@rig/runtime/control-plane/errors";
-
-// packages/cli/src/commands/_server-client.ts
-import { ensureLocalRigServerConnection } from "@rig/runtime/local-server";
-
-// packages/cli/src/commands/_connection-state.ts
-import { existsSync, mkdirSync, readFileSync, writeFileSync } from "fs";
-import { homedir } from "os";
-import { dirname, resolve } from "path";
-function resolveGlobalConnectionsPath(env = process.env) {
-  const explicit = env.RIG_CONNECTIONS_FILE?.trim();
-  if (explicit)
-    return resolve(explicit);
-  const stateDir = env.RIG_GLOBAL_STATE_DIR?.trim();
-  if (stateDir)
-    return resolve(stateDir, "connections.json");
-  return resolve(homedir(), ".rig", "connections.json");
-}
-function resolveRepoConnectionPath(projectRoot) {
-  return resolve(projectRoot, ".rig", "state", "connection.json");
-}
-function readJsonFile(path) {
-  if (!existsSync(path))
-    return null;
-  try {
-    return JSON.parse(readFileSync(path, "utf8"));
-  } catch (error) {
-    throw new CliError2(`Invalid Rig connection state at ${path}: ${error instanceof Error ? error.message : String(error)}`, 1);
-  }
-}
-function writeJsonFile(path, value) {
-  mkdirSync(dirname(path), { recursive: true });
-  writeFileSync(path, `${JSON.stringify(value, null, 2)}
-`, "utf8");
-}
-function normalizeConnection(value) {
-  if (!value || typeof value !== "object" || Array.isArray(value))
-    return null;
-  const record = value;
-  if (record.kind === "local")
-    return { kind: "local", mode: "auto" };
-  if (record.kind === "remote" && typeof record.baseUrl === "string" && record.baseUrl.trim()) {
-    const baseUrl = record.baseUrl.trim().replace(/\/+$/, "");
-    return { kind: "remote", baseUrl };
-  }
-  return null;
-}
-function readGlobalConnections(options = {}) {
-  const path = resolveGlobalConnectionsPath(options.env ?? process.env);
-  const payload = readJsonFile(path);
-  if (!payload || typeof payload !== "object" || Array.isArray(payload)) {
-    return { connections: {} };
-  }
-  const rawConnections = payload.connections;
-  const connections = {};
-  if (rawConnections && typeof rawConnections === "object" && !Array.isArray(rawConnections)) {
-    for (const [alias, raw] of Object.entries(rawConnections)) {
-      const connection = normalizeConnection(raw);
-      if (connection)
-        connections[alias] = connection;
-    }
-  }
-  return { connections };
-}
-function readRepoConnection(projectRoot) {
-  const payload = readJsonFile(resolveRepoConnectionPath(projectRoot));
-  if (!payload || typeof payload !== "object" || Array.isArray(payload))
-    return null;
-  const record = payload;
-  const selected = typeof record.selected === "string" ? record.selected.trim() : "";
-  if (!selected)
-    return null;
-  return {
-    selected,
-    project: typeof record.project === "string" ? record.project : undefined,
-    linkedAt: typeof record.linkedAt === "string" ? record.linkedAt : undefined,
-    serverProjectRoot: typeof record.serverProjectRoot === "string" && record.serverProjectRoot.trim() ? record.serverProjectRoot.trim() : undefined
-  };
-}
-function writeRepoConnection(projectRoot, state) {
-  writeJsonFile(resolveRepoConnectionPath(projectRoot), state);
-}
-function resolveSelectedConnection(projectRoot, options = {}) {
-  const repo = readRepoConnection(projectRoot);
-  if (!repo)
-    return null;
-  if (repo.selected === "local")
-    return { alias: "local", connection: { kind: "local", mode: "auto" }, serverProjectRoot: repo.serverProjectRoot };
-  const global = readGlobalConnections(options);
-  const connection = global.connections[repo.selected];
-  if (!connection) {
-    throw new CliError2(`Selected Rig server "${repo.selected}" was not found. Run \`rig server list\` or \`rig server use local\`.`, 1);
-  }
-  return { alias: repo.selected, connection, serverProjectRoot: repo.serverProjectRoot };
-}
-function writeRepoServerProjectRoot(projectRoot, serverProjectRoot) {
-  const repo = readRepoConnection(projectRoot);
-  if (!repo)
-    return;
-  writeRepoConnection(projectRoot, { ...repo, serverProjectRoot });
-}
-
-// packages/cli/src/commands/_server-client.ts
-var scopedGitHubBearerTokens = new Map;
-function cleanToken(value) {
-  const trimmed = value?.trim();
-  return trimmed ? trimmed : null;
-}
-function readPrivateRemoteSessionToken(projectRoot) {
-  const path = resolve2(projectRoot, ".rig", "state", "github-auth.json");
-  if (!existsSync2(path))
-    return null;
-  try {
-    const parsed = JSON.parse(readFileSync2(path, "utf8"));
-    return cleanToken(typeof parsed.apiSessionToken === "string" ? parsed.apiSessionToken : typeof parsed.sessionToken === "string" ? parsed.sessionToken : undefined);
-  } catch {
-    return null;
-  }
-}
-function readGitHubBearerTokenForRemote(projectRoot) {
-  const scopedKey = resolve2(projectRoot);
-  if (scopedGitHubBearerTokens.has(scopedKey))
-    return scopedGitHubBearerTokens.get(scopedKey) ?? null;
-  const privateSession = readPrivateRemoteSessionToken(projectRoot);
-  if (privateSession)
-    return privateSession;
-  return cleanToken(process.env.RIG_SERVER_AUTH_TOKEN) ?? cleanToken(process.env.RIG_REMOTE_AUTH_TOKEN);
-}
-function readStoredGitHubAuthToken(projectRoot) {
-  const path = resolve2(projectRoot, ".rig", "state", "github-auth.json");
-  if (!existsSync2(path))
-    return null;
-  try {
-    const parsed = JSON.parse(readFileSync2(path, "utf8"));
-    return cleanToken(typeof parsed.token === "string" ? parsed.token : undefined);
-  } catch {
-    return null;
-  }
-}
-function readLocalConnectionFallbackToken(projectRoot) {
-  return readGitHubBearerTokenForRemote(projectRoot) ?? cleanToken(process.env.RIG_GITHUB_TOKEN) ?? readStoredGitHubAuthToken(projectRoot);
-}
-async function ensureServerForCli(projectRoot) {
-  try {
-    const selected = resolveSelectedConnection(projectRoot);
-    if (selected?.connection.kind === "remote") {
-      const authToken = readGitHubBearerTokenForRemote(projectRoot);
-      const serverProjectRoot = selected.serverProjectRoot ?? await backfillRemoteServerProjectRoot(projectRoot, selected.connection.baseUrl, authToken);
-      return {
-        baseUrl: selected.connection.baseUrl,
-        authToken,
-        connectionKind: "remote",
-        serverProjectRoot
-      };
-    }
-    const connection = await ensureLocalRigServerConnection(projectRoot);
-    return {
-      baseUrl: connection.baseUrl,
-      authToken: connection.authToken ?? readLocalConnectionFallbackToken(projectRoot),
-      connectionKind: "local",
-      serverProjectRoot: resolve2(projectRoot)
-    };
-  } catch (error) {
-    if (error instanceof Error) {
-      throw new CliError2(error.message, 1);
-    }
-    throw error;
-  }
-}
-async function backfillRemoteServerProjectRoot(projectRoot, baseUrl, authToken) {
-  const repo = readRepoConnection(projectRoot);
-  const slug = repo?.project?.trim();
-  if (!slug)
-    return null;
-  try {
-    const response = await fetch(`${baseUrl}/api/projects/${encodeURIComponent(slug)}`, {
-      headers: mergeHeaders(undefined, authToken)
-    });
-    if (!response.ok)
-      return null;
-    const payload = await response.json();
-    const project = payload.project && typeof payload.project === "object" && !Array.isArray(payload.project) ? payload.project : null;
-    const checkouts = Array.isArray(project?.checkouts) ? project.checkouts : [];
-    const latestCheckout = [...checkouts].reverse().find((entry) => Boolean(entry && typeof entry === "object" && !Array.isArray(entry) && typeof entry.path === "string"));
-    const path = typeof latestCheckout?.path === "string" && latestCheckout.path.trim() ? latestCheckout.path.trim() : null;
-    if (path)
-      writeRepoServerProjectRoot(projectRoot, path);
-    return path;
-  } catch {
-    return null;
-  }
-}
-function mergeHeaders(headers, authToken) {
-  const merged = new Headers(headers);
-  if (authToken) {
-    merged.set("authorization", `Bearer ${authToken}`);
-  }
-  return merged;
-}
-function diagnosticMessage(payload) {
-  if (!payload || typeof payload !== "object" || Array.isArray(payload))
-    return null;
-  const record = payload;
-  const diagnostics = Array.isArray(record.diagnostics) ? record.diagnostics : [];
-  const messages = diagnostics.flatMap((entry) => {
-    if (!entry || typeof entry !== "object" || Array.isArray(entry))
-      return [];
-    const diagnostic = entry;
-    const kind = typeof diagnostic.kind === "string" ? diagnostic.kind : "task-source";
-    const message = typeof diagnostic.message === "string" ? diagnostic.message : null;
-    return message ? [`${kind}: ${message}`] : [];
-  });
-  return messages.length > 0 ? messages.join("; ") : null;
-}
-async function requestServerJson(context, pathname, init = {}) {
-  const server = await ensureServerForCli(context.projectRoot);
-  const headers = mergeHeaders(init.headers, server.authToken);
-  if (server.serverProjectRoot)
-    headers.set("x-rig-project-root", server.serverProjectRoot);
-  const response = await fetch(`${server.baseUrl}${pathname}`, {
-    ...init,
-    headers
-  });
-  const text = await response.text();
-  const payload = text.trim().length > 0 ? (() => {
-    try {
-      return JSON.parse(text);
-    } catch {
-      return null;
-    }
-  })() : null;
-  if (!response.ok) {
-    const diagnostics = diagnosticMessage(payload);
-    const detail = diagnostics ?? (text || response.statusText);
-    throw new CliError2(`Rig server request failed (${response.status}): ${detail}`, 1);
-  }
-  return payload;
-}
-async function getRunPiSessionViaServer(context, runId) {
-  const payload = await requestServerJson(context, `/api/runs/${encodeURIComponent(runId)}/pi`);
-  return payload && typeof payload === "object" && !Array.isArray(payload) ? payload : {};
-}
-async function getRunPiMessagesViaServer(context, runId) {
-  const payload = await requestServerJson(context, `/api/runs/${encodeURIComponent(runId)}/pi/messages`);
-  return payload && typeof payload === "object" && !Array.isArray(payload) ? payload : { messages: [] };
-}
-async function getRunPiStatusViaServer(context, runId) {
-  const payload = await requestServerJson(context, `/api/runs/${encodeURIComponent(runId)}/pi/status`);
-  return payload && typeof payload === "object" && !Array.isArray(payload) ? payload : {};
-}
-async function getRunPiCommandsViaServer(context, runId) {
-  const payload = await requestServerJson(context, `/api/runs/${encodeURIComponent(runId)}/pi/commands`);
-  return payload && typeof payload === "object" && !Array.isArray(payload) ? payload : { commands: [] };
-}
-async function getRunPiCapabilitiesViaServer(context, runId) {
-  const payload = await requestServerJson(context, `/api/runs/${encodeURIComponent(runId)}/pi/capabilities`);
-  return payload && typeof payload === "object" && !Array.isArray(payload) ? payload : { capabilities: null };
-}
-async function sendRunPiPromptViaServer(context, runId, text, streamingBehavior) {
-  const payload = await requestServerJson(context, `/api/runs/${encodeURIComponent(runId)}/pi/prompt`, {
-    method: "POST",
-    headers: { "content-type": "application/json" },
-    body: JSON.stringify({ text, streamingBehavior })
-  });
-  return payload && typeof payload === "object" && !Array.isArray(payload) ? payload : { accepted: true };
-}
-async function sendRunPiShellViaServer(context, runId, text) {
-  const payload = await requestServerJson(context, `/api/runs/${encodeURIComponent(runId)}/pi/shell`, {
-    method: "POST",
-    headers: { "content-type": "application/json" },
-    body: JSON.stringify({ text })
-  });
-  return payload && typeof payload === "object" && !Array.isArray(payload) ? payload : { accepted: true };
-}
-async function runRunPiCommandViaServer(context, runId, text) {
-  const payload = await requestServerJson(context, `/api/runs/${encodeURIComponent(runId)}/pi/commands/run`, {
-    method: "POST",
-    headers: { "content-type": "application/json" },
-    body: JSON.stringify({ text })
-  });
-  return payload && typeof payload === "object" && !Array.isArray(payload) ? payload : { type: "done" };
-}
-async function respondRunPiExtensionUiViaServer(context, runId, requestId, valueOrCancel) {
-  const payload = await requestServerJson(context, `/api/runs/${encodeURIComponent(runId)}/pi/extension-ui/respond`, {
-    method: "POST",
-    headers: { "content-type": "application/json" },
-    body: JSON.stringify({ requestId, ...valueOrCancel })
-  });
-  return payload && typeof payload === "object" && !Array.isArray(payload) ? payload : { accepted: true };
-}
-async function abortRunPiViaServer(context, runId) {
-  const payload = await requestServerJson(context, `/api/runs/${encodeURIComponent(runId)}/pi/abort`, { method: "POST" });
-  return payload && typeof payload === "object" && !Array.isArray(payload) ? payload : { aborted: true };
-}
-async function buildRunPiEventsWebSocketUrl(context, runId) {
-  const server = await ensureServerForCli(context.projectRoot);
-  const url = new URL(`${server.baseUrl.replace(/\/+$/, "")}/api/runs/${encodeURIComponent(runId)}/pi/events`);
-  url.protocol = url.protocol === "https:" ? "wss:" : "ws:";
-  if (server.authToken)
-    url.searchParams.set("token", server.authToken);
-  if (server.serverProjectRoot)
-    url.searchParams.set("rigProjectRoot", server.serverProjectRoot);
-  return url.toString();
-}
-
-// packages/cli/src/commands/_pi-remote-session.ts
-import { AgentSession as PiAgentSession, expandPromptTemplate } from "@earendil-works/pi-coding-agent";
-var TERMINAL_RUN_STATUSES = new Set(["completed", "failed", "stopped", "cancelled", "canceled", "closed", "merged", "needs_attention", "needs-attention"]);
-function defaultTransport() {
-  return {
-    getSession: getRunPiSessionViaServer,
-    getMessages: getRunPiMessagesViaServer,
-    getStatus: getRunPiStatusViaServer,
-    getCommands: getRunPiCommandsViaServer,
-    getCapabilities: getRunPiCapabilitiesViaServer,
-    sendPrompt: sendRunPiPromptViaServer,
-    sendShell: sendRunPiShellViaServer,
-    runCommand: runRunPiCommandViaServer,
-    abort: abortRunPiViaServer,
-    buildEventsWebSocketUrl: buildRunPiEventsWebSocketUrl
-  };
-}
-function recordOf(value) {
-  return value && typeof value === "object" && !Array.isArray(value) ? value : null;
-}
-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;
-}
-
-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 = () => {
-      this.hooks.onConnectionChange?.(true);
-      this.hooks.onStatusText?.("worker session live");
-    };
-    socket.onmessage = (message) => {
-      try {
-        const payload = typeof message.data === "string" ? JSON.parse(message.data) : JSON.parse(Buffer.from(message.data).toString("utf8"));
-        if (!catchupDone)
-          buffered.push(payload);
-        else
-          this.applyEnvelope(payload);
-      } catch (error) {
-        this.hooks.onError?.(`Unparseable worker Pi event: ${error instanceof Error ? error.message : String(error)}`);
-      }
-    };
-    socket.onerror = () => socket.close();
-    socket.onclose = () => {
-      this.hooks.onConnectionChange?.(false);
-      if (!this.closed)
-        this.hooks.onStatusText?.("worker Pi WebSocket disconnected");
-    };
-    try {
-      const [messagesPayload, statusPayload, commandsPayload, capabilitiesPayload] = await Promise.all([
-        this.transport.getMessages(this.context, this.runId),
-        this.transport.getStatus(this.context, this.runId),
-        this.transport.getCommands(this.context, this.runId),
-        this.transport.getCapabilities(this.context, this.runId).catch(() => ({ capabilities: null }))
-      ]);
-      const messages = Array.isArray(messagesPayload.messages) ? messagesPayload.messages : [];
-      this.applyStatusPayload(statusPayload);
-      this.session?.replaceRemoteMessages(messages);
-      const commands = Array.isArray(commandsPayload.commands) ? commandsPayload.commands : [];
-      const capabilities = capabilitiesPayload.capabilities && typeof capabilitiesPayload.capabilities === "object" && !Array.isArray(capabilitiesPayload.capabilities) ? capabilitiesPayload.capabilities : null;
-      this.hooks.onCatchUp?.(messages, commands, capabilities);
-      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
-      pending.resolve({});
-  }
-}
-function createRemoteBashOperations(controller, excludeFromContext) {
-  return {
-    exec(command, _cwd, execOptions) {
-      return new Promise((resolve3, reject) => {
-        const pending = { onData: execOptions.onData, resolve: resolve3, reject, sawChunk: false };
-        const cleanup = () => {
-          execOptions.signal?.removeEventListener("abort", onAbort);
-          if (timer)
-            clearTimeout(timer);
-        };
-        const onAbort = () => {
-          cleanup();
-          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();
-          controller.failPendingShell(pending, new Error(`Remote worker shell command timed out after ${timeoutMs}ms.`));
-        }, timeoutMs) : null;
-        const wrappedResolve = pending.resolve;
-        const wrappedReject = pending.reject;
-        pending.resolve = (result) => {
-          cleanup();
-          wrappedResolve(result);
-        };
-        pending.reject = (error) => {
-          cleanup();
-          wrappedReject(error);
-        };
-        execOptions.signal?.addEventListener("abort", onAbort, { once: true });
-        controller.registerPendingShell(pending);
-        controller.sendShell(`${excludeFromContext ? "!!" : "!"}${command}`).catch((error) => {
-          cleanup();
-          controller.failPendingShell(pending, error instanceof Error ? error : new Error(String(error)));
-        });
-      });
-    }
-  };
-}
-
-// packages/cli/src/commands/_spinner.ts
-var SPINNER_FRAMES = ["\u280B", "\u2819", "\u2839", "\u2838", "\u283C", "\u2834", "\u2826", "\u2827", "\u2807", "\u280F"];
-
-// 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);
-  }
-}
-function names(value, key = "name") {
-  if (!Array.isArray(value))
-    return [];
-  return value.flatMap((entry) => {
-    if (typeof entry === "string")
-      return [entry];
-    const record = recordOf2(entry);
-    const name = record?.[key];
-    return typeof name === "string" ? [name] : [];
-  });
-}
-function renderWorkerCapabilities(capabilities) {
-  const lines = ["Worker session capabilities (in effect for this run)"];
-  const tools = Array.isArray(capabilities.tools) ? capabilities.tools : [];
-  const activeTools = tools.flatMap((tool) => {
-    const record = recordOf2(tool);
-    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 = recordOf2(entry);
-    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)"}`);
-  const hookEvents = names(capabilities.hookEvents);
-  const hookList = Array.isArray(capabilities.hookEvents) ? capabilities.hookEvents.map(asText).filter(Boolean) : hookEvents;
-  lines.push(`  hooks       ${hookList.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 : "";
-  const cwd = typeof capabilities.cwd === "string" ? capabilities.cwd : "";
-  lines.push(`  model       ${model}${thinking ? ` \xB7 ${thinking}` : ""}`);
-  if (cwd)
-    lines.push(`  cwd         ${cwd}`);
-  lines.push("  (worker commands are in your palette as [worker \u2026] \xB7 /worker shows this again)");
-  return lines;
-}
-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");
-  }
-}
-var LOCALLY_OWNED_COMMANDS = new Set(["detach", "quit", "q", "stop", "worker", "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 {}
-  }
-}
-function createRigWorkerPiBridgeExtension(options) {
-  return (pi) => {
-    const registeredDaemonCommands = new Set;
-    let capabilityLines = null;
-    let statusText = "connecting to worker session";
-    let busy = true;
-    let connected = false;
-    let frame = 0;
-    let spinnerTimer = null;
-    const renderStatus = (ctx) => {
-      const prefix = busy ? `${SPINNER_FRAMES[frame]} ` : connected ? "\u25CF " : "\u25CB ";
-      ctx.ui.setStatus("rig-worker-pi", `${prefix}${statusText}`);
-    };
-    const setStatus = (ctx, text, isBusy) => {
-      statusText = text;
-      busy = isBusy;
-      renderStatus(ctx);
-    };
-    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.registerCommand("worker", {
-      description: "Show the worker session's real capabilities (tools, extensions, hooks, skills)",
-      handler: async (_args, ctx) => {
-        if (capabilityLines)
-          ctx.ui.setWidget("rig-worker-capabilities", capabilityLines, { placement: "aboveEditor" });
-        else
-          ctx.ui.notify("Worker capabilities are not available yet (still connecting).", "info");
-      }
-    });
-    pi.on("user_bash", (event) => ({
-      operations: createRemoteBashOperations(options.controller, event.excludeFromContext === true)
-    }));
-    pi.on("session_start", async (_event, ctx) => {
-      ctx.ui.setTitle("Rig \xB7 enriched bundled Pi");
-      setStatus(ctx, "waiting for worker Pi daemon", true);
-      spinnerTimer = setInterval(() => {
-        frame = (frame + 1) % SPINNER_FRAMES.length;
-        if (busy)
-          renderStatus(ctx);
-      }, 150);
-      ctx.ui.notify(`Enriched bundled Pi \u2014 native UI + Rig layers, worker brain. Attached to run ${options.runId}. /worker shows live capabilities \xB7 /detach exits \xB7 /stop cancels.`, "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) => setStatus(ctx, text, !connected),
-        onActivity: (label, detail) => {
-          const active = label !== "idle" && !/complete|ready/i.test(label);
-          setStatus(ctx, [label, detail].filter(Boolean).join(" \u2014 "), active);
-          if (/agent running|tool:/.test(label))
-            ctx.ui.setWidget("rig-worker-capabilities", undefined);
-        },
-        onConnectionChange: (nextConnected) => {
-          connected = nextConnected;
-          setStatus(ctx, nextConnected ? "worker session live" : "worker session disconnected", false);
-        },
-        onError: (message) => ctx.ui.notify(message, "error"),
-        onExtensionUiRequest: (request) => {
-          answerExtensionUiRequest(options, ctx, request);
-        },
-        onCatchUp: (messages, commands, capabilities) => {
-          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 (capabilities) {
-            capabilityLines = renderWorkerCapabilities(capabilities);
-            ctx.ui.setWidget("rig-worker-capabilities", capabilityLines, { placement: "aboveEditor" });
-          }
-        }
-      });
-      options.controller.connect().catch((error) => {
-        ctx.ui.notify(error instanceof Error ? error.message : String(error), "error");
-      });
-    });
-    pi.on("session_shutdown", () => {
-      if (spinnerTimer)
-        clearInterval(spinnerTimer);
-      options.controller.close();
-    });
-  };
-}
-export {
-  createRigWorkerPiBridgeExtension
-};