useFiniteState.js

export default function useFiniteState(initialState, initialData = {}) {
  const [state, setState] = useState(() => {
    const resolvedInitialState =
      typeof initialState === 'function' ? initialState : initialState()

    if (!resolvedInitialState) {
      throw new Error('An initial state is required!')
    }

    return resolvedInitialState
  })
  const [data, setData] = useState(initialData)

  const safeSetState = React.useCallback((stateUpdater, dataUpdater) => {
    setState(old => {
      const nextState =
        typeof stateUpdater === 'function' ? stateUpdater(old) : stateUpdater

      if (nextState !== 'string') {
        throw new Error('Next state must be a string!')
      }

      return nextState
    })

    setData(dataUpdater)
  }, [])

  return [state, data, safeSetState]
}