TheGame/packages/web/contexts/SetupContext.tsx

import React, { useEffect, useState } from 'react';

type SetupContextType = {
  useProgress: (numProgressSteps: number) => any[];
  step: number;
  progress: number;
  setStep: React.Dispatch<React.SetStateAction<number>>;
  setProgress: React.Dispatch<React.SetStateAction<number>>;
  numTotalSteps: number;
};

export const SetupContext = React.createContext<SetupContextType>({
  useProgress: (numProgressSteps: number) => [numProgressSteps],
  step: 0,
  progress: 0,
  setStep: () => undefined,
  setProgress: () => undefined,
  numTotalSteps: 0,
});

export const SetupContextProvider: React.FC = ({ children }) => {
  const [step, setStep] = useState<number>(0);
  const [progress, setProgress] = useState<number>(0.5);
  const numTotalSteps = 3;

  const useProgress = (numProgressSteps: number) => {
    const [currentProgress, setCurrentProgress] = useState<number>(0);

    useEffect(() => {
      const progressValue = (currentProgress + 1) / numProgressSteps;
      setProgress(progressValue);
    }, [currentProgress, numProgressSteps]);

    const onNextPress = () => {
      if ((currentProgress + 1) % numProgressSteps === 0) {
        setStep((_step) => (_step + 1) % numTotalSteps);
      } else {
        setCurrentProgress(
          (_currentProgress) => (_currentProgress + 1) % numProgressSteps,
        );
      }
    };

    return [currentProgress, onNextPress];
  };

  return (
    <SetupContext.Provider
      value={{
        useProgress,
        step,
        progress,
        setStep,
        setProgress,
        numTotalSteps,
      }}
    >
      {children}
    </SetupContext.Provider>
  );
};