@dxos/echo-pipeline 0.7.4-staging.f7e8224 → 0.7.4

package/dist/lib/node/index.cjs

@@ -48,9 +48,11 @@ __export(node_exports, {
   QueryServiceImpl: () => QueryServiceImpl,
   QueryState: () => QueryState,
   Space: () => import_chunk_MACQJ2EP.Space,
+  SpaceDocumentListUpdatedEvent: () => SpaceDocumentListUpdatedEvent,
   SpaceManager: () => import_chunk_MACQJ2EP.SpaceManager,
   SpaceProtocol: () => import_chunk_MACQJ2EP.SpaceProtocol,
   SpaceProtocolSession: () => import_chunk_MACQJ2EP.SpaceProtocolSession,
+  SpaceStateManager: () => SpaceStateManager,
   TimeframeClock: () => import_chunk_MACQJ2EP.TimeframeClock,
   codec: () => import_chunk_MACQJ2EP.codec,
   createIdFromSpaceKey: () => import_chunk_MACQJ2EP.createIdFromSpaceKey,
@@ -329,6 +331,7 @@ var CollectionSynchronizer = class extends import_context3.Resource {
   constructor(params) {
     super();
     this._perCollectionStates = /* @__PURE__ */ new Map();
+    this._activeCollections = /* @__PURE__ */ new Set();
     this._connectedPeers = /* @__PURE__ */ new Set();
     this.remoteStateUpdated = new import_async4.Event();
     this._sendCollectionState = params.sendCollectionState;
@@ -338,43 +341,58 @@ var CollectionSynchronizer = class extends import_context3.Resource {
   async _open(ctx) {
     (0, import_async4.scheduleTaskInterval)(this._ctx, async () => {
       for (const collectionId of this._perCollectionStates.keys()) {
-        this.refreshCollection(collectionId);
-        await (0, import_async4.asyncReturn)();
+        if (this._activeCollections.has(collectionId)) {
+          this.refreshCollection(collectionId);
+          await (0, import_async4.asyncReturn)();
+        }
       }
     }, POLL_INTERVAL);
   }
   getRegisteredCollectionIds() {
     return [
-      ...this._perCollectionStates.keys()
+      ...this._activeCollections
     ];
   }
   getLocalCollectionState(collectionId) {
-    return this._getPerCollectionState(collectionId).localState;
+    return this._perCollectionStates.get(collectionId)?.localState;
   }
   setLocalCollectionState(collectionId, state) {
+    this._activeCollections.add(collectionId);
     (0, import_log4.log)("setLocalCollectionState", {
       collectionId,
       state
     }, {
       F: __dxlog_file2,
-      L: 68,
+      L: 73,
       S: this,
       C: (f, a) => f(...a)
     });
-    this._getPerCollectionState(collectionId).localState = state;
+    this._getOrCreatePerCollectionState(collectionId).localState = state;
     queueMicrotask(async () => {
-      if (!this._ctx.disposed) {
+      if (!this._ctx.disposed && this._activeCollections.has(collectionId)) {
         this._refreshInterestedPeers(collectionId);
         this.refreshCollection(collectionId);
       }
     });
   }
+  clearLocalCollectionState(collectionId) {
+    this._activeCollections.delete(collectionId);
+    this._perCollectionStates.delete(collectionId);
+    (0, import_log4.log)("clearLocalCollectionState", {
+      collectionId
+    }, {
+      F: __dxlog_file2,
+      L: 87,
+      S: this,
+      C: (f, a) => f(...a)
+    });
+  }
   getRemoteCollectionStates(collectionId) {
-    return this._getPerCollectionState(collectionId).remoteStates;
+    return this._getOrCreatePerCollectionState(collectionId).remoteStates;
   }
   refreshCollection(collectionId) {
     let scheduleAnotherRefresh = false;
-    const state = this._getPerCollectionState(collectionId);
+    const state = this._getOrCreatePerCollectionState(collectionId);
     for (const peerId of this._connectedPeers) {
       if (state.interestedPeers.has(peerId)) {
         const lastQueried = state.lastQueried.get(peerId) ?? 0;
@@ -421,7 +439,7 @@ var CollectionSynchronizer = class extends import_context3.Resource {
   * Callback when a peer queries the state of a collection.
   */
   onCollectionStateQueried(collectionId, peerId) {
-    const perCollectionState = this._getPerCollectionState(collectionId);
+    const perCollectionState = this._getOrCreatePerCollectionState(collectionId);
     if (perCollectionState.localState) {
       this._sendCollectionState(collectionId, peerId, perCollectionState.localState);
     }
@@ -436,19 +454,19 @@ var CollectionSynchronizer = class extends import_context3.Resource {
       state
     }, {
       F: __dxlog_file2,
-      L: 148,
+      L: 159,
       S: this,
       C: (f, a) => f(...a)
     });
     validateCollectionState(state);
-    const perCollectionState = this._getPerCollectionState(collectionId);
+    const perCollectionState = this._getOrCreatePerCollectionState(collectionId);
     perCollectionState.remoteStates.set(peerId, state);
     this.remoteStateUpdated.emit({
       peerId,
       collectionId
     });
   }
-  _getPerCollectionState(collectionId) {
+  _getOrCreatePerCollectionState(collectionId) {
     return (0, import_util.defaultMap)(this._perCollectionStates, collectionId, () => ({
       localState: void 0,
       remoteStates: /* @__PURE__ */ new Map(),
@@ -459,9 +477,9 @@ var CollectionSynchronizer = class extends import_context3.Resource {
   _refreshInterestedPeers(collectionId) {
     for (const peerId of this._connectedPeers) {
       if (this._shouldSyncCollection(collectionId, peerId)) {
-        this._getPerCollectionState(collectionId).interestedPeers.add(peerId);
+        this._getOrCreatePerCollectionState(collectionId).interestedPeers.add(peerId);
       } else {
-        this._getPerCollectionState(collectionId).interestedPeers.delete(peerId);
+        this._getOrCreatePerCollectionState(collectionId).interestedPeers.delete(peerId);
       }
     }
   }
@@ -1326,6 +1344,9 @@ var AutomergeHost = class extends import_context2.Resource {
       documents
     });
   }
+  async clearLocalCollectionState(collectionId) {
+    this._collectionSynchronizer.clearLocalCollectionState(collectionId);
+  }
   _onCollectionStateQueried(collectionId, peerId) {
     this._collectionSynchronizer.onCollectionStateQueried(collectionId, peerId);
   }
@@ -1363,7 +1384,7 @@ var AutomergeHost = class extends import_context2.Resource {
       count: toReplicate.length
     }, {
       F: __dxlog_file4,
-      L: 495,
+      L: 499,
       S: this,
       C: (f, a) => f(...a)
     });
@@ -1430,7 +1451,7 @@ var changeIsPresentInDoc = (doc, changeHash) => {
 var decodeCollectionState = (state) => {
   (0, import_invariant3.invariant)(typeof state === "object" && state !== null, "Invalid state", {
     F: __dxlog_file4,
-    L: 553,
+    L: 557,
     S: void 0,
     A: [
       "typeof state === 'object' && state !== null",
@@ -1581,12 +1602,12 @@ var logSendSync = (message) => {
   });
 };
 var __dxlog_file6 = "/home/runner/work/dxos/dxos/packages/core/echo/echo-pipeline/src/automerge/space-collection.ts";
-var deriveCollectionIdFromSpaceId = (spaceId) => `space:${spaceId}`;
+var deriveCollectionIdFromSpaceId = (spaceId, rootDocumentId) => rootDocumentId ? `space:${spaceId}:${rootDocumentId}` : `space:${spaceId}`;
 var getSpaceIdFromCollectionId = (collectionId) => {
-  const spaceId = collectionId.replace(/^space:/, "");
+  const spaceId = collectionId.split(":")[1];
   (0, import_invariant7.invariant)(import_keys4.SpaceId.isValid(spaceId), void 0, {
     F: __dxlog_file6,
-    L: 13,
+    L: 15,
     S: void 0,
     A: [
       "SpaceId.isValid(spaceId)",
@@ -2152,6 +2173,7 @@ var DataServiceImpl = class {
   constructor(params) {
     this._subscriptions = /* @__PURE__ */ new Map();
     this._automergeHost = params.automergeHost;
+    this._spaceStateManager = params.spaceStateManager;
     this._updateIndexes = params.updateIndexes;
   }
   subscribe(request) {
@@ -2165,7 +2187,7 @@ var DataServiceImpl = class {
         ready();
       }).catch((err) => import_log.log.catch(err, void 0, {
         F: __dxlog_file8,
-        L: 66,
+        L: 70,
         S: this,
         C: (f, a) => f(...a)
       }));
@@ -2176,7 +2198,7 @@ var DataServiceImpl = class {
     const synchronizer = this._subscriptions.get(request.subscriptionId);
     (0, import_invariant.invariant)(synchronizer, "Subscription not found", {
       F: __dxlog_file8,
-      L: 73,
+      L: 77,
       S: this,
       A: [
         "synchronizer",
@@ -2197,7 +2219,7 @@ var DataServiceImpl = class {
     const synchronizer = this._subscriptions.get(request.subscriptionId);
     (0, import_invariant.invariant)(synchronizer, "Subscription not found", {
       F: __dxlog_file8,
-      L: 88,
+      L: 92,
       S: this,
       A: [
         "synchronizer",
@@ -2239,18 +2261,29 @@ var DataServiceImpl = class {
   }
   subscribeSpaceSyncState(request) {
     return new import_stream.Stream(({ ctx, next, ready }) => {
-      (0, import_invariant.invariant)(import_keys.SpaceId.isValid(request.spaceId), void 0, {
+      const spaceId = request.spaceId;
+      (0, import_invariant.invariant)(import_keys.SpaceId.isValid(spaceId), void 0, {
         F: __dxlog_file8,
-        L: 127,
+        L: 132,
         S: this,
         A: [
-          "SpaceId.isValid(request.spaceId)",
+          "SpaceId.isValid(spaceId)",
           ""
         ]
       });
-      const collectionId = deriveCollectionIdFromSpaceId(request.spaceId);
+      const rootDocumentId = this._spaceStateManager.getSpaceRootDocumentId(spaceId);
+      let collectionId = rootDocumentId && deriveCollectionIdFromSpaceId(spaceId, rootDocumentId);
+      this._spaceStateManager.spaceDocumentListUpdated.on(ctx, (event) => {
+        const newId = deriveCollectionIdFromSpaceId(spaceId, event.spaceRootId);
+        if (newId !== collectionId) {
+          collectionId = newId;
+          scheduler.trigger();
+        }
+      });
       const scheduler = new import_async.UpdateScheduler(ctx, async () => {
-        const state = await this._automergeHost.getCollectionSyncState(collectionId);
+        const state = collectionId ? await this._automergeHost.getCollectionSyncState(collectionId) : {
+          peers: []
+        };
         next({
           peers: state.peers.map((peer) => ({
             peerId: peer.peerId,
@@ -2821,6 +2854,9 @@ var SpaceStateManager = class extends import_context11.Resource {
   getRootByDocumentId(documentId) {
     return this._roots.get(documentId);
   }
+  getSpaceRootDocumentId(spaceId) {
+    return this._rootBySpace.get(spaceId);
+  }
   async assignRootToSpace(spaceId, handle) {
     let root;
     if (this._roots.has(handle.documentId)) {
@@ -2840,7 +2876,7 @@ var SpaceStateManager = class extends import_context11.Resource {
     this._rootBySpace.set(spaceId, root.handle.documentId);
     const ctx = new import_context11.Context(void 0, {
       F: __dxlog_file14,
-      L: 58
+      L: 62
     });
     this._perRootContext.set(root.handle.documentId, ctx);
     await root.handle.whenReady();
@@ -2851,7 +2887,7 @@ var SpaceStateManager = class extends import_context11.Resource {
       ];
       if (!(0, import_lodash.default)(documentIds, this._lastSpaceDocumentList.get(spaceId))) {
         this._lastSpaceDocumentList.set(spaceId, documentIds);
-        this.spaceDocumentListUpdated.emit(new SpaceDocumentListUpdatedEvent(spaceId, documentIds));
+        this.spaceDocumentListUpdated.emit(new SpaceDocumentListUpdatedEvent(spaceId, root.documentId, prevRootId, documentIds));
       }
     }, {
       maxFrequency: 50
@@ -2864,8 +2900,10 @@ var SpaceStateManager = class extends import_context11.Resource {
   }
 };
 var SpaceDocumentListUpdatedEvent = class {
-  constructor(spaceId, documentIds) {
+  constructor(spaceId, spaceRootId, previousRootId, documentIds) {
     this.spaceId = spaceId;
+    this.spaceRootId = spaceRootId;
+    this.previousRootId = previousRootId;
     this.documentIds = documentIds;
   }
 };
@@ -2908,6 +2946,7 @@ var EchoHost = class extends import_context7.Resource {
     });
     this._dataService = new DataServiceImpl({
       automergeHost: this._automergeHost,
+      spaceStateManager: this._spaceStateManager,
       updateIndexes: async () => {
         await this._indexer.updateIndexes();
       }
@@ -2971,7 +3010,11 @@ var EchoHost = class extends import_context7.Resource {
     await this._queryService.open(ctx);
     await this._spaceStateManager.open(ctx);
     this._spaceStateManager.spaceDocumentListUpdated.on(this._ctx, (e) => {
+      if (e.previousRootId) {
+        void this._automergeHost.clearLocalCollectionState(deriveCollectionIdFromSpaceId(e.spaceId, e.previousRootId));
+      }
       void this._automergeHost.updateLocalCollectionState(deriveCollectionIdFromSpaceId(e.spaceId), e.documentIds);
+      void this._automergeHost.updateLocalCollectionState(deriveCollectionIdFromSpaceId(e.spaceId, e.spaceRootId), e.documentIds);
     });
   }
   async _close(ctx) {
@@ -3010,7 +3053,7 @@ var EchoHost = class extends import_context7.Resource {
   async createSpaceRoot(spaceKey) {
     (0, import_invariant8.invariant)(this._lifecycleState === import_context7.LifecycleState.OPEN, void 0, {
       F: __dxlog_file15,
-      L: 209,
+      L: 217,
       S: this,
       A: [
         "this._lifecycleState === LifecycleState.OPEN",
@@ -3038,7 +3081,7 @@ var EchoHost = class extends import_context7.Resource {
   async openSpaceRoot(spaceId, automergeUrl) {
     (0, import_invariant8.invariant)(this._lifecycleState === import_context7.LifecycleState.OPEN, void 0, {
       F: __dxlog_file15,
-      L: 228,
+      L: 236,
       S: this,
       A: [
         "this._lifecycleState === LifecycleState.OPEN",
@@ -3064,10 +3107,6 @@ var EchoHost = class extends import_context7.Resource {
   async removeReplicator(replicator) {
     await this._automergeHost.removeReplicator(replicator);
   }
-  async getSpaceSyncState(spaceId) {
-    const collectionId = deriveCollectionIdFromSpaceId(spaceId);
-    return this._automergeHost.getCollectionSyncState(collectionId);
-  }
 };
 function _using_ctx() {
   var _disposeSuppressedError = typeof SuppressedError === "function" ? SuppressedError : function(error, suppressed) {
@@ -3485,9 +3524,11 @@ var findInlineObjectOfType = (spaceDoc, typename) => {
   QueryServiceImpl,
   QueryState,
   Space,
+  SpaceDocumentListUpdatedEvent,
   SpaceManager,
   SpaceProtocol,
   SpaceProtocolSession,
+  SpaceStateManager,
   TimeframeClock,
   codec,
   createIdFromSpaceKey,