48 lines
973 B
TypeScript
48 lines
973 B
TypeScript
|
import { useState } from "react";
|
||
|
|
|
||
|
|
export interface UseWizardOpts<T> {
|
||
|
|
next: () => void;
|
||
|
|
prev: () => void;
|
||
|
|
set: (step: T) => void;
|
||
|
|
}
|
||
|
|
export type UseWizard<T> = [T, UseWizardOpts<T>];
|
||
|
|
function useWizard<T>(initial: T, steps: T[]): UseWizard<T> {
|
||
|
|
const [stepIndex, setStepIndex] = useState(steps.indexOf(initial));
|
||
|
|
|
||
|
|
function next() {
|
||
|
|
if (stepIndex === steps.length - 1) {
|
||
|
|
console.error("This is the last step");
|
||
|
|
} else {
|
||
|
|
setStepIndex(stepIndex + 1);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
function prev() {
|
||
|
|
if (stepIndex === 0) {
|
||
|
|
console.error("This is the first step");
|
||
|
|
} else {
|
||
|
|
setStepIndex(stepIndex - 1);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
function set(step: T) {
|
||
|
|
const newStepIndex = steps.findIndex(s => s === step);
|
||
|
|
if (newStepIndex === -1) {
|
||
|
|
console.error("Step does not exist");
|
||
|
|
} else {
|
||
|
|
setStepIndex(newStepIndex);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return [
|
||
|
|
steps[stepIndex],
|
||
|
|
{
|
||
|
|
next,
|
||
|
|
prev,
|
||
|
|
set
|
||
|
|
}
|
||
|
|
];
|
||
|
|
}
|
||
|
|
|
||
|
|
export default useWizard;
|