fix: create first workspace logic (#3241)

(cherry picked from commit c49cf1c53c)

apps/web/src/bootstrap/index.ts

@@ -6,7 +6,11 @@ import type {
 } from '@affine/env/workspace';
 import { WorkspaceFlavour, WorkspaceVersion } from '@affine/env/workspace';
 import type { RootWorkspaceMetadata } from '@affine/workspace/atom';
-import { workspaceAdaptersAtom } from '@affine/workspace/atom';
+import {
+  type RootWorkspaceMetadataV2,
+  rootWorkspacesMetadataAtom,
+  workspaceAdaptersAtom,
+} from '@affine/workspace/atom';
 import {
   migrateLocalBlobStorage,
   upgradeV1ToV2,
@@ -77,6 +81,36 @@ if (!environment.isServer) {
     }
     return;
   };
+
+  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);
 }
 
 if (runtimeConfig.enablePlugin && !environment.isServer) {

apps/web/src/pages/index.tsx

@@ -1,15 +1,14 @@
 import { WorkspaceFallback } from '@affine/component/workspace';
 import { DebugLogger } from '@affine/debug';
-import { WorkspaceSubPath, WorkspaceVersion } from '@affine/env/workspace';
-import type { RootWorkspaceMetadataV2 } from '@affine/workspace/atom';
+import { WorkspaceSubPath } from '@affine/env/workspace';
 import { rootWorkspacesMetadataAtom } from '@affine/workspace/atom';
+import { NoSsr } from '@mui/material';
 import { getWorkspace } from '@toeverything/plugin-infra/__internal__/workspace';
-import { useAtom } from 'jotai';
+import { useAtomValue } from 'jotai';
 import type { NextPage } from 'next';
 import { useRouter } from 'next/router';
 import { Suspense, useEffect } from 'react';
 
-import { WorkspaceAdapters } from '../adapters/workspace';
 import { RouteLogic, useRouterHelper } from '../hooks/use-router-helper';
 import { useWorkspace } from '../hooks/use-workspace';
 import { useAppHelper } from '../hooks/use-workspaces';
@@ -29,46 +28,9 @@ const WorkspaceLoader = (props: AllWorkspaceLoaderProps): null => {
 const IndexPageInner = () => {
   const router = useRouter();
   const { jumpToPage, jumpToSubPath } = useRouterHelper(router);
-  const [meta, setMeta] = useAtom(rootWorkspacesMetadataAtom);
+  const meta = useAtomValue(rootWorkspacesMetadataAtom);
   const helper = useAppHelper();
 
-  useEffect(() => {
-    const abortController = new AbortController();
-    const signal = abortController.signal;
-    const createFirst = (): RootWorkspaceMetadataV2[] => {
-      if (signal.aborted) {
-        return [];
-      }
-
-      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);
-    };
-
-    if (meta.length === 0 && localStorage.getItem('is-first-open') === null) {
-      meta.push(...createFirst());
-      console.info('create first workspace', meta);
-      localStorage.setItem('is-first-open', 'false');
-      setMeta(meta).catch(console.error);
-    }
-
-    return () => {
-      abortController.abort();
-    };
-  }, [meta, setMeta]);
-
   useEffect(() => {
     if (!router.isReady) {
       return;
@@ -136,9 +98,11 @@ const IndexPageInner = () => {
 
 const IndexPage: NextPage = () => {
   return (
-    <Suspense fallback={<WorkspaceFallback />}>
-      <IndexPageInner />
-    </Suspense>
+    <NoSsr>
+      <Suspense fallback={<WorkspaceFallback />}>
+        <IndexPageInner />
+      </Suspense>
+    </NoSsr>
   );
 };