45 lines
1.6 KiB
TypeScript
45 lines
1.6 KiB
TypeScript
import { useState } from "preact/hooks";
|
|
import storage from 'local-storage-fallback'
|
|
|
|
export function useLocalStorage<T>(key: string, initialValue: T) {
|
|
//Define storage to localStorage and fallback if not avalaible (in wkhtmltopdf for exemple)
|
|
let localStorage: any;
|
|
localStorage = window.localStorage
|
|
if (null === localStorage) {
|
|
localStorage = storage;
|
|
}
|
|
|
|
// State to store our value
|
|
// Pass initial state function to useState so logic is only executed once
|
|
const [storedValue, setStoredValue] = useState(() => {
|
|
try {
|
|
// Get from local storage by key
|
|
const item = localStorage.getItem(key);
|
|
// Parse stored json or if none return initialValue
|
|
return item ? JSON.parse(item) : initialValue;
|
|
} catch (error) {
|
|
// If error also return initialValue
|
|
console.error(error);
|
|
return initialValue;
|
|
}
|
|
});
|
|
|
|
// Return a wrapped version of useState's setter function that ...
|
|
// ... persists the new value to localStorage.
|
|
const setValue = (value: T) => {
|
|
try {
|
|
// Allow value to be a function so we have same API as useState
|
|
const valueToStore =
|
|
value instanceof Function ? value(storedValue) : value;
|
|
// Save state
|
|
setStoredValue(valueToStore);
|
|
// Save to local storage
|
|
localStorage.setItem(key, JSON.stringify(valueToStore));
|
|
} catch (error) {
|
|
// A more advanced implementation would handle the error case
|
|
console.error(error);
|
|
}
|
|
};
|
|
|
|
return [storedValue, setValue];
|
|
} |