@h-rig/cli 0.0.6-alpha.4 → 0.0.6-alpha.41

package/dist/src/commands/_pi-install.js

@@ -3,7 +3,8 @@
 import { existsSync, readFileSync, rmSync } from "fs";
 import { homedir } from "os";
 import { resolve } from "path";
-var PI_RIG_PACKAGE_NAME = "@rig/pi-rig";
+var PI_RIG_PACKAGE_NAME = "@h-rig/pi-rig";
+var LEGACY_PI_RIG_PACKAGE_NAME = "@rig/pi-rig";
 var LEGACY_PI_RIG_MARKER = `// Managed by Rig. Source package: @rig/pi-rig.
 export { default } from '@rig/pi-rig';
 `;
@@ -31,7 +32,7 @@ function resolvePiHomeDir(inputHomeDir) {
 function piListContainsPiRig(output) {
   return output.split(/\r?\n/).some((line) => {
     const normalized = line.trim();
-    return normalized.includes(PI_RIG_PACKAGE_NAME) || /(?:^|[\\/])packages[\\/]pi-rig(?:$|\s)/.test(normalized);
+    return normalized.includes(PI_RIG_PACKAGE_NAME) || normalized.includes(LEGACY_PI_RIG_PACKAGE_NAME) || /(?:^|[\\/])packages[\\/]pi-rig(?:$|\s)/.test(normalized);
   });
 }
 async function safeRun(runner, command, options) {
@@ -147,7 +148,7 @@ async function ensureRemotePiRigInstalled(input) {
   const payload = await input.requestJson("/api/pi-rig/install", {
     method: "POST",
     headers: { "content-type": "application/json" },
-    body: JSON.stringify({ package: "@rig/pi-rig", scope: "global" })
+    body: JSON.stringify({ package: PI_RIG_PACKAGE_NAME, scope: "global" })
   });
   const record = payload && typeof payload === "object" && !Array.isArray(payload) ? payload : {};
   const piOk = record.piOk === true || record.ok === true;

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

@@ -0,0 +1,665 @@
+// @bun
+// packages/cli/src/commands/_pi-remote-session.ts
+import { AgentSession as PiAgentSession } 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 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
+  };
+}
+function resolveSelectedConnection(projectRoot, options = {}) {
+  const repo = readRepoConnection(projectRoot);
+  if (!repo)
+    return null;
+  if (repo.selected === "local")
+    return { alias: "local", connection: { kind: "local", mode: "auto" } };
+  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 };
+}
+
+// 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);
+}
+async function ensureServerForCli(projectRoot) {
+  try {
+    const selected = resolveSelectedConnection(projectRoot);
+    if (selected?.connection.kind === "remote") {
+      return {
+        baseUrl: selected.connection.baseUrl,
+        authToken: readGitHubBearerTokenForRemote(projectRoot),
+        connectionKind: "remote"
+      };
+    }
+    const connection = await ensureLocalRigServerConnection(projectRoot);
+    return {
+      baseUrl: connection.baseUrl,
+      authToken: connection.authToken,
+      connectionKind: "local"
+    };
+  } catch (error) {
+    if (error instanceof Error) {
+      throw new CliError2(error.message, 1);
+    }
+    throw error;
+  }
+}
+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 response = await fetch(`${server.baseUrl}${pathname}`, {
+    ...init,
+    headers: mergeHeaders(init.headers, server.authToken)
+  });
+  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 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);
+  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,
+    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] = 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
+      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 prompt(text, options) {
+    const trimmed = text.trim();
+    if (!trimmed)
+      return;
+    if (trimmed.startsWith("/")) {
+      if (await this._tryExecuteExtensionCommand(trimmed))
+        return;
+      await this.remote.sendCommand(trimmed);
+      return;
+    }
+    if (trimmed.startsWith("!")) {
+      await this.remote.sendShell(trimmed);
+      return;
+    }
+    const behavior = options?.streamingBehavior ?? (this.isStreaming || this.isCompacting ? "steer" : undefined);
+    options?.preflightResult?.(true);
+    await this.remote.sendPrompt(trimmed, behavior);
+  }
+  async steer(text) {
+    await this.remote.sendPrompt(text, "steer");
+  }
+  async followUp(text) {
+    await this.remote.sendPrompt(text, "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
+};