53 lines
1.8 KiB
JavaScript
53 lines
1.8 KiB
JavaScript
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];
|
|
}); |