parent
860f24a680
commit
6fd03b1f4c
@ -0,0 +1,36 @@ |
|||||||
|
import { useState } from 'react' |
||||||
|
|
||||||
|
// Hook
|
||||||
|
export const useLocalStorage = (key: string, initialValue: any) => { |
||||||
|
// State to store our value
|
||||||
|
// Pass initial state function to useState so logic is only executed once
|
||||||
|
const [storedValue, setStoredValue] = useState<any>(() => { |
||||||
|
try { |
||||||
|
// Get from local storage by key
|
||||||
|
const item = window.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.log(error) |
||||||
|
return initialValue |
||||||
|
} |
||||||
|
}) |
||||||
|
// Return a wrapped version of useState's setter function that ...
|
||||||
|
// ... persists the new value to localStorage.
|
||||||
|
const setValue = (value: any | ((val: any) => any)) => { |
||||||
|
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.localStorage.setItem(key, JSON.stringify(valueToStore)) |
||||||
|
} catch (error) { |
||||||
|
// A more advanced implementation would handle the error case
|
||||||
|
console.log(error) |
||||||
|
} |
||||||
|
} |
||||||
|
return [storedValue, setValue] as const |
||||||
|
} |
Loading…
Reference in new issue