From 1f03ece1a4c0957c22334ccc384b5612f758ab4a Mon Sep 17 00:00:00 2001
From: Alex Yang <himself65@outlook.com>
Date: Tue, 22 Aug 2023 21:56:47 -0500
Subject: [PATCH] fix(storybook): lazy load app (#3905)

---
 apps/storybook/.storybook/preview.tsx | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/apps/storybook/.storybook/preview.tsx b/apps/storybook/.storybook/preview.tsx
index 3dcc7c59ed..c42a947695 100644
--- a/apps/storybook/.storybook/preview.tsx
+++ b/apps/storybook/.storybook/preview.tsx
@@ -9,9 +9,7 @@ import { AffineContext } from '@affine/component/context';
 import useSWR from 'swr';
 import type { Decorator } from '@storybook/react';
 import { createStore } from 'jotai/vanilla';
-import { setup } from '@affine/core/bootstrap/setup';
 import { _setCurrentStore } from '@toeverything/infra/atom';
-import { bootstrapPluginSystem } from '@affine/core/bootstrap/register-plugins';
 import { setupGlobal } from '@affine/env/global';
 
 setupGlobal();
@@ -54,6 +52,14 @@ const ThemeChange = () => {
 
 const storeMap = new Map<string, ReturnType<typeof createStore>>();
 
+const bootstrapPluginSystemPromise = import(
+  '@affine/core/bootstrap/register-plugins'
+).then(({ bootstrapPluginSystem }) => bootstrapPluginSystem);
+
+const setupPromise = import('@affine/core/bootstrap/setup').then(
+  ({ setup }) => setup
+);
+
 const withContextDecorator: Decorator = (Story, context) => {
   const { data: store } = useSWR(
     context.id,
@@ -64,7 +70,9 @@ const withContextDecorator: Decorator = (Story, context) => {
       localStorage.clear();
       const store = createStore();
       _setCurrentStore(store);
+      const setup = await setupPromise;
       await setup(store);
+      const bootstrapPluginSystem = await bootstrapPluginSystemPromise;
       await bootstrapPluginSystem(store);
       storeMap.set(context.id, store);
       return store;