Fix stories (#11585)
<img width="1512" alt="image" src="https://github.com/user-attachments/assets/f77a006c-9538-4734-b80e-b58b9f5e7835" />
This commit is contained in:
@ -3,13 +3,40 @@ import { useSetRecoilState } from 'recoil';
|
||||
|
||||
import { prefetchIsLoadedFamilyState } from '@/prefetch/states/prefetchIsLoadedFamilyState';
|
||||
import { PrefetchKey } from '@/prefetch/types/PrefetchKey';
|
||||
import { useEffect, useState } from 'react';
|
||||
|
||||
export const PrefetchLoadingDecorator: Decorator = (Story) => {
|
||||
const setAreFavoritesPrefetched = useSetRecoilState(
|
||||
export const PrefetchLoadingDecorator: Decorator = (Story, context) => {
|
||||
const { parameters } = context;
|
||||
|
||||
const prefetchLoadingSetDelay = parameters.prefetchLoadingSetDelay ?? 0;
|
||||
|
||||
const setAreFavoritesFoldersPrefetched = useSetRecoilState(
|
||||
prefetchIsLoadedFamilyState(PrefetchKey.AllFavoritesFolders),
|
||||
);
|
||||
|
||||
setAreFavoritesPrefetched(false);
|
||||
const setAreFavoritesPrefetched = useSetRecoilState(
|
||||
prefetchIsLoadedFamilyState(PrefetchKey.AllFavorites),
|
||||
);
|
||||
|
||||
const [isInitialized, setIsInitialized] = useState(false);
|
||||
|
||||
useEffect(() => {
|
||||
if (isInitialized) {
|
||||
return;
|
||||
}
|
||||
|
||||
setIsInitialized(true);
|
||||
|
||||
setTimeout(() => {
|
||||
setAreFavoritesPrefetched(false);
|
||||
setAreFavoritesFoldersPrefetched(false);
|
||||
}, prefetchLoadingSetDelay);
|
||||
}, [
|
||||
isInitialized,
|
||||
prefetchLoadingSetDelay,
|
||||
setAreFavoritesFoldersPrefetched,
|
||||
setAreFavoritesPrefetched,
|
||||
]);
|
||||
|
||||
return <Story />;
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user