feat: add bootstrap (#3299)

apps/core/src/bootstrap/before-app.ts

@@ -0,0 +1,132 @@
+import { migrateToSubdoc } from '@affine/env/blocksuite';
+import { setupGlobal } from '@affine/env/global';
+import type {
+  LocalIndexedDBDownloadProvider,
+  WorkspaceAdapter,
+} from '@affine/env/workspace';
+import { WorkspaceFlavour, WorkspaceVersion } from '@affine/env/workspace';
+import type { RootWorkspaceMetadata } from '@affine/workspace/atom';
+import {
+  type RootWorkspaceMetadataV2,
+  rootWorkspacesMetadataAtom,
+  workspaceAdaptersAtom,
+} 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();
+
+rootStore.set(
+  workspaceAdaptersAtom,
+  WorkspaceAdapters as Record<
+    WorkspaceFlavour,
+    WorkspaceAdapter<WorkspaceFlavour>
+  >
+);
+
+const value = localStorage.getItem('jotai-workspaces');
+if (value) {
+  try {
+    const metadata = JSON.parse(value) as RootWorkspaceMetadata[];
+    const promises: Promise<void>[] = [];
+    const newMetadata = [...metadata];
+    metadata.forEach(oldMeta => {
+      if (!('version' in oldMeta)) {
+        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');
+            return;
+          }
+          const newWorkspace = upgradeV1ToV2(workspace);
+
+          const newId = await adapter.CRUD.create(
+            newWorkspace.blockSuiteWorkspace
+          );
+
+          await adapter.CRUD.delete(workspace as any);
+          console.log('migrated', oldMeta.id, newId);
+          const index = newMetadata.findIndex(meta => meta.id === oldMeta.id);
+          newMetadata[index] = {
+            ...oldMeta,
+            id: newId,
+            version: WorkspaceVersion.SubDoc,
+          };
+          await migrateLocalBlobStorage(workspace.id, newId);
+        };
+
+        // create a new workspace and push it to metadata
+        promises.push(upgrade());
+      }
+    });
+
+    await Promise.all(promises)
+      .then(() => {
+        console.log('migration done');
+      })
+      .catch(() => {
+        console.error('migration failed');
+      })
+      .finally(() => {
+        localStorage.setItem('jotai-workspaces', JSON.stringify(newMetadata));
+        window.dispatchEvent(new CustomEvent('migration-done'));
+        window.$migrationDone = true;
+      });
+  } catch (e) {
+    console.error('error when migrating data', e);
+  }
+}
+
+const createFirst = (): RootWorkspaceMetadataV2[] => {
+  const Plugins = Object.values(WorkspaceAdapters).sort(
+    (a, b) => a.loadPriority - b.loadPriority
+  );
+
+  return Plugins.flatMap(Plugin => {
+    return Plugin.Events['app:init']?.().map(
+      id =>
+        ({
+          id,
+          flavour: Plugin.flavour,
+          // new workspace should all support sub-doc feature
+          version: WorkspaceVersion.SubDoc,
+        }) satisfies RootWorkspaceMetadataV2
+    );
+  }).filter((ids): ids is RootWorkspaceMetadataV2 => !!ids);
+};
+
+await rootStore
+  .get(rootWorkspacesMetadataAtom)
+  .then(meta => {
+    if (meta.length === 0 && localStorage.getItem('is-first-open') === null) {
+      const result = createFirst();
+      console.info('create first workspace', result);
+      localStorage.setItem('is-first-open', 'false');
+      rootStore.set(rootWorkspacesMetadataAtom, result).catch(console.error);
+    }
+  })
+  .catch(console.error);

apps/core/src/bootstrap/register-plugins.ts

@@ -0,0 +1,3 @@
+import('@affine/bookmark-block');
+
+export {};