fix(infra): avoid data loss (#6111)

packages/common/infra/src/blocksuite/migration/fixing.ts

@@ -13,6 +13,9 @@ export function fixWorkspaceVersion(rootDoc: YDoc) {
    * Blocksuite just set the value, do nothing else.
    */
   function doFix() {
+    if (meta.size === 0) {
+      return;
+    }
     const workspaceVersion = meta.get('workspaceVersion');
     if (typeof workspaceVersion !== 'number' || workspaceVersion < 2) {
       transact(

packages/common/infra/src/workspace/engine/sync/engine.ts

@@ -68,15 +68,17 @@ export class SyncEngine {
     this.onStatusChange.emit(s);
   }
   isRootDocLoaded = LiveData.from(
-    new Observable(observer => {
+    new Observable<boolean>(observer => {
       observer.next(
-        this.status.local
-          ? this.status.local.step > SyncPeerStep.LoadingRootDoc
-          : false
+        [this.status?.local, ...(this.status?.remotes ?? [])].some(
+          p => p?.rootDocLoaded === true
+        )
       );
       this.onStatusChange.on(status => {
         observer.next(
-          status.local ? status.local.step > SyncPeerStep.LoadingRootDoc : false
+          [status?.local, ...(status?.remotes ?? [])].some(
+            p => p?.rootDocLoaded === true
+          )
         );
       });
     }),

packages/common/infra/src/workspace/engine/sync/peer.ts

@@ -20,6 +20,7 @@ export interface SyncPeerStatus {
   pendingPullUpdates: number;
   pendingPushUpdates: number;
   lastError: string | null;
+  rootDocLoaded: boolean;
 }
 
 /**
@@ -56,6 +57,7 @@ export class SyncPeer {
     pendingPullUpdates: 0,
     pendingPushUpdates: 0,
     lastError: null,
+    rootDocLoaded: false,
   };
   onStatusChange = new Slot<SyncPeerStatus>();
   readonly abort = new AbortController();
@@ -122,6 +124,7 @@ export class SyncPeer {
           pendingPullUpdates: 0,
           pendingPushUpdates: 0,
           lastError: 'Retrying sync after 5 seconds',
+          rootDocLoaded: this.status.rootDocLoaded,
         };
         await Promise.race([
           new Promise<void>(resolve => {
@@ -295,6 +298,13 @@ export class SyncPeer {
       (await this.storage.pull(doc.guid, encodeStateVector(doc))) ?? {};
     throwIfAborted(abort);
 
+    if (docData !== undefined && doc.guid === this.rootDoc.guid) {
+      this.status = {
+        ...this.status,
+        rootDocLoaded: true,
+      };
+    }
+
     if (docData) {
       applyUpdate(doc, docData, 'load');
     }
@@ -400,6 +410,7 @@ export class SyncPeer {
       pendingPushUpdates:
         this.state.pushUpdatesQueue.length + (this.state.pushingUpdate ? 1 : 0),
       lastError,
+      rootDocLoaded: this.status.rootDocLoaded,
     };
   }