department-of-veterans-affairs/vets-website

import React from 'react';

export const useLocalStorage = (storageKey, fallbackState) => {
  const [value, setValue] = React.useState(
    JSON.parse(localStorage.getItem(storageKey)) ?? fallbackState,
  );

  React.useEffect(
    () => {
      if (value === '') {
        localStorage.removeItem(storageKey);
        return;
      }
      localStorage.setItem(storageKey, JSON.stringify(value));
    },
    [value, storageKey],
  );

  return [value, setValue];
};