import { getSessionStorageItem, removeSessionStorageItem } from '@/utils/sessionStorage';
import { useEffect, useState } from 'react';

// 读取缓存数据，组件卸载时清除缓存
export function useSessionStorage<T>(key: string, isObject: boolean, initialValue: T) {
  const [storage, setStorage] = useState<T>(initialValue);

  useEffect(() => {
    const res = getSessionStorageItem(key, isObject);
    if (res) {
      setStorage(res);
    }
    return () => {
      removeSessionStorageItem(key);
    };
  }, []);

  return [storage];
}