luulam/AFFiNE
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix(storybook): lazy load app (#3905)

Browse Source
This commit is contained in:
Alex Yang
2023-08-22 21:56:47 -05:00
committed by GitHub
parent 2b71c69732
commit 1f03ece1a4
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
apps/storybook/.storybook/preview.tsx
View File

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