useStep.ts

useStep.ts

import { useState } from 'react'

export const useStep = <
  T extends {
    [step: string]: string
  }
>({
  steps,
  initialStep,
}: {
  steps: T
  initialStep: T[keyof T]
}) => {
  const [step, setStep] = useState(initialStep)
  const stepTo = Object.keys(steps).reduce(
    (acc, k) => ({
      ...acc,
      [k]: () => setStep(k as T[keyof T]),
    }),
    { reset: () => setStep(initialStep) }
  ) as { [step in keyof T | 'reset']: () => void }
  return { step, stepTo }
}