feat: support migration (#2852)

apps/web/src/bootstrap/index.ts

@@ -1,4 +1,18 @@
+import { migrateToSubdoc } from '@affine/env/blocksuite';
 import { config, setupGlobal } from '@affine/env/config';
+import type { LocalIndexedDBDownloadProvider } from '@affine/env/workspace';
+import { WorkspaceFlavour, WorkspaceVersion } from '@affine/env/workspace';
+import type { RootWorkspaceMetadata } from '@affine/workspace/atom';
+import { rootWorkspacesMetadataAtom } from '@affine/workspace/atom';
+import {
+  migrateLocalBlobStorage,
+  upgradeV1ToV2,
+} from '@affine/workspace/migration';
+import { createIndexedDBDownloadProvider } from '@affine/workspace/providers';
+import { assertExists } from '@blocksuite/global/utils';
+import { rootStore } from '@toeverything/plugin-infra/manager';
+
+import { WorkspaceAdapters } from '../adapters/workspace';
 
 setupGlobal();
 
@@ -32,3 +46,66 @@ if (!environment.isDesktop && !environment.isServer) {
     writable: false,
   });
 }
+
+rootStore.sub(rootWorkspacesMetadataAtom, () => {
+  const metadata = rootStore.get(rootWorkspacesMetadataAtom);
+  metadata.forEach(oldMeta => {
+    if (!oldMeta.version) {
+      const adapter = WorkspaceAdapters[oldMeta.flavour];
+      assertExists(adapter);
+      const upgrade = async () => {
+        const workspace = await adapter.CRUD.get(oldMeta.id);
+        if (!workspace) {
+          console.warn('cannot find workspace', oldMeta.id);
+          return;
+        }
+        if (workspace.flavour !== WorkspaceFlavour.LOCAL) {
+          console.warn('not supported');
+          return;
+        }
+        const doc = workspace.blockSuiteWorkspace.doc;
+        const provider = createIndexedDBDownloadProvider(workspace.id, doc, {
+          awareness: workspace.blockSuiteWorkspace.awarenessStore.awareness,
+        }) as LocalIndexedDBDownloadProvider;
+        provider.sync();
+        await provider.whenReady;
+        const newDoc = migrateToSubdoc(doc);
+        if (doc === newDoc) {
+          console.log('doc not changed');
+          rootStore.set(rootWorkspacesMetadataAtom, metadata =>
+            metadata.map(newMeta =>
+              newMeta.id === oldMeta.id
+                ? {
+                    ...newMeta,
+                    version: WorkspaceVersion.SubDoc,
+                  }
+                : newMeta
+            )
+          );
+          return;
+        }
+        const newWorkspace = upgradeV1ToV2(workspace);
+
+        const newId = await adapter.CRUD.create(
+          newWorkspace.blockSuiteWorkspace
+        );
+
+        await adapter.CRUD.delete(workspace as any);
+        await migrateLocalBlobStorage(workspace.id, newId);
+        rootStore.set(rootWorkspacesMetadataAtom, metadata => [
+          ...metadata
+            .map(newMeta => (newMeta.id === oldMeta.id ? null : newMeta))
+            .filter((meta): meta is RootWorkspaceMetadata => !!meta),
+          {
+            id: newId,
+            flavour: oldMeta.flavour,
+            version: WorkspaceVersion.SubDoc,
+          },
+        ]);
+      };
+
+      // create a new workspace and push it to metadata
+      upgrade().catch(console.error);
+    }
+  });
+});