useRefState useStateWithRef

const useStateWithRef = (initialState) => {
  const [state, _setState] = React.useState(initialState);
  const ref = React.useRef(state);
  const setState = React.useCallback(
    (newState) => {
      if (typeof newState === 'function') {
        _setState(prevState => {
          const computedState = newState(prevState);
          ref.current = computedState;
          return computedState;
        });
      } else {
        ref.current = newState;
        _setState(newState);
      }
    },
    []
  );
  return [state, setState, ref];
}

typescript

export const useStateWithRef = <T, >(initialState: T): [T, (arg1: T) => void, RefObject<T>] => {
    const [state, _setState] = React.useState(initialState);
    const ref = React.useRef(state);
    const setState = React.useCallback(
        (newState) => {
            if (typeof newState === 'function') {
                _setState((prevState: any) => {
                    const computedState = newState(prevState);
                    ref.current = computedState;
                    return computedState;
                });
            } else {
                ref.current = newState;
                _setState(newState);
            }
        },
        []
    );
    return [state, setState, ref];
}