Function drainModuleResult

packages/computer/src/runtime/runtime.ts:413–449

(
  events: ReadableStream<WorkspaceRuntimeEvent>,
  encoding: E | undefined,
  setReader: (reader: ReadableStreamDefaultReader<WorkspaceRuntimeEvent> | undefined) => void,
)

}

async function drainModuleResult<E extends ExecEncoding>(
  events: ReadableStream<WorkspaceRuntimeEvent>,
  encoding: E | undefined,
  setReader: (reader: ReadableStreamDefaultReader<WorkspaceRuntimeEvent> | undefined) => void,
): Promise<WorkspaceRuntimeResult<E>> {
  const stdout: Uint8Array[] = [];
  const stderr: Uint8Array[] = [];
  let value: WorkspaceRuntimeResult<E>["value"];
  let exitCode = 1;
  const reader = events.getReader();
  setReader(reader);
  try {
    while (true) {
      const next = await reader.read();
      if (next.done) break;
      const event = next.value;
      if (event.name === "stdout") stdout.push(event.value);
      if (event.name === "stderr") stderr.push(event.value);
      if (event.name === "result") value = event.value;
      if (event.name === "exit") exitCode = event.value;
    }
  } finally {
    reader.releaseLock();
    setReader(undefined);
  }
  return {
    status: exitCode === 0 ? "completed" : exitCode === 130 ? "cancelled" : "failed",
    exitCode,
    stdout: join(stdout, encoding) as WorkspaceRuntimeResult<E>["stdout"],
    stderr: join(stderr, encoding) as WorkspaceRuntimeResult<E>["stderr"],
    ...(value === undefined ? {} : { value }),
    pushed: 0,
    pulled: 0,
    skipped: [] as SkippedEntry[],
    sync: { status: "complete", applied: 0, skipped: [] },
  };
}

function isCancellationExitCode(exitCode: number) {
  return exitCode === 129 || exitCode === 130 || exitCode === 137 || exitCode === 143;