import { useState, useEffect } from "react"
// https://usehooks.com/useLocalStorage/
export function useStorage(key, initialValue, storage = window.localStorage) {
const getValue = (key, initialValue) => {
try {
// Get from local storage by key
const item = storage.getItem(key)
// Parse stored json or if none return initialValue
return item ? JSON.parse(item) : initialValue
} catch (error) {
// If error also return initialValue
console.log(error)
return initialValue
}
}
// State to store our value
// Pass initial state function to useState so logic is only executed once
const [storedValue, setStoredValue] = useState(() => getValue(key, initialValue))
useEffect(() => {
const val = getValue(key, initialValue)
setStoredValue(val)
}, [key])
// Return a wrapped version of useState's setter function that ...
// ... persists the new value to localStorage.
const setValue = value => {
try {
// Allow value to be a function so we have same API as useState
const valueToStore = value instanceof Function ? value(storedValue) : value
// Save to local storage
storage.setItem(key, JSON.stringify(valueToStore))
// Save state
setStoredValue(valueToStore)
} catch (error) {
// A more advanced implementation would handle the error case
console.log(error)
}
}
return [storedValue, setValue]
}
|