react/components/utils-shared/use-current-state.ts

import { Dispatch, MutableRefObject, SetStateAction, useEffect, useRef, useState } from 'react'

export type CurrentStateType<S> = [
  S, Dispatch<SetStateAction<S>>, MutableRefObject<S>
]

const useCurrentState = <S,>(initialState: S): CurrentStateType<S> => {
  const [state, setState] = useState<S>(initialState as S)
  const ref = useRef<S>(initialState as S)

  useEffect(() => {
    ref.current = state
  }, [state])

  const setValue = (val: SetStateAction<S>) => {
    const result = typeof val === 'function' ? (val as ((prevState: S) => S))(ref.current) : val
    ref.current = result
    setState(result)
  }

  return [state, setValue, ref]
}

export default useCurrentState