useStateWithRef.ts

interface StateRefObject<T> {
  readonly current: T
}

export const useStateWithRef = <T>(
  initialValue: T | (() => T)
): [T, React.Dispatch<React.SetStateAction<T>>, StateRefObject<T>] => {
  const [state, __setState] = useState(initialValue)
  const stateRef = useRef(state)

  const setState = useCallback((value: T | ((prevState: T) => T)) => {
    __setState(value)
    stateRef.current = isFunction(value) ? value(stateRef.current) : value
  }, [])

  return [state, setState, stateRef]
}