91 lines
2.4 KiB
TypeScript
91 lines
2.4 KiB
TypeScript
import {
|
|
useCallback,
|
|
useEffect,
|
|
useMemo,
|
|
useSyncExternalStore,
|
|
useState,
|
|
} from 'react';
|
|
import { LocalStorage } from '@vegaprotocol/utils';
|
|
|
|
type LocalStorageCallback = (key: string) => void;
|
|
const LOCAL_STORAGE_CALLBACKS = new Set<LocalStorageCallback>();
|
|
const registerCallback = (callback: LocalStorageCallback) =>
|
|
LOCAL_STORAGE_CALLBACKS.add(callback);
|
|
const unregisterCallback = (callback: LocalStorageCallback) =>
|
|
LOCAL_STORAGE_CALLBACKS.delete(callback);
|
|
const triggerCallbacks = (key: string) =>
|
|
LOCAL_STORAGE_CALLBACKS.forEach((cb) => cb(key));
|
|
|
|
type UseLocalStorageHook = [
|
|
string | null | undefined,
|
|
(value: string) => void,
|
|
() => void
|
|
];
|
|
|
|
export const useLocalStorage = (key: string) => {
|
|
const subscribe = useCallback(
|
|
(onStoreChange: () => void) => {
|
|
const callback = (cbKey: string) => {
|
|
if (cbKey === key) onStoreChange();
|
|
};
|
|
registerCallback(callback);
|
|
return () => unregisterCallback(callback);
|
|
},
|
|
[key]
|
|
);
|
|
const getSnapshot = () => {
|
|
const item = LocalStorage.getItem(key);
|
|
return item;
|
|
};
|
|
const setValue = useCallback(
|
|
(value: string) => {
|
|
LocalStorage.setItem(key, value);
|
|
triggerCallbacks(key);
|
|
},
|
|
[key]
|
|
);
|
|
const removeValue = useCallback(() => {
|
|
LocalStorage.removeItem(key);
|
|
triggerCallbacks(key);
|
|
}, [key]);
|
|
const value = useSyncExternalStore(subscribe, getSnapshot, () => null);
|
|
|
|
// sync in all tabs
|
|
const onStorage = useCallback(
|
|
(ev: StorageEvent) => {
|
|
if (ev.storageArea === window.localStorage && ev.key === key) {
|
|
triggerCallbacks(key);
|
|
}
|
|
},
|
|
[key]
|
|
);
|
|
useEffect(() => {
|
|
window.addEventListener('storage', onStorage);
|
|
return () => window.removeEventListener('storage', onStorage);
|
|
}, [key, onStorage]);
|
|
|
|
return useMemo<UseLocalStorageHook>(
|
|
() => [value, setValue, removeValue],
|
|
[removeValue, setValue, value]
|
|
);
|
|
};
|
|
|
|
export const useLocalStorageSnapshot = (key: string) => {
|
|
const [value, setStoredValue] = useState(LocalStorage.getItem(key));
|
|
const setValue = useCallback(
|
|
(value: string) => {
|
|
LocalStorage.setItem(key, value);
|
|
setStoredValue(value);
|
|
},
|
|
[key]
|
|
);
|
|
const removeValue = useCallback(() => {
|
|
LocalStorage.removeItem(key);
|
|
setStoredValue(null);
|
|
}, [key]);
|
|
return useMemo<UseLocalStorageHook>(
|
|
() => [value, setValue, removeValue],
|
|
[removeValue, setValue, value]
|
|
);
|
|
};
|