Class WorkspaceRuntimeBridge

packages/computer/src/runtime/bridge.ts:8–335

import type { WorkspaceTrustedModule } from "./types.js";

export class WorkspaceRuntimeBridge extends RpcTarget {
  readonly #capability: WorkspaceRuntimeCapability;
  readonly #git: GitClient | undefined;
  readonly #artifacts: ArtifactClient | undefined;
  readonly #trustedModules: Record<string, WorkspaceTrustedModule>;
  readonly #allowGitNetwork: boolean;
  readonly #allowArtifactNetwork: boolean;
  readonly #maxPayloadBytes: number;
  readonly #maxCallDurationMs: number;
  readonly #maxConcurrentCalls: number;
  readonly #maxCalls: number;
  readonly #maxTotalRequestBytes: number;
  readonly #maxTotalResponseBytes: number;
  readonly #maxResultBytes: number;
  readonly #onAttachOutput?: (readable: ReadableStream<Uint8Array>) => Promise<void>;
  readonly #inFlight = new Set<Promise<string>>();
  readonly #abortControllers = new Set<AbortController>();
  #cancelled = false;
  #callTimedOut = false;
  #calls = 0;
  #requestBytes = 0;
  #responseBytes = 0;

  constructor(
    capability: WorkspaceRuntimeCapability,
    integrations: {
      git?: GitClient;
      artifacts?: ArtifactClient;
      trustedModules?: Record<string, WorkspaceTrustedModule>;
      allowGitNetwork?: boolean;
      allowArtifactNetwork?: boolean;
      maxPayloadBytes?: number;
      maxCallDurationMs?: number;
      maxConcurrentCalls?: number;
      maxCalls?: number;
      maxTotalRequestBytes?: number;
      maxTotalResponseBytes?: number;
      maxResultBytes?: number;
      onAttachOutput?: (readable: ReadableStream<Uint8Array>) => Promise<void>;
    } = {},
  ) {
    super();
    this.#capability = capability;
    this.#git = integrations.git;
    this.#artifacts = integrations.artifacts;
    this.#trustedModules = integrations.trustedModules ?? {};
    this.#allowGitNetwork = integrations.allowGitNetwork ?? false;
    this.#allowArtifactNetwork = integrations.allowArtifactNetwork ?? false;
    this.#maxPayloadBytes = integrations.maxPayloadBytes ?? 1024 * 1024;
    this.#maxCallDurationMs = integrations.maxCallDurationMs ?? 30_000;
    this.#maxConcurrentCalls = integrations.maxConcurrentCalls ?? 16;
    this.#maxCalls = integrations.maxCalls ?? 256;
    this.#maxTotalRequestBytes = integrations.maxTotalRequestBytes ?? 8 * 1024 * 1024;
    this.#maxTotalResponseBytes = integrations.maxTotalResponseBytes ?? 8 * 1024 * 1024;
    this.#maxResultBytes = integrations.maxResultBytes ?? 1024 * 1024;
    this.#onAttachOutput = integrations.onAttachOutput;
  }