Class WorkspaceRuntimeExecHandleStub

packages/computer/src/stub.ts:280–372

}

export class WorkspaceRuntimeExecHandleStub<
  E extends "utf8" | undefined = undefined,
> extends RpcTarget {
  readonly #handle: WorkspaceRuntimeExecHandle<E>;
  #consumed = false;
  #disposed = false;
  #reader: ReadableStreamDefaultReader<WorkspaceRuntimeEvent<"utf8" | undefined>> | undefined;

  constructor(handle: WorkspaceRuntimeExecHandle<E>) {
    super();
    this.#handle = handle;
    trackStub(this);
  }

  [Symbol.dispose](): void {
    if (this.#disposed) return;
    this.#disposed = true;
    const reader = this.#reader;
    this.#reader = undefined;
    if (reader) {
      void reader
        .cancel()
        .catch(() => undefined)
        .finally(() => reader.releaseLock());
    } else {
      const cancel = (this.#handle as { cancel?: () => Promise<void> }).cancel;
      if (typeof cancel === "function") void cancel.call(this.#handle).catch(() => undefined);
    }
    untrackStub(this);
  }

  get id(): string {
    return this.#handle.id;
  }

  get backend(): string {
    return this.#handle.backend;
  }

  async result(): Promise<WorkspaceRuntimeResult<E>> {
    this.#claim();
    return this.#handle.result();
  }

  stream(): ReadableStream<Uint8Array> {
    this.#claim();
    const reader = (
      this.#handle as ReadableStream<WorkspaceRuntimeEvent<"utf8" | undefined>>
    ).getReader();
    this.#reader = reader;
    const owner = this;
    return new ReadableStream<Uint8Array>(
      {
        async pull(controller) {
          try {
            const { value, done } = await reader.read();
            if (done) {
              reader.releaseLock();