@dxos/echo-pipeline 0.7.3 → 0.7.4-staging.99db212

package/dist/lib/browser/index.mjs

@@ -223,6 +223,7 @@ var CollectionSynchronizer = class extends Resource2 {
     * CollectionId -> State.
     */
     this._perCollectionStates = /* @__PURE__ */ new Map();
+    this._activeCollections = /* @__PURE__ */ new Set();
     this._connectedPeers = /* @__PURE__ */ new Set();
     this.remoteStateUpdated = new Event();
     this._sendCollectionState = params.sendCollectionState;
@@ -232,43 +233,58 @@ var CollectionSynchronizer = class extends Resource2 {
   async _open(ctx) {
     scheduleTaskInterval(this._ctx, async () => {
       for (const collectionId of this._perCollectionStates.keys()) {
-        this.refreshCollection(collectionId);
-        await asyncReturn();
+        if (this._activeCollections.has(collectionId)) {
+          this.refreshCollection(collectionId);
+          await 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);
     log2("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);
+    log2("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;
@@ -315,7 +331,7 @@ var CollectionSynchronizer = class extends Resource2 {
   * 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);
     }
@@ -330,19 +346,19 @@ var CollectionSynchronizer = class extends Resource2 {
       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 defaultMap(this._perCollectionStates, collectionId, () => ({
       localState: void 0,
       remoteStates: /* @__PURE__ */ new Map(),
@@ -353,9 +369,9 @@ var CollectionSynchronizer = class extends Resource2 {
   _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);
       }
     }
   }
@@ -1241,6 +1257,9 @@ var AutomergeHost = class extends Resource4 {
       documents
     });
   }
+  async clearLocalCollectionState(collectionId) {
+    this._collectionSynchronizer.clearLocalCollectionState(collectionId);
+  }
   _onCollectionStateQueried(collectionId, peerId) {
     this._collectionSynchronizer.onCollectionStateQueried(collectionId, peerId);
   }
@@ -1278,7 +1297,7 @@ var AutomergeHost = class extends Resource4 {
       count: toReplicate.length
     }, {
       F: __dxlog_file4,
-      L: 495,
+      L: 499,
       S: this,
       C: (f, a) => f(...a)
     });
@@ -1345,7 +1364,7 @@ var changeIsPresentInDoc = (doc, changeHash) => {
 var decodeCollectionState = (state) => {
   invariant3(typeof state === "object" && state !== null, "Invalid state", {
     F: __dxlog_file4,
-    L: 553,
+    L: 557,
     S: void 0,
     A: [
       "typeof state === 'object' && state !== null",
@@ -1514,12 +1533,12 @@ var logSendSync = (message) => {
 import { invariant as invariant5 } from "@dxos/invariant";
 import { SpaceId } from "@dxos/keys";
 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];
   invariant5(SpaceId.isValid(spaceId), void 0, {
     F: __dxlog_file6,
-    L: 13,
+    L: 15,
     S: void 0,
     A: [
       "SpaceId.isValid(spaceId)",
@@ -2103,6 +2122,7 @@ var DataServiceImpl = class {
     */
     this._subscriptions = /* @__PURE__ */ new Map();
     this._automergeHost = params.automergeHost;
+    this._spaceStateManager = params.spaceStateManager;
     this._updateIndexes = params.updateIndexes;
   }
   subscribe(request) {
@@ -2116,7 +2136,7 @@ var DataServiceImpl = class {
         ready();
       }).catch((err) => log7.catch(err, void 0, {
         F: __dxlog_file8,
-        L: 66,
+        L: 70,
         S: this,
         C: (f, a) => f(...a)
       }));
@@ -2127,7 +2147,7 @@ var DataServiceImpl = class {
     const synchronizer = this._subscriptions.get(request.subscriptionId);
     invariant7(synchronizer, "Subscription not found", {
       F: __dxlog_file8,
-      L: 73,
+      L: 77,
       S: this,
       A: [
         "synchronizer",
@@ -2148,7 +2168,7 @@ var DataServiceImpl = class {
     const synchronizer = this._subscriptions.get(request.subscriptionId);
     invariant7(synchronizer, "Subscription not found", {
       F: __dxlog_file8,
-      L: 88,
+      L: 92,
       S: this,
       A: [
         "synchronizer",
@@ -2190,18 +2210,29 @@ var DataServiceImpl = class {
   }
   subscribeSpaceSyncState(request) {
     return new Stream(({ ctx, next, ready }) => {
-      invariant7(SpaceId2.isValid(request.spaceId), void 0, {
+      const spaceId = request.spaceId;
+      invariant7(SpaceId2.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 UpdateScheduler2(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,
@@ -2827,6 +2858,9 @@ var SpaceStateManager = class extends Resource8 {
   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)) {
@@ -2846,7 +2880,7 @@ var SpaceStateManager = class extends Resource8 {
     this._rootBySpace.set(spaceId, root.handle.documentId);
     const ctx = new Context5(void 0, {
       F: __dxlog_file14,
-      L: 58
+      L: 62
     });
     this._perRootContext.set(root.handle.documentId, ctx);
     await root.handle.whenReady();
@@ -2857,7 +2891,7 @@ var SpaceStateManager = class extends Resource8 {
       ];
       if (!isEqual(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
@@ -2870,8 +2904,10 @@ var SpaceStateManager = class extends Resource8 {
   }
 };
 var SpaceDocumentListUpdatedEvent = class {
-  constructor(spaceId, documentIds) {
+  constructor(spaceId, spaceRootId, previousRootId, documentIds) {
     this.spaceId = spaceId;
+    this.spaceRootId = spaceRootId;
+    this.previousRootId = previousRootId;
     this.documentIds = documentIds;
   }
 };
@@ -2916,6 +2952,7 @@ var EchoHost = class extends Resource9 {
     });
     this._dataService = new DataServiceImpl({
       automergeHost: this._automergeHost,
+      spaceStateManager: this._spaceStateManager,
       updateIndexes: async () => {
         await this._indexer.updateIndexes();
       }
@@ -2979,7 +3016,11 @@ var EchoHost = class extends Resource9 {
     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) {
@@ -3018,7 +3059,7 @@ var EchoHost = class extends Resource9 {
   async createSpaceRoot(spaceKey) {
     invariant11(this._lifecycleState === LifecycleState4.OPEN, void 0, {
       F: __dxlog_file15,
-      L: 209,
+      L: 217,
       S: this,
       A: [
         "this._lifecycleState === LifecycleState.OPEN",
@@ -3046,7 +3087,7 @@ var EchoHost = class extends Resource9 {
   async openSpaceRoot(spaceId, automergeUrl) {
     invariant11(this._lifecycleState === LifecycleState4.OPEN, void 0, {
       F: __dxlog_file15,
-      L: 228,
+      L: 236,
       S: this,
       A: [
         "this._lifecycleState === LifecycleState.OPEN",
@@ -3072,10 +3113,6 @@ var EchoHost = class extends Resource9 {
   async removeReplicator(replicator) {
     await this._automergeHost.removeReplicator(replicator);
   }
-  async getSpaceSyncState(spaceId) {
-    const collectionId = deriveCollectionIdFromSpaceId(spaceId);
-    return this._automergeHost.getCollectionSyncState(collectionId);
-  }
 };
 
 // packages/core/echo/echo-pipeline/src/edge/echo-edge-replicator.ts
@@ -3508,9 +3545,11 @@ export {
   QueryServiceImpl,
   QueryState,
   Space,
+  SpaceDocumentListUpdatedEvent,
   SpaceManager,
   SpaceProtocol,
   SpaceProtocolSession,
+  SpaceStateManager,
   TimeframeClock,
   codec,
   createIdFromSpaceKey,