feat: add bootstrap (#3299)

apps/core/package.json

@@ -10,6 +10,7 @@
   },
   "dependencies": {
     "@affine-test/fixtures": "workspace:*",
+    "@affine/bookmark-block": "workspace:*",
     "@affine/component": "workspace:*",
     "@affine/debug": "workspace:*",
     "@affine/env": "workspace:*",

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 {};

apps/core/src/index.tsx

@@ -1,106 +1,14 @@
-import { migrateToSubdoc } from '@affine/env/blocksuite';
-import { setupGlobal } from '@affine/env/global';
-import type { LocalIndexedDBDownloadProvider } from '@affine/env/workspace';
-import { WorkspaceFlavour, WorkspaceVersion } from '@affine/env/workspace';
-import { type WorkspaceAdapter } from '@affine/env/workspace';
-import type { RootWorkspaceMetadata } from '@affine/workspace/atom';
-import { 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 { createRoot } from 'react-dom/client';
 
-import { WorkspaceAdapters } from './adapters/workspace';
-
-// bootstrap
-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());
-      }
-    });
-
-    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);
-  }
-}
-
-// start app
-import('./app').then(({ App }) => {
+async function main() {
+  await import('./bootstrap/before-app');
+  const { App } = await import('./app');
   const root = document.getElementById('app');
   assertExists(root);
 
   createRoot(root).render(<App />);
-});
+  await import('./bootstrap/register-plugins');
+}
+
+await main();

apps/core/src/pages/index.tsx

@@ -1,15 +1,10 @@
 import { DebugLogger } from '@affine/debug';
-import { WorkspaceSubPath, WorkspaceVersion } from '@affine/env/workspace';
-import {
-  type RootWorkspaceMetadataV2,
-  rootWorkspacesMetadataAtom,
-} from '@affine/workspace/atom';
+import { WorkspaceSubPath } from '@affine/env/workspace';
+import { rootWorkspacesMetadataAtom } from '@affine/workspace/atom';
 import { getWorkspace } from '@toeverything/plugin-infra/__internal__/workspace';
-import { rootStore } from '@toeverything/plugin-infra/manager';
 import { useAtomValue } from 'jotai';
 import { lazy, useEffect, useRef } from 'react';
 
-import { WorkspaceAdapters } from '../adapters/workspace';
 import { RouteLogic, useNavigateHelper } from '../hooks/use-navigate-helper';
 import { useWorkspace } from '../hooks/use-workspace';
 
@@ -23,36 +18,6 @@ type WorkspaceLoaderProps = {
   id: string;
 };
 
-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);
-};
-
-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);
-
 const WorkspaceLoader = (props: WorkspaceLoaderProps): null => {
   useWorkspace(props.id);
   return null;