@h-rig/cli 0.0.6-alpha.64 → 0.0.6-alpha.66

package/dist/src/commands/_pi-remote-session.js

@@ -1,771 +0,0 @@
-// @bun
-// packages/cli/src/commands/_pi-remote-session.ts
-import { AgentSession as PiAgentSession, expandPromptTemplate } from "@earendil-works/pi-coding-agent";
-
-// 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 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
-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({});
-  }
-}
-
-class RigRemoteAgentSession extends PiAgentSession {
-  remote;
-  constructor(config, remote) {
-    super(config);
-    this.remote = remote;
-    remote.bindSession(this);
-  }
-  handleRemoteSessionEvent(eventValue) {
-    const event = recordOf(eventValue);
-    if (!event)
-      return;
-    const type = String(event.type ?? "");
-    if ((type === "message_end" || type === "turn_end") && recordOf(event.message)) {
-      this.agent.state.messages = [...this.agent.state.messages, event.message];
-    }
-    this._emit(eventValue);
-  }
-  replaceRemoteMessages(messages) {
-    this.agent.state.messages = messages;
-  }
-  async expandLocalInput(text) {
-    let current = text;
-    const runner = this.extensionRunner;
-    if (runner.hasHandlers("input")) {
-      const result = await runner.emitInput(current, undefined, "interactive");
-      if (result.action === "handled")
-        return null;
-      if (result.action === "transform")
-        current = result.text;
-    }
-    current = this._expandSkillCommand(current);
-    current = expandPromptTemplate(current, [...this.promptTemplates]);
-    return current;
-  }
-  async prompt(text, options) {
-    const trimmed = text.trim();
-    if (!trimmed)
-      return;
-    if (trimmed.startsWith("/")) {
-      if (await this._tryExecuteExtensionCommand(trimmed))
-        return;
-      const expanded2 = await this.expandLocalInput(trimmed);
-      if (expanded2 === null)
-        return;
-      if (expanded2 !== trimmed) {
-        const behavior2 = options?.streamingBehavior ?? (this.isStreaming || this.isCompacting ? "steer" : undefined);
-        options?.preflightResult?.(true);
-        await this.remote.sendPrompt(expanded2, behavior2);
-        return;
-      }
-      await this.remote.sendCommand(trimmed);
-      return;
-    }
-    if (trimmed.startsWith("!")) {
-      await this.remote.sendShell(trimmed);
-      return;
-    }
-    const expanded = await this.expandLocalInput(trimmed);
-    if (expanded === null)
-      return;
-    const behavior = options?.streamingBehavior ?? (this.isStreaming || this.isCompacting ? "steer" : undefined);
-    options?.preflightResult?.(true);
-    await this.remote.sendPrompt(expanded, behavior);
-  }
-  async steer(text) {
-    const trimmed = text.trim();
-    if (trimmed.startsWith("/") && await this._tryExecuteExtensionCommand(trimmed))
-      return;
-    const expanded = await this.expandLocalInput(trimmed);
-    if (expanded === null)
-      return;
-    await this.remote.sendPrompt(expanded, "steer");
-  }
-  async followUp(text) {
-    const trimmed = text.trim();
-    if (trimmed.startsWith("/") && await this._tryExecuteExtensionCommand(trimmed))
-      return;
-    const expanded = await this.expandLocalInput(trimmed);
-    if (expanded === null)
-      return;
-    await this.remote.sendPrompt(expanded, "followUp");
-  }
-  async abort() {
-    await this.remote.abort();
-  }
-  async compact(customInstructions) {
-    const pending = new Promise((resolve3, reject) => {
-      this.remote.registerPendingCompaction({ resolve: resolve3, reject });
-    });
-    await this.remote.sendCommand(`/compact${customInstructions ? ` ${customInstructions}` : ""}`);
-    return pending;
-  }
-  clearQueue() {
-    const cleared = {
-      steering: [...this.remote.status.steeringMessages],
-      followUp: [...this.remote.status.followUpMessages]
-    };
-    this.remote.status.steeringMessages = [];
-    this.remote.status.followUpMessages = [];
-    this.remote.status.pendingMessageCount = 0;
-    this.remote.sendCommand("/queue-clear").catch(() => {});
-    return cleared;
-  }
-  get isStreaming() {
-    return this.remote.status.isStreaming;
-  }
-  get isCompacting() {
-    return this.remote.status.isCompacting;
-  }
-  get isBashRunning() {
-    return this.remote.status.isBashRunning;
-  }
-  get pendingMessageCount() {
-    return this.remote.status.pendingMessageCount;
-  }
-  getSteeringMessages() {
-    return this.remote.status.steeringMessages;
-  }
-  getFollowUpMessages() {
-    return this.remote.status.followUpMessages;
-  }
-  get model() {
-    return this.remote.status.model ?? super.model;
-  }
-  async setModel(model) {
-    await this.remote.sendCommand(`/model ${model.provider}/${model.id}`);
-    this.remote.status.model = model;
-  }
-  get thinkingLevel() {
-    return this.remote.status.thinkingLevel ?? super.thinkingLevel;
-  }
-  setThinkingLevel(level) {
-    this.remote.status.thinkingLevel = level;
-    this.remote.sendCommand(`/thinking ${level}`).catch(() => {});
-  }
-  get sessionName() {
-    return this.remote.status.sessionName ?? super.sessionName;
-  }
-  setSessionName(name) {
-    this.remote.status.sessionName = name;
-    this.remote.sendCommand(`/name ${name}`).catch(() => {});
-  }
-  getSessionStats() {
-    return this.remote.status.stats ?? super.getSessionStats();
-  }
-  getContextUsage() {
-    return this.remote.status.contextUsage ?? super.getContextUsage();
-  }
-  dispose() {
-    this.remote.close();
-    super.dispose();
-  }
-}
-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)));
-        });
-      });
-    }
-  };
-}
-export {
-  createRemoteBashOperations,
-  RigRemoteSessionController,
-  RigRemoteAgentSession
-};