Function getWorkspace

packages/computer/src/client.ts:366–406

(handle: WorkspaceHandle)

export type WorkspaceHandle = { [WORKSPACE]?: unknown } | WorkspaceStubHost;

export async function getWorkspace(handle: WorkspaceHandle): Promise<WorkspaceClient> {
  const local = (handle as { [WORKSPACE]?: unknown })[WORKSPACE];
  if (local instanceof Workspace) {
    // Local path: delegate straight to the in-isolate Workspace.
    // Nothing to dispose — the durable object owns the Workspace
    // lifecycle.
    await local.ready();
    return makeClient(
      {
        fs: local.fs,
        runtime: local.runtime,
        get git() {
          return local.git;
        },
        artifacts: local.artifacts,
        assets: local.assets,
      },
      // Local handle is already a host ExecHandle — pass it through.
      (h) => h,
      () => {},
      local.useThink,
    );
  }
  // Remote path: fetch the stub over RPC and delegate to it. Handle
  // stubs need inflating from their JSONL stream into a host-shaped
  // ExecHandle.
  const stub = await (handle as WorkspaceStubHost).__getWorkspaceStub();
  try {
    return makeClient(
      stub,
      (h, metadata) => rebuildExecHandle(h as RemoteExecHandle, metadata),
      () => {
        (stub as { [Symbol.dispose]?: () => void })[Symbol.dispose]?.();
      },
      await stub.useThink,
    );
  } catch (error) {
    (stub as { [Symbol.dispose]?: () => void })[Symbol.dispose]?.();
    throw error;
  }
}