2021-01-07 09:22:40 -05:00
|
|
|
import { useState, useCallback } from 'react'
|
2021-05-18 06:56:56 -04:00
|
|
|
import localStorage from '../../infrastructure/local-storage'
|
2021-01-07 09:22:40 -05:00
|
|
|
|
|
|
|
function usePersistedState(key, defaultValue) {
|
|
|
|
const [value, setValue] = useState(() => {
|
2021-09-23 06:35:50 -04:00
|
|
|
return localStorage.getItem(key) ?? defaultValue
|
2021-01-07 09:22:40 -05:00
|
|
|
})
|
|
|
|
|
|
|
|
const updateFunction = useCallback(
|
|
|
|
newValue => {
|
2021-09-23 06:35:50 -04:00
|
|
|
setValue(value => {
|
|
|
|
const actualNewValue =
|
|
|
|
typeof newValue === 'function' ? newValue(value) : newValue
|
|
|
|
|
|
|
|
if (actualNewValue === defaultValue) {
|
|
|
|
localStorage.removeItem(key)
|
|
|
|
} else {
|
|
|
|
localStorage.setItem(key, actualNewValue)
|
|
|
|
}
|
|
|
|
|
|
|
|
return actualNewValue
|
|
|
|
})
|
2021-01-07 09:22:40 -05:00
|
|
|
},
|
|
|
|
[key, defaultValue]
|
|
|
|
)
|
|
|
|
|
|
|
|
return [value, updateFunction]
|
|
|
|
}
|
|
|
|
|
|
|
|
export default usePersistedState
|