useStorage.js

import localforage from 'localforage';

const useStorage = (state, setState) => {
  const [rehydrated, setRehydrated] = useState(false);
  const [error, setError] = useState(null);

  useEffect(() => {
    // Async rehydration
    localforage.getItem(config.key, (err, value) => {
      if (err) {
        setRehydrated(true);
        return setError(err);
      }
      setState(rehydrated(value));
      setRehydrated(true);
    });
  }, []);

  useEffect(() => {
    if (isNil(state) || isEmpty(state)) {
      localforage.removeItem(config.key);
    }
    localforage.setItem(config.key, hydrate(state));
  }, [state]);
  
  return {
    rehydrated,
    error,
  }
};