useStateRef

useStateRef.js

import { useState, useCallback } from 'react';

export const useStateRef = initialState => {
  const [state, setState] = useState(() => ({
    current: typeof initialState === 'function' ? initialState() : initialState
  }));

  const setStateRef = useCallback(
    (setter, forceRender) => {
      setState(prev => {
        const current = typeof setter === 'function' ? setter(prev.current) : setter;
        if (forceRender) return { current };
        prev.current = current;
        return prev;
      });
    },
    [setState]
  );
  return [state, setStateRef];
};