@dxos/edge-client 0.8.2-main.12df754 → 0.8.2-main.30e4dbb

package/src/edge-ws-connection.ts

@@ -8,21 +8,18 @@ import { scheduleTask, scheduleTaskInterval } from '@dxos/async';
 import { Context, Resource } from '@dxos/context';
 import { invariant } from '@dxos/invariant';
 import { log, logInfo } from '@dxos/log';
+import { EdgeWebsocketProtocol } from '@dxos/protocols';
 import { buf } from '@dxos/protocols/buf';
 import { MessageSchema, type Message } from '@dxos/protocols/buf/dxos/edge/messenger_pb';
 
 import { protocol } from './defs';
 import { type EdgeIdentity } from './edge-identity';
+import { CLOUDFLARE_MESSAGE_MAX_BYTES, WebSocketMuxer } from './edge-ws-muxer';
 import { toUint8Array } from './protocol';
 
 const SIGNAL_KEEPALIVE_INTERVAL = 4_000;
 const SIGNAL_KEEPALIVE_TIMEOUT = 12_000;
 
-/**
- * 1MB websocket message limit: https://developers.cloudflare.com/durable-objects/platform/limits/
- */
-const CLOUDFLARE_MESSAGE_LENGTH_LIMIT = 1024 * 1024;
-
 export type EdgeWsConnectionCallbacks = {
   onConnected: () => void;
   onMessage: (message: Message) => void;
@@ -32,6 +29,7 @@ export type EdgeWsConnectionCallbacks = {
 export class EdgeWsConnection extends Resource {
   private _inactivityTimeoutCtx: Context | undefined;
   private _ws: WebSocket | undefined;
+  private _wsMuxer: WebSocketMuxer | undefined;
 
   constructor(
     private readonly _identity: EdgeIdentity,
@@ -52,24 +50,34 @@ export class EdgeWsConnection extends Resource {
 
   public send(message: Message) {
     invariant(this._ws);
+    invariant(this._wsMuxer);
     log('sending...', { peerKey: this._identity.peerKey, payload: protocol.getPayloadType(message) });
-    const encoded = buf.toBinary(MessageSchema, message);
-    if (encoded.byteLength >= CLOUDFLARE_MESSAGE_LENGTH_LIMIT) {
-      log.error('edge message dropped due to websocket message limit', {
-        byteLength: encoded.byteLength,
-        serviceId: message.serviceId,
-        payload: protocol.getPayloadType(message),
-      });
-      return;
+    if (this._ws?.protocol.includes(EdgeWebsocketProtocol.V0)) {
+      const binary = buf.toBinary(MessageSchema, message);
+      if (binary.length > CLOUDFLARE_MESSAGE_MAX_BYTES) {
+        log.error('Message dropped because it was too large (>1MB).', {
+          byteLength: binary.byteLength,
+          serviceId: message.serviceId,
+          payload: protocol.getPayloadType(message),
+        });
+        return;
+      }
+      this._ws.send(binary);
+    } else {
+      this._wsMuxer.send(message).catch((e) => log.catch(e));
     }
-    this._ws.send(encoded);
   }
 
   protected override async _open() {
+    const baseProtocols = [...Object.values(EdgeWebsocketProtocol)];
     this._ws = new WebSocket(
       this._connectionInfo.url.toString(),
-      this._connectionInfo.protocolHeader ? [this._connectionInfo.protocolHeader] : [],
+      this._connectionInfo.protocolHeader
+        ? [...baseProtocols, this._connectionInfo.protocolHeader]
+        : [...baseProtocols],
     );
+    const muxer = new WebSocketMuxer(this._ws);
+    this._wsMuxer = muxer;
 
     this._ws.onopen = () => {
       if (this.isOpen) {
@@ -84,6 +92,7 @@ export class EdgeWsConnection extends Resource {
       if (this.isOpen) {
         log.warn('disconnected while being open', { code: event.code, reason: event.reason });
         this._callbacks.onRestartRequired();
+        muxer.destroy();
       }
     };
     this._ws.onerror = (event) => {
@@ -106,9 +115,16 @@ export class EdgeWsConnection extends Resource {
         this._rescheduleHeartbeatTimeout();
         return;
       }
-      const data = await toUint8Array(event.data);
-      if (this.isOpen) {
-        const message = buf.fromBinary(MessageSchema, data);
+      const bytes = await toUint8Array(event.data);
+      if (!this.isOpen) {
+        return;
+      }
+
+      const message = this._ws?.protocol?.includes(EdgeWebsocketProtocol.V0)
+        ? buf.fromBinary(MessageSchema, bytes)
+        : muxer.receiveData(bytes);
+
+      if (message) {
         log('received', { from: message.source, payload: protocol.getPayloadType(message) });
         this._callbacks.onMessage(message);
       }
@@ -121,6 +137,8 @@ export class EdgeWsConnection extends Resource {
     try {
       this._ws?.close();
       this._ws = undefined;
+      this._wsMuxer?.destroy();
+      this._wsMuxer = undefined;
     } catch (err) {
       if (err instanceof Error && err.message.includes('WebSocket is closed before the connection is established.')) {
         return;

package/src/edge-ws-muxer.test.ts

@@ -0,0 +1,55 @@
+//
+// Copyright 2024 DXOS.org
+//
+
+import { describe, expect, test } from 'vitest';
+
+import { bufWkt } from '@dxos/protocols/buf';
+import { TextMessageSchema } from '@dxos/protocols/buf/dxos/edge/messenger_pb';
+
+import { protocol } from './defs';
+import type { EdgeIdentity } from './edge-identity';
+import { WebSocketMuxer } from './edge-ws-muxer';
+
+const MAX_CHUNK_LENGTH = 16;
+
+describe('WebSocketMuxerTest', () => {
+  test('basic message reassembly', async () => {
+    const { muxer: muxer1, sentMessages } = await createMuxer();
+    const { muxer: muxer2 } = await createMuxer();
+    const content = 'A'.repeat(MAX_CHUNK_LENGTH);
+    await muxer1.send(textMessage(content));
+
+    expect(sentMessages.length).toBeGreaterThan(1);
+    for (const chunk of sentMessages.slice(0, -1)) {
+      expect(muxer2.receiveData(chunk)).toBeUndefined();
+    }
+
+    const reassembledMessage = muxer2.receiveData(sentMessages.at(-1)!)!;
+    expect(reassembledMessage).toBeDefined();
+
+    const decoded = bufWkt.anyUnpack(reassembledMessage.payload!, TextMessageSchema);
+    expect(decoded?.message).toStrictEqual(content);
+  });
+});
+
+const textMessage = (message: string, source?: EdgeIdentity) =>
+  protocol.createMessage(TextMessageSchema, {
+    source: source && { peerKey: source.peerKey, identityKey: source.identityKey },
+    serviceId: 'test-service',
+    payload: { message },
+  });
+
+const createMuxer = async () => {
+  const sentMessages: Uint8Array[] = [];
+  const muxer = new WebSocketMuxer(
+    {
+      readyState: 1,
+      send: (message: string) => {
+        sentMessages.push(Buffer.from(message));
+      },
+    },
+    { maxChunkLength: MAX_CHUNK_LENGTH },
+  );
+  return { muxer, sentMessages };
+};

package/src/edge-ws-muxer.ts

@@ -0,0 +1,217 @@
+//
+// Copyright 2025 DXOS.org
+//
+
+import { Trigger } from '@dxos/async';
+import { log } from '@dxos/log';
+import { buf } from '@dxos/protocols/buf';
+import { MessageSchema, type Message } from '@dxos/protocols/buf/dxos/edge/messenger_pb';
+
+import { protocol } from './defs';
+
+/**
+ * 0000 0001 - message contains a part of segmented message chunk sequence.
+ * The next byte defines a channel id and the rest of the message contains a part of Message proto binary.
+ * Messages from different channels might interleave.
+ * When the flag is NOT set the rest of the message should be interpreted as the valid Message proto binary.
+ */
+const FLAG_SEGMENT_SEQ = 1;
+/**
+ * 0000 0010 - message terminates a segmented message chunk sequence.
+ * All the chunks accumulated for the channel specified by the second byte can be concatenated
+ * and interpreted as a valid Message proto binary.
+ */
+const FLAG_SEGMENT_SEQ_TERMINATED = 1 << 1;
+
+/**
+ * https://developers.cloudflare.com/durable-objects/platform/limits/
+ */
+export const CLOUDFLARE_MESSAGE_MAX_BYTES = 1000 * 1000; // 1MB
+export const CLOUDFLARE_RPC_MAX_BYTES = 32 * 1000 * 1000; // 32MB
+
+const MAX_CHUNK_LENGTH = 16384;
+const MAX_BUFFERED_AMOUNT = CLOUDFLARE_MESSAGE_MAX_BYTES;
+const BUFFER_FULL_BACKOFF_TIMEOUT = 100;
+
+export class WebSocketMuxer {
+  private readonly _inMessageAccumulator = new Map<number, Buffer[]>();
+  private readonly _outMessageChunks = new Map<number, MessageChunk[]>();
+  private readonly _outMessageChannelByService = new Map<string, number>();
+
+  private _sendTimeout: any | undefined;
+
+  private readonly _maxChunkLength: number;
+
+  constructor(
+    private readonly _ws: WebSocketCompat,
+    config?: { maxChunkLength: number },
+  ) {
+    this._maxChunkLength = config?.maxChunkLength ?? MAX_CHUNK_LENGTH;
+  }
+
+  /**
+   * Resolves when all the message chunks get enqueued for sending.
+   */
+  public async send(message: Message): Promise<void> {
+    const binary = buf.toBinary(MessageSchema, message);
+    const channelId = this._resolveChannel(message);
+    if (
+      (channelId == null && binary.byteLength > CLOUDFLARE_MESSAGE_MAX_BYTES) ||
+      binary.byteLength > CLOUDFLARE_RPC_MAX_BYTES
+    ) {
+      log.error('Large message dropped', {
+        byteLength: binary.byteLength,
+        serviceId: message.serviceId,
+        payload: protocol.getPayloadType(message),
+        channelId,
+      });
+      return;
+    }
+
+    if (channelId == null || binary.length < this._maxChunkLength) {
+      const flags = Buffer.from([0]);
+      this._ws.send(Buffer.concat([flags, binary]));
+      return;
+    }
+
+    const terminatorSentTrigger = new Trigger();
+    const messageChunks: MessageChunk[] = [];
+    for (let i = 0; i < binary.length; i += this._maxChunkLength) {
+      const chunk = binary.slice(i, i + this._maxChunkLength);
+      const isLastChunk = i + this._maxChunkLength >= binary.length;
+      if (isLastChunk) {
+        const flags = Buffer.from([FLAG_SEGMENT_SEQ | FLAG_SEGMENT_SEQ_TERMINATED, channelId]);
+        messageChunks.push({ payload: Buffer.concat([flags, chunk]), trigger: terminatorSentTrigger });
+      } else {
+        const flags = Buffer.from([FLAG_SEGMENT_SEQ, channelId]);
+        messageChunks.push({ payload: Buffer.concat([flags, chunk]) });
+      }
+    }
+
+    const queuedMessages = this._outMessageChunks.get(channelId);
+    if (queuedMessages) {
+      queuedMessages.push(...messageChunks);
+    } else {
+      this._outMessageChunks.set(channelId, messageChunks);
+    }
+
+    this._sendChunkedMessages();
+
+    return terminatorSentTrigger.wait();
+  }
+
+  public receiveData(data: Uint8Array): Message | undefined {
+    if ((data[0] & FLAG_SEGMENT_SEQ) === 0) {
+      return buf.fromBinary(MessageSchema, data.slice(1));
+    }
+
+    const [flags, channelId, ...payload] = data;
+    let chunkAccumulator = this._inMessageAccumulator.get(channelId);
+    if (chunkAccumulator) {
+      chunkAccumulator.push(Buffer.from(payload));
+    } else {
+      chunkAccumulator = [Buffer.from(payload)];
+      this._inMessageAccumulator.set(channelId, chunkAccumulator);
+    }
+
+    if ((flags & FLAG_SEGMENT_SEQ_TERMINATED) === 0) {
+      return undefined;
+    }
+
+    const message = buf.fromBinary(MessageSchema, Buffer.concat(chunkAccumulator));
+    this._inMessageAccumulator.delete(channelId);
+    return message;
+  }
+
+  public destroy() {
+    if (this._sendTimeout) {
+      clearTimeout(this._sendTimeout);
+      this._sendTimeout = undefined;
+    }
+    for (const channelChunks of this._outMessageChunks.values()) {
+      channelChunks.forEach((chunk) => chunk.trigger?.wake());
+    }
+    this._outMessageChunks.clear();
+    this._inMessageAccumulator.clear();
+    this._outMessageChannelByService.clear();
+  }
+
+  private _sendChunkedMessages() {
+    if (this._sendTimeout) {
+      return;
+    }
+
+    const send = () => {
+      if (this._ws.readyState === WebSocket.CLOSING || this._ws.readyState === WebSocket.CLOSED) {
+        log.warn('send called for closed websocket');
+        this._sendTimeout = undefined;
+        return;
+      }
+
+      let timeout = 0;
+      const emptyChannels: number[] = [];
+      for (const [channelId, messages] of this._outMessageChunks.entries()) {
+        if (this._ws.bufferedAmount != null) {
+          if (this._ws.bufferedAmount + MAX_CHUNK_LENGTH > MAX_BUFFERED_AMOUNT) {
+            timeout = BUFFER_FULL_BACKOFF_TIMEOUT;
+            break;
+          }
+        }
+
+        const nextMessage = messages.shift();
+        if (nextMessage) {
+          this._ws.send(nextMessage.payload);
+          nextMessage.trigger?.wake();
+        } else {
+          emptyChannels.push(channelId);
+        }
+      }
+
+      emptyChannels.forEach((channelId) => this._outMessageChunks.delete(channelId));
+
+      if (this._outMessageChunks.size > 0) {
+        this._sendTimeout = setTimeout(send, timeout);
+      } else {
+        this._sendTimeout = undefined;
+      }
+    };
+    this._sendTimeout = setTimeout(send);
+  }
+
+  private _resolveChannel(message: Message): number | undefined {
+    if (!message.serviceId) {
+      return undefined;
+    }
+    let id = this._outMessageChannelByService.get(message.serviceId);
+    if (!id) {
+      id = this._outMessageChannelByService.size + 1;
+      this._outMessageChannelByService.set(message.serviceId, id);
+    }
+    return id;
+  }
+}
+
+type WebSocketCompat = {
+  readonly readyState: number;
+  /**
+   * Not available in workerd.
+   */
+  bufferedAmount?: number;
+  send(message: (ArrayBuffer | ArrayBufferView) | string): void;
+};
+
+type MessageChunk = {
+  payload: Buffer;
+  /**
+   * Wakes when the payload is enqueued by WebSocket.
+   */
+  trigger?: Trigger;
+};
+
+/**
+ * To avoid using isomorphic-ws on edge.
+ */
+enum WebSocket {
+  CLOSING = 2,
+  CLOSED = 3,
+}

package/src/index.ts

@@ -11,3 +11,4 @@ export * from './errors';
 export * from './auth';
 export * from './edge-http-client';
 export * from './edge-identity';
+export * from './edge-ws-muxer';

package/src/testing/test-utils.ts

@@ -6,10 +6,12 @@ import WebSocket from 'isomorphic-ws';
 
 import { Trigger } from '@dxos/async';
 import { log } from '@dxos/log';
+import { EdgeWebsocketProtocol } from '@dxos/protocols';
 import { buf } from '@dxos/protocols/buf';
 import { MessageSchema, TextMessageSchema, type Message } from '@dxos/protocols/buf/dxos/edge/messenger_pb';
 
 import { protocol } from '../defs';
+import { WebSocketMuxer } from '../edge-ws-muxer';
 import { toUint8Array } from '../protocol';
 
 export const DEFAULT_PORT = 8080;
@@ -21,24 +23,33 @@ type TestEdgeWsServerParams = {
 };
 
 export const createTestEdgeWsServer = async (port = DEFAULT_PORT, params?: TestEdgeWsServerParams) => {
-  const server = new WebSocket.Server({ port, verifyClient: createConnectionDelayHandler(params) });
+  const wsServer = new WebSocket.Server({
+    port,
+    verifyClient: createConnectionDelayHandler(params),
+    handleProtocols: () => [EdgeWebsocketProtocol.V1],
+  });
 
-  let connection: WebSocket | undefined;
+  let connection: { ws: WebSocket; muxer: WebSocketMuxer } | undefined;
 
   const messageSink: any[] = [];
   const messageSourceLog: any[] = [];
   const closeTrigger = new Trigger();
-  const sendResponseMessage = createResponseSender(() => connection!);
+  const sendResponseMessage = createResponseSender(() => connection!.muxer);
 
-  server.on('connection', (ws) => {
-    connection = ws;
+  wsServer.on('connection', (ws) => {
+    const muxer = new WebSocketMuxer(ws);
+    connection = { ws, muxer };
     ws.on('error', (err) => log.catch(err));
     ws.on('message', async (data) => {
       if (String(data) === '__ping__') {
         ws.send('__pong__');
         return;
       }
-      const { request, requestPayload } = await decodeRequest(params, data);
+      const message = muxer.receiveData(await toUint8Array(data));
+      if (!message) {
+        return;
+      }
+      const { request, requestPayload } = await decodePayload(message, params);
       messageSourceLog.push(request.source);
       if (params?.messageHandler) {
         const responsePayload = await params.messageHandler(requestPayload);
@@ -57,19 +68,19 @@ export const createTestEdgeWsServer = async (port = DEFAULT_PORT, params?: TestE
   });
 
   return {
-    server,
+    server: wsServer,
     messageSink,
     messageSourceLog,
-    endpoint: `ws://localhost:${port}`,
-    cleanup: () => server.close(),
+    endpoint: `ws://127.0.0.1:${port}`,
+    cleanup: () => wsServer.close(),
     currentConnection: () => connection,
     sendResponseMessage,
     sendMessage: (msg: Message) => {
-      connection!.send(buf.toBinary(MessageSchema, msg));
+      return connection!.muxer.send(msg);
     },
     closeConnection: () => {
       closeTrigger.reset();
-      connection!.close(1011);
+      connection!.ws.close(1011);
       return closeTrigger.wait();
     },
   };
@@ -89,27 +100,23 @@ const createConnectionDelayHandler = (params: TestEdgeWsServerParams | undefined
   };
 };
 
-const createResponseSender = (connection: () => WebSocket) => {
+const createResponseSender = (connection: () => WebSocketMuxer) => {
   return (request: Message, responsePayload: Uint8Array) => {
     const recipient = request.source!;
-    connection().send(
-      buf.toBinary(
-        MessageSchema,
-        buf.create(MessageSchema, {
-          source: {
-            identityKey: recipient.identityKey,
-            peerKey: recipient.peerKey,
-          },
-          serviceId: request.serviceId!,
-          payload: { value: responsePayload },
-        }),
-      ),
+    void connection().send(
+      buf.create(MessageSchema, {
+        source: {
+          identityKey: recipient.identityKey,
+          peerKey: recipient.peerKey,
+        },
+        serviceId: request.serviceId!,
+        payload: { value: responsePayload },
+      }),
     );
   };
 };
 
-const decodeRequest = async (params: TestEdgeWsServerParams | undefined, data: any) => {
-  const request = buf.fromBinary(MessageSchema, await toUint8Array(data));
+const decodePayload = async (request: Message, params: TestEdgeWsServerParams | undefined) => {
   const requestPayload = params?.payloadDecoder
     ? params.payloadDecoder(request.payload!.value!)
     : protocol.getPayload(request, TextMessageSchema);

package/dist/lib/browser/chunk-ZWJXA37R.mjs

@@ -1,113 +0,0 @@
-// packages/core/mesh/edge-client/src/protocol.ts
-import { invariant } from "@dxos/invariant";
-import { buf, bufWkt } from "@dxos/protocols/buf";
-import { MessageSchema } from "@dxos/protocols/buf/dxos/edge/messenger_pb";
-import { bufferToArray } from "@dxos/util";
-var __dxlog_file = "/home/runner/work/dxos/dxos/packages/core/mesh/edge-client/src/protocol.ts";
-var getTypename = (typeName) => `type.googleapis.com/${typeName}`;
-var Protocol = class {
-  constructor(types) {
-    this._typeRegistry = buf.createRegistry(...types);
-  }
-  get typeRegistry() {
-    return this._typeRegistry;
-  }
-  toJson(message) {
-    try {
-      return buf.toJson(MessageSchema, message, {
-        registry: this.typeRegistry
-      });
-    } catch (err) {
-      return {
-        type: this.getPayloadType(message)
-      };
-    }
-  }
-  /**
-  * Return the payload with the given type.
-  */
-  getPayload(message, type) {
-    invariant(message.payload, void 0, {
-      F: __dxlog_file,
-      L: 40,
-      S: this,
-      A: [
-        "message.payload",
-        ""
-      ]
-    });
-    const payloadTypename = this.getPayloadType(message);
-    if (type && type.typeName !== payloadTypename) {
-      throw new Error(`Unexpected payload type: ${payloadTypename}; expected ${type.typeName}`);
-    }
-    invariant(bufWkt.anyIs(message.payload, type), `Unexpected payload type: ${payloadTypename}}`, {
-      F: __dxlog_file,
-      L: 46,
-      S: this,
-      A: [
-        "bufWkt.anyIs(message.payload, type)",
-        "`Unexpected payload type: ${payloadTypename}}`"
-      ]
-    });
-    const payload = bufWkt.anyUnpack(message.payload, this.typeRegistry);
-    invariant(payload, `Empty payload: ${payloadTypename}}`, {
-      F: __dxlog_file,
-      L: 48,
-      S: this,
-      A: [
-        "payload",
-        "`Empty payload: ${payloadTypename}}`"
-      ]
-    });
-    return payload;
-  }
-  /**
-  * Get the payload type.
-  */
-  getPayloadType(message) {
-    if (!message.payload) {
-      return void 0;
-    }
-    const [, type] = message.payload.typeUrl.split("/");
-    return type;
-  }
-  /**
-  * Create a packed message.
-  */
-  createMessage(type, { source, target, payload, serviceId }) {
-    return buf.create(MessageSchema, {
-      timestamp: (/* @__PURE__ */ new Date()).toISOString(),
-      source,
-      target,
-      serviceId,
-      payload: payload ? bufWkt.anyPack(type, buf.create(type, payload)) : void 0
-    });
-  }
-};
-var toUint8Array = async (data) => {
-  if (data instanceof Buffer) {
-    return bufferToArray(data);
-  }
-  if (data instanceof Blob) {
-    return new Uint8Array(await data.arrayBuffer());
-  }
-  throw new Error(`Unexpected datatype: ${data}`);
-};
-
-// packages/core/mesh/edge-client/src/defs.ts
-import { bufWkt as bufWkt2 } from "@dxos/protocols/buf";
-import { SwarmRequestSchema, SwarmResponseSchema, TextMessageSchema } from "@dxos/protocols/buf/dxos/edge/messenger_pb";
-var protocol = new Protocol([
-  SwarmRequestSchema,
-  SwarmResponseSchema,
-  TextMessageSchema,
-  bufWkt2.AnySchema
-]);
-
-export {
-  getTypename,
-  Protocol,
-  toUint8Array,
-  protocol
-};
-//# sourceMappingURL=chunk-ZWJXA37R.mjs.map

package/dist/lib/browser/chunk-ZWJXA37R.mjs.map

@@ -1,7 +0,0 @@
-{
-  "version": 3,
-  "sources": ["../../../src/protocol.ts", "../../../src/defs.ts"],
-  "sourcesContent": ["//\n// Copyright 2024 DXOS.org\n//\n\nimport { invariant } from '@dxos/invariant';\nimport { buf, bufWkt } from '@dxos/protocols/buf';\nimport { type Message, MessageSchema, type PeerSchema } from '@dxos/protocols/buf/dxos/edge/messenger_pb';\nimport { bufferToArray } from '@dxos/util';\n\nexport type PeerData = buf.MessageInitShape<typeof PeerSchema>;\n\nexport const getTypename = (typeName: string) => `type.googleapis.com/${typeName}`;\n\n/**\n * NOTE: The type registry should be extended with all message types.\n */\nexport class Protocol {\n  private readonly _typeRegistry: buf.Registry;\n\n  constructor(types: buf.DescMessage[]) {\n    this._typeRegistry = buf.createRegistry(...types);\n  }\n\n  get typeRegistry(): buf.Registry {\n    return this._typeRegistry;\n  }\n\n  toJson(message: Message): any {\n    try {\n      return buf.toJson(MessageSchema, message, { registry: this.typeRegistry });\n    } catch (err) {\n      return { type: this.getPayloadType(message) };\n    }\n  }\n\n  /**\n   * Return the payload with the given type.\n   */\n  getPayload<Desc extends buf.DescMessage>(message: Message, type: Desc): buf.MessageShape<Desc> {\n    invariant(message.payload);\n    const payloadTypename = this.getPayloadType(message);\n    if (type && type.typeName !== payloadTypename) {\n      throw new Error(`Unexpected payload type: ${payloadTypename}; expected ${type.typeName}`);\n    }\n\n    invariant(bufWkt.anyIs(message.payload, type), `Unexpected payload type: ${payloadTypename}}`);\n    const payload = bufWkt.anyUnpack(message.payload, this.typeRegistry) as buf.MessageShape<Desc>;\n    invariant(payload, `Empty payload: ${payloadTypename}}`);\n    return payload;\n  }\n\n  /**\n   * Get the payload type.\n   */\n  getPayloadType(message: Message): string | undefined {\n    if (!message.payload) {\n      return undefined;\n    }\n\n    const [, type] = message.payload.typeUrl.split('/');\n    return type;\n  }\n\n  /**\n   * Create a packed message.\n   */\n  createMessage<Desc extends buf.DescMessage>(\n    type: Desc,\n    {\n      source,\n      target,\n      payload,\n      serviceId,\n    }: {\n      source?: PeerData;\n      target?: PeerData[];\n      payload?: buf.MessageInitShape<Desc>;\n      serviceId?: string;\n    },\n  ) {\n    return buf.create(MessageSchema, {\n      timestamp: new Date().toISOString(),\n      source,\n      target,\n      serviceId,\n      payload: payload ? bufWkt.anyPack(type, buf.create(type, payload)) : undefined,\n    });\n  }\n}\n\n/**\n * Convert websocket data to Uint8Array.\n */\nexport const toUint8Array = async (data: any): Promise<Uint8Array> => {\n  // Node.\n  if (data instanceof Buffer) {\n    return bufferToArray(data);\n  }\n\n  // Browser.\n  if (data instanceof Blob) {\n    return new Uint8Array(await (data as Blob).arrayBuffer());\n  }\n\n  throw new Error(`Unexpected datatype: ${data}`);\n};\n", "//\n// Copyright 2024 DXOS.org\n//\n\nimport { bufWkt } from '@dxos/protocols/buf';\nimport { SwarmRequestSchema, SwarmResponseSchema, TextMessageSchema } from '@dxos/protocols/buf/dxos/edge/messenger_pb';\n\nimport { Protocol } from './protocol';\n\nexport const protocol = new Protocol([SwarmRequestSchema, SwarmResponseSchema, TextMessageSchema, bufWkt.AnySchema]);\n"],
-  "mappings": ";AAIA,SAASA,iBAAiB;AAC1B,SAASC,KAAKC,cAAc;AAC5B,SAAuBC,qBAAsC;AAC7D,SAASC,qBAAqB;;AAIvB,IAAMC,cAAc,CAACC,aAAqB,uBAAuBA,QAAAA;AAKjE,IAAMC,WAAN,MAAMA;EAGXC,YAAYC,OAA0B;AACpC,SAAKC,gBAAgBT,IAAIU,eAAc,GAAIF,KAAAA;EAC7C;EAEA,IAAIG,eAA6B;AAC/B,WAAO,KAAKF;EACd;EAEAG,OAAOC,SAAuB;AAC5B,QAAI;AACF,aAAOb,IAAIY,OAAOV,eAAeW,SAAS;QAAEC,UAAU,KAAKH;MAAa,CAAA;IAC1E,SAASI,KAAK;AACZ,aAAO;QAAEC,MAAM,KAAKC,eAAeJ,OAAAA;MAAS;IAC9C;EACF;;;;EAKAK,WAAyCL,SAAkBG,MAAoC;AAC7FjB,cAAUc,QAAQM,SAAO,QAAA;;;;;;;;;AACzB,UAAMC,kBAAkB,KAAKH,eAAeJ,OAAAA;AAC5C,QAAIG,QAAQA,KAAKX,aAAae,iBAAiB;AAC7C,YAAM,IAAIC,MAAM,4BAA4BD,eAAAA,cAA6BJ,KAAKX,QAAQ,EAAE;IAC1F;AAEAN,cAAUE,OAAOqB,MAAMT,QAAQM,SAASH,IAAAA,GAAO,4BAA4BI,eAAAA,KAAkB;;;;;;;;;AAC7F,UAAMD,UAAUlB,OAAOsB,UAAUV,QAAQM,SAAS,KAAKR,YAAY;AACnEZ,cAAUoB,SAAS,kBAAkBC,eAAAA,KAAkB;;;;;;;;;AACvD,WAAOD;EACT;;;;EAKAF,eAAeJ,SAAsC;AACnD,QAAI,CAACA,QAAQM,SAAS;AACpB,aAAOK;IACT;AAEA,UAAM,CAAA,EAAGR,IAAAA,IAAQH,QAAQM,QAAQM,QAAQC,MAAM,GAAA;AAC/C,WAAOV;EACT;;;;EAKAW,cACEX,MACA,EACEY,QACAC,QACAV,SACAW,UAAS,GAOX;AACA,WAAO9B,IAAI+B,OAAO7B,eAAe;MAC/B8B,YAAW,oBAAIC,KAAAA,GAAOC,YAAW;MACjCN;MACAC;MACAC;MACAX,SAASA,UAAUlB,OAAOkC,QAAQnB,MAAMhB,IAAI+B,OAAOf,MAAMG,OAAAA,CAAAA,IAAYK;IACvE,CAAA;EACF;AACF;AAKO,IAAMY,eAAe,OAAOC,SAAAA;AAEjC,MAAIA,gBAAgBC,QAAQ;AAC1B,WAAOnC,cAAckC,IAAAA;EACvB;AAGA,MAAIA,gBAAgBE,MAAM;AACxB,WAAO,IAAIC,WAAW,MAAOH,KAAcI,YAAW,CAAA;EACxD;AAEA,QAAM,IAAIpB,MAAM,wBAAwBgB,IAAAA,EAAM;AAChD;;;ACrGA,SAASK,UAAAA,eAAc;AACvB,SAASC,oBAAoBC,qBAAqBC,yBAAyB;AAIpE,IAAMC,WAAW,IAAIC,SAAS;EAACC;EAAoBC;EAAqBC;EAAmBC,QAAOC;CAAU;",
-  "names": ["invariant", "buf", "bufWkt", "MessageSchema", "bufferToArray", "getTypename", "typeName", "Protocol", "constructor", "types", "_typeRegistry", "createRegistry", "typeRegistry", "toJson", "message", "registry", "err", "type", "getPayloadType", "getPayload", "payload", "payloadTypename", "Error", "anyIs", "anyUnpack", "undefined", "typeUrl", "split", "createMessage", "source", "target", "serviceId", "create", "timestamp", "Date", "toISOString", "anyPack", "toUint8Array", "data", "Buffer", "Blob", "Uint8Array", "arrayBuffer", "bufWkt", "SwarmRequestSchema", "SwarmResponseSchema", "TextMessageSchema", "protocol", "Protocol", "SwarmRequestSchema", "SwarmResponseSchema", "TextMessageSchema", "bufWkt", "AnySchema"]
-}