42 lines
1.6 KiB
JavaScript
42 lines
1.6 KiB
JavaScript
|
import { useRef, useEffect } from 'react';
|
||
|
|
import { useInstantLayoutTransition } from '../projection/use-instant-layout-transition.mjs';
|
||
|
|
import { useForceUpdate } from './use-force-update.mjs';
|
||
|
|
import { instantAnimationState } from './use-instant-transition-state.mjs';
|
||
|
|
import { frame } from '../frameloop/frame.mjs';
|
||
|
|
|
||
|
|
function useInstantTransition() {
|
||
|
|
const [forceUpdate, forcedRenderCount] = useForceUpdate();
|
||
|
|
const startInstantLayoutTransition = useInstantLayoutTransition();
|
||
|
|
const unlockOnFrameRef = useRef();
|
||
|
|
useEffect(() => {
|
||
|
|
/**
|
||
|
|
* Unblock after two animation frames, otherwise this will unblock too soon.
|
||
|
|
*/
|
||
|
|
frame.postRender(() => frame.postRender(() => {
|
||
|
|
/**
|
||
|
|
* If the callback has been called again after the effect
|
||
|
|
* triggered this 2 frame delay, don't unblock animations. This
|
||
|
|
* prevents the previous effect from unblocking the current
|
||
|
|
* instant transition too soon. This becomes more likely when
|
||
|
|
* used in conjunction with React.startTransition().
|
||
|
|
*/
|
||
|
|
if (forcedRenderCount !== unlockOnFrameRef.current)
|
||
|
|
return;
|
||
|
|
instantAnimationState.current = false;
|
||
|
|
}));
|
||
|
|
}, [forcedRenderCount]);
|
||
|
|
return (callback) => {
|
||
|
|
startInstantLayoutTransition(() => {
|
||
|
|
instantAnimationState.current = true;
|
||
|
|
forceUpdate();
|
||
|
|
callback();
|
||
|
|
unlockOnFrameRef.current = forcedRenderCount + 1;
|
||
|
|
});
|
||
|
|
};
|
||
|
|
}
|
||
|
|
function disableInstantTransitions() {
|
||
|
|
instantAnimationState.current = false;
|
||
|
|
}
|
||
|
|
|
||
|
|
export { disableInstantTransitions, useInstantTransition };
|