kiss-translator/src/hooks/Storage.js

import { useCallback, useEffect, useState } from "react";
import { storage } from "../libs/storage";

export function useStorage(key, defaultVal = null) {
  const [loading, setLoading] = useState(true);
  const [data, setData] = useState(defaultVal);

  const save = useCallback(
    async (val) => {
      setData(val);
      await storage.setObj(key, val);
    },
    [key]
  );

  const update = useCallback(
    async (obj) => {
      setData((pre) => ({ ...pre, ...obj }));
      await storage.putObj(key, obj);
    },
    [key]
  );

  const remove = useCallback(async () => {
    setData(null);
    await storage.del(key);
  }, [key]);

  const reload = useCallback(async () => {
    const val = await storage.getObj(key);
    if (val) {
      setData(val);
    } else if (defaultVal) {
      setData(defaultVal);
      await storage.setObj(key, defaultVal);
    }
  }, [key, defaultVal]);

  useEffect(() => {
    (async () => {
      try {
        setLoading(true);
        await reload();
      } catch (err) {
        //
      } finally {
        setLoading(false);
      }
    })();
  }, [reload]);

  return { data, save, update, remove, reload, loading };
}