@dxos/messaging 0.6.13 → 0.6.14-main.2b6a0f3

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

@@ -0,0 +1,688 @@
+"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_WDA7H2PO_exports = {};
+__export(chunk_WDA7H2PO_exports, {
+  MemorySignalManager: () => MemorySignalManager,
+  MemorySignalManagerContext: () => MemorySignalManagerContext,
+  Messenger: () => Messenger,
+  PeerInfoHash: () => PeerInfoHash
+});
+module.exports = __toCommonJS(chunk_WDA7H2PO_exports);
+var import_async = require("@dxos/async");
+var import_context = require("@dxos/context");
+var import_invariant = require("@dxos/invariant");
+var import_keys = require("@dxos/keys");
+var import_log = require("@dxos/log");
+var import_protocols = require("@dxos/protocols");
+var import_proto = require("@dxos/protocols/proto");
+var import_util = require("@dxos/util");
+var import_tracing = require("@dxos/tracing");
+var import_async2 = require("@dxos/async");
+var import_context2 = require("@dxos/context");
+var import_invariant2 = require("@dxos/invariant");
+var import_keys2 = require("@dxos/keys");
+var import_log2 = require("@dxos/log");
+var import_proto2 = require("@dxos/protocols/proto");
+var import_util2 = require("@dxos/util");
+var MessengerMonitor = class {
+  recordMessageAckFailed() {
+    import_tracing.trace.metrics.increment("dxos.mesh.signal.messenger.failed-ack", 1);
+  }
+  recordReliableMessage(params) {
+    import_tracing.trace.metrics.increment("dxos.mesh.signal.messenger.reliable-send", 1, {
+      tags: {
+        success: params.sent,
+        attempts: params.sendAttempts
+      }
+    });
+  }
+};
+var MESSAGE_TIMEOUT = 1e4;
+var __dxlog_file = "/home/runner/work/dxos/dxos/packages/core/mesh/messaging/src/messenger.ts";
+var ReliablePayload = import_proto.schema.getCodecForType("dxos.mesh.messaging.ReliablePayload");
+var Acknowledgement = import_proto.schema.getCodecForType("dxos.mesh.messaging.Acknowledgement");
+var RECEIVED_MESSAGES_GC_INTERVAL = 12e4;
+var Messenger = class {
+  constructor({ signalManager, retryDelay = 1e3 }) {
+    this._monitor = new MessengerMonitor();
+    this._listeners = new import_util.ComplexMap(({ peerId, payloadType }) => peerId + payloadType);
+    this._defaultListeners = /* @__PURE__ */ new Map();
+    this._onAckCallbacks = new import_util.ComplexMap(import_keys.PublicKey.hash);
+    this._receivedMessages = new import_util.ComplexSet(import_keys.PublicKey.hash);
+    this._toClear = new import_util.ComplexSet(import_keys.PublicKey.hash);
+    this._closed = true;
+    this._signalManager = signalManager;
+    this._retryDelay = retryDelay;
+    this.open();
+  }
+  open() {
+    if (!this._closed) {
+      return;
+    }
+    const traceId = import_keys.PublicKey.random().toHex();
+    import_log.log.trace("dxos.mesh.messenger.open", import_protocols.trace.begin({
+      id: traceId
+    }), {
+      F: __dxlog_file,
+      L: 72,
+      S: this,
+      C: (f, a) => f(...a)
+    });
+    this._ctx = new import_context.Context({
+      onError: (err) => import_log.log.catch(err, void 0, {
+        F: __dxlog_file,
+        L: 74,
+        S: this,
+        C: (f, a) => f(...a)
+      })
+    }, {
+      F: __dxlog_file,
+      L: 73
+    });
+    this._ctx.onDispose(this._signalManager.onMessage.on(async (message) => {
+      (0, import_log.log)("received message", {
+        from: message.author
+      }, {
+        F: __dxlog_file,
+        L: 78,
+        S: this,
+        C: (f, a) => f(...a)
+      });
+      await this._handleMessage(message);
+    }));
+    (0, import_async.scheduleTaskInterval)(this._ctx, async () => {
+      this._performGc();
+    }, RECEIVED_MESSAGES_GC_INTERVAL);
+    this._closed = false;
+    import_log.log.trace("dxos.mesh.messenger.open", import_protocols.trace.end({
+      id: traceId
+    }), {
+      F: __dxlog_file,
+      L: 93,
+      S: this,
+      C: (f, a) => f(...a)
+    });
+  }
+  async close() {
+    if (this._closed) {
+      return;
+    }
+    this._closed = true;
+    await this._ctx.dispose();
+  }
+  async sendMessage({ author, recipient, payload }) {
+    (0, import_invariant.invariant)(!this._closed, "Closed", {
+      F: __dxlog_file,
+      L: 105,
+      S: this,
+      A: [
+        "!this._closed",
+        "'Closed'"
+      ]
+    });
+    const messageContext = this._ctx.derive();
+    const reliablePayload = {
+      messageId: import_keys.PublicKey.random(),
+      payload
+    };
+    (0, import_invariant.invariant)(!this._onAckCallbacks.has(reliablePayload.messageId), void 0, {
+      F: __dxlog_file,
+      L: 112,
+      S: this,
+      A: [
+        "!this._onAckCallbacks.has(reliablePayload.messageId!)",
+        ""
+      ]
+    });
+    (0, import_log.log)("send message", {
+      messageId: reliablePayload.messageId,
+      author,
+      recipient
+    }, {
+      F: __dxlog_file,
+      L: 113,
+      S: this,
+      C: (f, a) => f(...a)
+    });
+    let messageReceived;
+    let timeoutHit;
+    let sendAttempts = 0;
+    const promise = new Promise((resolve, reject) => {
+      messageReceived = resolve;
+      timeoutHit = reject;
+    });
+    (0, import_async.scheduleExponentialBackoffTaskInterval)(messageContext, async () => {
+      (0, import_log.log)("retrying message", {
+        messageId: reliablePayload.messageId
+      }, {
+        F: __dxlog_file,
+        L: 128,
+        S: this,
+        C: (f, a) => f(...a)
+      });
+      sendAttempts++;
+      await this._encodeAndSend({
+        author,
+        recipient,
+        reliablePayload
+      }).catch((err) => (0, import_log.log)("failed to send message", {
+        err
+      }, {
+        F: __dxlog_file,
+        L: 131,
+        S: this,
+        C: (f, a) => f(...a)
+      }));
+    }, this._retryDelay);
+    (0, import_async.scheduleTask)(messageContext, () => {
+      (0, import_log.log)("message not delivered", {
+        messageId: reliablePayload.messageId
+      }, {
+        F: __dxlog_file,
+        L: 140,
+        S: this,
+        C: (f, a) => f(...a)
+      });
+      this._onAckCallbacks.delete(reliablePayload.messageId);
+      timeoutHit(new import_protocols.TimeoutError("signaling message not delivered", new import_async.TimeoutError(MESSAGE_TIMEOUT, "Message not delivered")));
+      void messageContext.dispose();
+      this._monitor.recordReliableMessage({
+        sendAttempts,
+        sent: false
+      });
+    }, MESSAGE_TIMEOUT);
+    this._onAckCallbacks.set(reliablePayload.messageId, () => {
+      messageReceived();
+      this._onAckCallbacks.delete(reliablePayload.messageId);
+      void messageContext.dispose();
+      this._monitor.recordReliableMessage({
+        sendAttempts,
+        sent: true
+      });
+    });
+    await this._encodeAndSend({
+      author,
+      recipient,
+      reliablePayload
+    });
+    return promise;
+  }
+  /**
+  * Subscribes onMessage function to messages that contains payload with payloadType.
+  * @param payloadType if not specified, onMessage will be subscribed to all types of messages.
+  */
+  async listen({ peer, payloadType, onMessage }) {
+    (0, import_invariant.invariant)(!this._closed, "Closed", {
+      F: __dxlog_file,
+      L: 178,
+      S: this,
+      A: [
+        "!this._closed",
+        "'Closed'"
+      ]
+    });
+    await this._signalManager.subscribeMessages(peer);
+    let listeners;
+    (0, import_invariant.invariant)(peer.peerKey, "Peer key is required", {
+      F: __dxlog_file,
+      L: 182,
+      S: this,
+      A: [
+        "peer.peerKey",
+        "'Peer key is required'"
+      ]
+    });
+    if (!payloadType) {
+      listeners = this._defaultListeners.get(peer.peerKey);
+      if (!listeners) {
+        listeners = /* @__PURE__ */ new Set();
+        this._defaultListeners.set(peer.peerKey, listeners);
+      }
+    } else {
+      listeners = this._listeners.get({
+        peerId: peer.peerKey,
+        payloadType
+      });
+      if (!listeners) {
+        listeners = /* @__PURE__ */ new Set();
+        this._listeners.set({
+          peerId: peer.peerKey,
+          payloadType
+        }, listeners);
+      }
+    }
+    listeners.add(onMessage);
+    return {
+      unsubscribe: async () => {
+        listeners.delete(onMessage);
+      }
+    };
+  }
+  async _encodeAndSend({ author, recipient, reliablePayload }) {
+    await this._signalManager.sendMessage({
+      author,
+      recipient,
+      payload: {
+        type_url: "dxos.mesh.messaging.ReliablePayload",
+        value: ReliablePayload.encode(reliablePayload, {
+          preserveAny: true
+        })
+      }
+    });
+  }
+  async _handleMessage(message) {
+    switch (message.payload.type_url) {
+      case "dxos.mesh.messaging.ReliablePayload": {
+        await this._handleReliablePayload(message);
+        break;
+      }
+      case "dxos.mesh.messaging.Acknowledgement": {
+        await this._handleAcknowledgement({
+          payload: message.payload
+        });
+        break;
+      }
+    }
+  }
+  async _handleReliablePayload({ author, recipient, payload }) {
+    (0, import_invariant.invariant)(payload.type_url === "dxos.mesh.messaging.ReliablePayload", void 0, {
+      F: __dxlog_file,
+      L: 240,
+      S: this,
+      A: [
+        "payload.type_url === 'dxos.mesh.messaging.ReliablePayload'",
+        ""
+      ]
+    });
+    const reliablePayload = ReliablePayload.decode(payload.value, {
+      preserveAny: true
+    });
+    (0, import_log.log)("handling message", {
+      messageId: reliablePayload.messageId
+    }, {
+      F: __dxlog_file,
+      L: 243,
+      S: this,
+      C: (f, a) => f(...a)
+    });
+    try {
+      await this._sendAcknowledgement({
+        author,
+        recipient,
+        messageId: reliablePayload.messageId
+      });
+    } catch (err) {
+      this._monitor.recordMessageAckFailed();
+      throw err;
+    }
+    if (this._receivedMessages.has(reliablePayload.messageId)) {
+      return;
+    }
+    this._receivedMessages.add(reliablePayload.messageId);
+    await this._callListeners({
+      author,
+      recipient,
+      payload: reliablePayload.payload
+    });
+  }
+  async _handleAcknowledgement({ payload }) {
+    (0, import_invariant.invariant)(payload.type_url === "dxos.mesh.messaging.Acknowledgement", void 0, {
+      F: __dxlog_file,
+      L: 271,
+      S: this,
+      A: [
+        "payload.type_url === 'dxos.mesh.messaging.Acknowledgement'",
+        ""
+      ]
+    });
+    this._onAckCallbacks.get(Acknowledgement.decode(payload.value).messageId)?.();
+  }
+  async _sendAcknowledgement({ author, recipient, messageId }) {
+    (0, import_log.log)("sending ACK", {
+      messageId,
+      from: recipient,
+      to: author
+    }, {
+      F: __dxlog_file,
+      L: 284,
+      S: this,
+      C: (f, a) => f(...a)
+    });
+    await this._signalManager.sendMessage({
+      author: recipient,
+      recipient: author,
+      payload: {
+        type_url: "dxos.mesh.messaging.Acknowledgement",
+        value: Acknowledgement.encode({
+          messageId
+        })
+      }
+    });
+  }
+  async _callListeners(message) {
+    {
+      (0, import_invariant.invariant)(message.recipient.peerKey, "Peer key is required", {
+        F: __dxlog_file,
+        L: 298,
+        S: this,
+        A: [
+          "message.recipient.peerKey",
+          "'Peer key is required'"
+        ]
+      });
+      const defaultListenerMap = this._defaultListeners.get(message.recipient.peerKey);
+      if (defaultListenerMap) {
+        for (const listener of defaultListenerMap) {
+          await listener(message);
+        }
+      }
+    }
+    {
+      const listenerMap = this._listeners.get({
+        peerId: message.recipient.peerKey,
+        payloadType: message.payload.type_url
+      });
+      if (listenerMap) {
+        for (const listener of listenerMap) {
+          await listener(message);
+        }
+      }
+    }
+  }
+  _performGc() {
+    const start = performance.now();
+    for (const key of this._toClear.keys()) {
+      this._receivedMessages.delete(key);
+    }
+    this._toClear.clear();
+    for (const key of this._receivedMessages.keys()) {
+      this._toClear.add(key);
+    }
+    const elapsed = performance.now() - start;
+    if (elapsed > 100) {
+      import_log.log.warn("GC took too long", {
+        elapsed
+      }, {
+        F: __dxlog_file,
+        L: 333,
+        S: this,
+        C: (f, a) => f(...a)
+      });
+    }
+  }
+};
+var PeerInfoHash = ({ peerKey }) => peerKey;
+var __dxlog_file2 = "/home/runner/work/dxos/dxos/packages/core/mesh/messaging/src/signal-manager/memory-signal-manager.ts";
+var MemorySignalManagerContext = class {
+  constructor() {
+    this.swarmEvent = new import_async2.Event();
+    this.swarms = new import_util2.ComplexMap(import_keys2.PublicKey.hash);
+    this.connections = new import_util2.ComplexMap(PeerInfoHash);
+  }
+};
+var MemorySignalManager = class {
+  constructor(_context) {
+    this._context = _context;
+    this.statusChanged = new import_async2.Event();
+    this.swarmEvent = new import_async2.Event();
+    this.onMessage = new import_async2.Event();
+    this._joinedSwarms = new import_util2.ComplexSet(({ topic, peer }) => topic.toHex() + peer.peerKey);
+    this._freezeTrigger = new import_async2.Trigger().wake();
+    this._ctx = new import_context2.Context(void 0, {
+      F: __dxlog_file2,
+      L: 51
+    });
+    this._ctx.onDispose(this._context.swarmEvent.on((data) => this.swarmEvent.emit(data)));
+  }
+  async open() {
+    if (!this._ctx.disposed) {
+      return;
+    }
+    this._ctx = new import_context2.Context(void 0, {
+      F: __dxlog_file2,
+      L: 60
+    });
+    this._ctx.onDispose(this._context.swarmEvent.on((data) => this.swarmEvent.emit(data)));
+    await Promise.all([
+      ...this._joinedSwarms.values()
+    ].map((value) => this.join(value)));
+  }
+  async close() {
+    if (this._ctx.disposed) {
+      return;
+    }
+    const joinedSwarmsCopy = new import_util2.ComplexSet(({ topic, peer }) => topic.toHex() + peer.peerKey, [
+      ...this._joinedSwarms.values()
+    ]);
+    await Promise.all([
+      ...this._joinedSwarms.values()
+    ].map((value) => this.leave(value)));
+    this._joinedSwarms = joinedSwarmsCopy;
+    await this._ctx.dispose();
+  }
+  getStatus() {
+    return [];
+  }
+  async join({ topic, peer }) {
+    (0, import_invariant2.invariant)(!this._ctx.disposed, "Closed", {
+      F: __dxlog_file2,
+      L: 89,
+      S: this,
+      A: [
+        "!this._ctx.disposed",
+        "'Closed'"
+      ]
+    });
+    this._joinedSwarms.add({
+      topic,
+      peer
+    });
+    if (!this._context.swarms.has(topic)) {
+      this._context.swarms.set(topic, new import_util2.ComplexSet(PeerInfoHash));
+    }
+    this._context.swarms.get(topic).add(peer);
+    this._context.swarmEvent.emit({
+      topic,
+      peerAvailable: {
+        peer,
+        since: /* @__PURE__ */ new Date()
+      }
+    });
+    for (const [topic2, peers] of this._context.swarms) {
+      Array.from(peers).forEach((peer2) => {
+        this.swarmEvent.emit({
+          topic: topic2,
+          peerAvailable: {
+            peer: peer2,
+            since: /* @__PURE__ */ new Date()
+          }
+        });
+      });
+    }
+  }
+  async leave({ topic, peer }) {
+    (0, import_invariant2.invariant)(!this._ctx.disposed, "Closed", {
+      F: __dxlog_file2,
+      L: 121,
+      S: this,
+      A: [
+        "!this._ctx.disposed",
+        "'Closed'"
+      ]
+    });
+    this._joinedSwarms.delete({
+      topic,
+      peer
+    });
+    if (!this._context.swarms.has(topic)) {
+      this._context.swarms.set(topic, new import_util2.ComplexSet(PeerInfoHash));
+    }
+    this._context.swarms.get(topic).delete(peer);
+    const swarmEvent = {
+      topic,
+      peerLeft: {
+        peer
+      }
+    };
+    this._context.swarmEvent.emit(swarmEvent);
+  }
+  async sendMessage({ author, recipient, payload }) {
+    (0, import_log2.log)("send message", {
+      author,
+      recipient,
+      ...dec(payload)
+    }, {
+      F: __dxlog_file2,
+      L: 142,
+      S: this,
+      C: (f, a) => f(...a)
+    });
+    (0, import_invariant2.invariant)(recipient, void 0, {
+      F: __dxlog_file2,
+      L: 144,
+      S: this,
+      A: [
+        "recipient",
+        ""
+      ]
+    });
+    (0, import_invariant2.invariant)(!this._ctx.disposed, "Closed", {
+      F: __dxlog_file2,
+      L: 145,
+      S: this,
+      A: [
+        "!this._ctx.disposed",
+        "'Closed'"
+      ]
+    });
+    await this._freezeTrigger.wait();
+    const remote = this._context.connections.get(recipient);
+    if (!remote) {
+      import_log2.log.warn("recipient is not subscribed for messages", {
+        author,
+        recipient
+      }, {
+        F: __dxlog_file2,
+        L: 151,
+        S: this,
+        C: (f, a) => f(...a)
+      });
+      return;
+    }
+    if (remote._ctx.disposed) {
+      import_log2.log.warn("recipient is disposed", {
+        author,
+        recipient
+      }, {
+        F: __dxlog_file2,
+        L: 156,
+        S: this,
+        C: (f, a) => f(...a)
+      });
+      return;
+    }
+    remote._freezeTrigger.wait().then(() => {
+      if (remote._ctx.disposed) {
+        import_log2.log.warn("recipient is disposed", {
+          author,
+          recipient
+        }, {
+          F: __dxlog_file2,
+          L: 164,
+          S: this,
+          C: (f, a) => f(...a)
+        });
+        return;
+      }
+      (0, import_log2.log)("receive message", {
+        author,
+        recipient,
+        ...dec(payload)
+      }, {
+        F: __dxlog_file2,
+        L: 168,
+        S: this,
+        C: (f, a) => f(...a)
+      });
+      remote.onMessage.emit({
+        author,
+        recipient,
+        payload
+      });
+    }).catch((err) => {
+      import_log2.log.error("error while waiting for freeze", {
+        err
+      }, {
+        F: __dxlog_file2,
+        L: 173,
+        S: this,
+        C: (f, a) => f(...a)
+      });
+    });
+  }
+  async subscribeMessages(peerInfo) {
+    (0, import_log2.log)("subscribing", {
+      peerInfo
+    }, {
+      F: __dxlog_file2,
+      L: 178,
+      S: this,
+      C: (f, a) => f(...a)
+    });
+    this._context.connections.set(peerInfo, this);
+  }
+  async unsubscribeMessages(peerInfo) {
+    (0, import_log2.log)("unsubscribing", {
+      peerInfo
+    }, {
+      F: __dxlog_file2,
+      L: 183,
+      S: this,
+      C: (f, a) => f(...a)
+    });
+    this._context.connections.delete(peerInfo);
+  }
+  freeze() {
+    this._freezeTrigger.reset();
+  }
+  unfreeze() {
+    this._freezeTrigger.wake();
+  }
+};
+var dec = (payload) => {
+  if (!payload.type_url.endsWith("ReliablePayload")) {
+    return {};
+  }
+  const relPayload = import_proto2.schema.getCodecForType("dxos.mesh.messaging.ReliablePayload").decode(payload.value);
+  if (typeof relPayload?.payload?.data === "object") {
+    return {
+      payload: Object.keys(relPayload?.payload?.data)[0],
+      sessionId: relPayload?.payload?.sessionId
+    };
+  }
+  return {};
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  MemorySignalManager,
+  MemorySignalManagerContext,
+  Messenger,
+  PeerInfoHash
+});
+//# sourceMappingURL=chunk-WDA7H2PO.cjs.map