Function createWorkspaceClient

packages/rpc/src/client.ts:136–178

(options: {
  url: string;
  WebSocketImpl?: typeof WebSocket;
})

// shim from createSyncClient doesn't compose cleanly. Add when a
// caller actually needs it.
export function createWorkspaceClient(options: {
  url: string;
  WebSocketImpl?: typeof WebSocket;
}): WorkspaceClient {
  const WS = options.WebSocketImpl ?? WebSocket;
  const ws = new WS(options.url);
  // Same WebSocket / RpcStub cast pattern as createSyncClient —
  // see comment there. WorkspaceRPC adds the composite shape so
  // callers can pipeline `.sync.push(...)` and `.shell.exec(...)`.
  const stub = newWebSocketRpcSession(
    ws as unknown as globalThis.WebSocket,
  ) as RpcStub<WorkspaceRPC>;
  return new Proxy(stub, {
    get(target, prop, receiver) {
      if (prop === "close") {
        return async () => {
          // Same dispose-before-close pattern as createSyncClient.
          try {
            (target as unknown as Disposable)[Symbol.dispose]?.();
          } catch {
            // already disposed; idempotent
          }
          await new Promise<void>((resolve) => {
            const w = ws as unknown as { readyState: number; close: () => void };
            if (w.readyState >= 2) {
              resolve();
              return;
            }
            (ws as unknown as EventTarget).addEventListener("close", () => resolve(), {
              once: true,
            });
            w.close();
            setTimeout(resolve, 200);
          });
        };
      }
      return Reflect.get(target, prop, receiver);
    },
    // As in createSyncClient, the Proxy is structurally a
    // WorkspaceRPC stub + close(); route through unknown so TS
    // accepts the WorkspaceClient landing type.
  }) as unknown as WorkspaceClient;
}