@dxos/edge-client 0.8.1 → 0.8.2-main.2f9c567

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"]
-}

package/dist/lib/node/chunk-ANV2HBEH.cjs

@@ -1,136 +0,0 @@
-"use strict";
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
-};
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
-      if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
-  }
-  return to;
-};
-var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-var chunk_ANV2HBEH_exports = {};
-__export(chunk_ANV2HBEH_exports, {
-  Protocol: () => Protocol,
-  getTypename: () => getTypename,
-  protocol: () => protocol,
-  toUint8Array: () => toUint8Array
-});
-module.exports = __toCommonJS(chunk_ANV2HBEH_exports);
-var import_invariant = require("@dxos/invariant");
-var import_buf = require("@dxos/protocols/buf");
-var import_messenger_pb = require("@dxos/protocols/buf/dxos/edge/messenger_pb");
-var import_util = require("@dxos/util");
-var import_buf2 = require("@dxos/protocols/buf");
-var import_messenger_pb2 = require("@dxos/protocols/buf/dxos/edge/messenger_pb");
-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 = import_buf.buf.createRegistry(...types);
-  }
-  get typeRegistry() {
-    return this._typeRegistry;
-  }
-  toJson(message) {
-    try {
-      return import_buf.buf.toJson(import_messenger_pb.MessageSchema, message, {
-        registry: this.typeRegistry
-      });
-    } catch (err) {
-      return {
-        type: this.getPayloadType(message)
-      };
-    }
-  }
-  /**
-  * Return the payload with the given type.
-  */
-  getPayload(message, type) {
-    (0, import_invariant.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}`);
-    }
-    (0, import_invariant.invariant)(import_buf.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 = import_buf.bufWkt.anyUnpack(message.payload, this.typeRegistry);
-    (0, import_invariant.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 import_buf.buf.create(import_messenger_pb.MessageSchema, {
-      timestamp: (/* @__PURE__ */ new Date()).toISOString(),
-      source,
-      target,
-      serviceId,
-      payload: payload ? import_buf.bufWkt.anyPack(type, import_buf.buf.create(type, payload)) : void 0
-    });
-  }
-};
-var toUint8Array = async (data) => {
-  if (data instanceof Buffer) {
-    return (0, import_util.bufferToArray)(data);
-  }
-  if (data instanceof Blob) {
-    return new Uint8Array(await data.arrayBuffer());
-  }
-  throw new Error(`Unexpected datatype: ${data}`);
-};
-var protocol = new Protocol([
-  import_messenger_pb2.SwarmRequestSchema,
-  import_messenger_pb2.SwarmResponseSchema,
-  import_messenger_pb2.TextMessageSchema,
-  import_buf2.bufWkt.AnySchema
-]);
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
-  Protocol,
-  getTypename,
-  protocol,
-  toUint8Array
-});
-//# sourceMappingURL=chunk-ANV2HBEH.cjs.map

package/dist/lib/node/chunk-ANV2HBEH.cjs.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,uBAA0B;AAC1B,iBAA4B;AAC5B,0BAA6D;AAC7D,kBAA8B;ACH9B,IAAAA,cAAuB;AACvB,IAAAC,uBAA2E;;ADMpE,IAAMC,cAAc,CAACC,aAAqB,uBAAuBA,QAAAA;AAKjE,IAAMC,WAAN,MAAMA;EAGXC,YAAYC,OAA0B;AACpC,SAAKC,gBAAgBC,eAAIC,eAAc,GAAIH,KAAAA;EAC7C;EAEA,IAAII,eAA6B;AAC/B,WAAO,KAAKH;EACd;EAEAI,OAAOC,SAAuB;AAC5B,QAAI;AACF,aAAOJ,eAAIG,OAAOE,mCAAeD,SAAS;QAAEE,UAAU,KAAKJ;MAAa,CAAA;IAC1E,SAASK,KAAK;AACZ,aAAO;QAAEC,MAAM,KAAKC,eAAeL,OAAAA;MAAS;IAC9C;EACF;;;;EAKAM,WAAyCN,SAAkBI,MAAoC;AAC7FG,oCAAUP,QAAQQ,SAAO,QAAA;;;;;;;;;AACzB,UAAMC,kBAAkB,KAAKJ,eAAeL,OAAAA;AAC5C,QAAII,QAAQA,KAAKb,aAAakB,iBAAiB;AAC7C,YAAM,IAAIC,MAAM,4BAA4BD,eAAAA,cAA6BL,KAAKb,QAAQ,EAAE;IAC1F;AAEAgB,oCAAUI,kBAAOC,MAAMZ,QAAQQ,SAASJ,IAAAA,GAAO,4BAA4BK,eAAAA,KAAkB;;;;;;;;;AAC7F,UAAMD,UAAUG,kBAAOE,UAAUb,QAAQQ,SAAS,KAAKV,YAAY;AACnES,oCAAUC,SAAS,kBAAkBC,eAAAA,KAAkB;;;;;;;;;AACvD,WAAOD;EACT;;;;EAKAH,eAAeL,SAAsC;AACnD,QAAI,CAACA,QAAQQ,SAAS;AACpB,aAAOM;IACT;AAEA,UAAM,CAAA,EAAGV,IAAAA,IAAQJ,QAAQQ,QAAQO,QAAQC,MAAM,GAAA;AAC/C,WAAOZ;EACT;;;;EAKAa,cACEb,MACA,EACEc,QACAC,QACAX,SACAY,UAAS,GAOX;AACA,WAAOxB,eAAIyB,OAAOpB,mCAAe;MAC/BqB,YAAW,oBAAIC,KAAAA,GAAOC,YAAW;MACjCN;MACAC;MACAC;MACAZ,SAASA,UAAUG,kBAAOc,QAAQrB,MAAMR,eAAIyB,OAAOjB,MAAMI,OAAAA,CAAAA,IAAYM;IACvE,CAAA;EACF;AACF;AAKO,IAAMY,eAAe,OAAOC,SAAAA;AAEjC,MAAIA,gBAAgBC,QAAQ;AAC1B,eAAOC,2BAAcF,IAAAA;EACvB;AAGA,MAAIA,gBAAgBG,MAAM;AACxB,WAAO,IAAIC,WAAW,MAAOJ,KAAcK,YAAW,CAAA;EACxD;AAEA,QAAM,IAAItB,MAAM,wBAAwBiB,IAAAA,EAAM;AAChD;AChGO,IAAMM,WAAW,IAAIzC,SAAS;EAAC0C;EAAoBC;EAAqBC;EAAmBzB,YAAAA,OAAO0B;CAAU;",
-  "names": ["import_buf", "import_messenger_pb", "getTypename", "typeName", "Protocol", "constructor", "types", "_typeRegistry", "buf", "createRegistry", "typeRegistry", "toJson", "message", "MessageSchema", "registry", "err", "type", "getPayloadType", "getPayload", "invariant", "payload", "payloadTypename", "Error", "bufWkt", "anyIs", "anyUnpack", "undefined", "typeUrl", "split", "createMessage", "source", "target", "serviceId", "create", "timestamp", "Date", "toISOString", "anyPack", "toUint8Array", "data", "Buffer", "bufferToArray", "Blob", "Uint8Array", "arrayBuffer", "protocol", "SwarmRequestSchema", "SwarmResponseSchema", "TextMessageSchema", "AnySchema"]
-}