1
This commit is contained in:
+53
@@ -0,0 +1,53 @@
|
||||
import useState from "@rc-component/util/es/hooks/useState";
|
||||
import * as React from 'react';
|
||||
import { STEP_ACTIVATED, STEP_ACTIVE, STEP_NONE, STEP_PREPARE, STEP_PREPARED, STEP_START } from "../interface";
|
||||
import useIsomorphicLayoutEffect from "./useIsomorphicLayoutEffect";
|
||||
import useNextFrame from "./useNextFrame";
|
||||
const FULL_STEP_QUEUE = [STEP_PREPARE, STEP_START, STEP_ACTIVE, STEP_ACTIVATED];
|
||||
const SIMPLE_STEP_QUEUE = [STEP_PREPARE, STEP_PREPARED];
|
||||
|
||||
/** Skip current step */
|
||||
export const SkipStep = false;
|
||||
/** Current step should be update in */
|
||||
export const DoStep = true;
|
||||
export function isActive(step) {
|
||||
return step === STEP_ACTIVE || step === STEP_ACTIVATED;
|
||||
}
|
||||
export default ((status, prepareOnly, callback) => {
|
||||
const [step, setStep] = useState(STEP_NONE);
|
||||
const [nextFrame, cancelNextFrame] = useNextFrame();
|
||||
function startQueue() {
|
||||
setStep(STEP_PREPARE, true);
|
||||
}
|
||||
const STEP_QUEUE = prepareOnly ? SIMPLE_STEP_QUEUE : FULL_STEP_QUEUE;
|
||||
useIsomorphicLayoutEffect(() => {
|
||||
if (step !== STEP_NONE && step !== STEP_ACTIVATED) {
|
||||
const index = STEP_QUEUE.indexOf(step);
|
||||
const nextStep = STEP_QUEUE[index + 1];
|
||||
const result = callback(step);
|
||||
if (result === SkipStep) {
|
||||
// Skip when no needed
|
||||
setStep(nextStep, true);
|
||||
} else if (nextStep) {
|
||||
// Do as frame for step update
|
||||
nextFrame(info => {
|
||||
function doNext() {
|
||||
// Skip since current queue is ood
|
||||
if (info.isCanceled()) return;
|
||||
setStep(nextStep, true);
|
||||
}
|
||||
if (result === true) {
|
||||
doNext();
|
||||
} else {
|
||||
// Only promise should be async
|
||||
Promise.resolve(result).then(doNext);
|
||||
}
|
||||
});
|
||||
}
|
||||
}
|
||||
}, [status, step]);
|
||||
React.useEffect(() => () => {
|
||||
cancelNextFrame();
|
||||
}, []);
|
||||
return [startQueue, step];
|
||||
});
|
||||
Reference in New Issue
Block a user