61 lines
1.7 KiB
TypeScript
61 lines
1.7 KiB
TypeScript
import { useState, useEffect } from 'react'
|
|
|
|
export function usePersistedState<T>({
|
|
key,
|
|
initialValue,
|
|
isSessionStorage,
|
|
}: {
|
|
key: string
|
|
initialValue: T
|
|
isSessionStorage?: boolean
|
|
}): [T, (x: T | ((prev: T) => T)) => void] {
|
|
// State to store our value
|
|
// Pass initial state function to useState so logic is only executed once
|
|
const [storedValue, setStoredValue] = useState(initialValue)
|
|
|
|
useEffect(() => {
|
|
if (typeof window === 'undefined') {
|
|
return
|
|
}
|
|
try {
|
|
// Get from local storage by key
|
|
const item =
|
|
window[isSessionStorage ? 'sessionStorage' : 'localStorage'].getItem(
|
|
key
|
|
)
|
|
// Parse stored json or if none return initialValue
|
|
if (item) {
|
|
setStoredValue(JSON.parse(item))
|
|
}
|
|
} catch (error) {
|
|
// If error also return initialValue
|
|
console.log(error)
|
|
}
|
|
}, [isSessionStorage, key, setStoredValue])
|
|
|
|
// Return a wrapped version of useState's setter function that ...
|
|
// ... persists the new value to localStorage.
|
|
const setValue = (value: T | ((prev: T) => T)): void => {
|
|
if (typeof window === 'undefined') {
|
|
return
|
|
}
|
|
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
|
|
window[isSessionStorage ? 'sessionStorage' : 'localStorage'].setItem(
|
|
key,
|
|
JSON.stringify(valueToStore)
|
|
)
|
|
} catch (error) {
|
|
// A more advanced implementation would handle the error case
|
|
console.log(error)
|
|
}
|
|
}
|
|
|
|
return [storedValue, setValue]
|
|
}
|