forked from ritz078/transform
-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathuseSessionStorage.ts
More file actions
47 lines (42 loc) · 1.5 KB
/
useSessionStorage.ts
File metadata and controls
47 lines (42 loc) · 1.5 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
import { useState } from "react";
import pkg from "../package.json";
const prefix = `transform:${pkg.version}:`;
export function useSessionStorage(key, initialValue) {
// 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 =
typeof window !== "undefined"
? window.sessionStorage.getItem(prefix + key) || initialValue
: initialValue;
// Parse stored json or if none return initialValue
return key.startsWith("data:") ? item : JSON.parse(item);
} catch (error) {
// If error also return initialValue
return initialValue;
}
});
// Return a wrapped version of useState's setter function that ...
// ... persists the new value to sessionStorage.
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 state
setStoredValue(valueToStore);
// Save to local storage
if (typeof window !== "undefined")
window.sessionStorage.setItem(
prefix + key,
key.startsWith("data:") ? valueToStore : JSON.stringify(valueToStore)
);
} catch (error) {
// A more advanced implementation would handle the error case
console.log(error);
}
};
return [storedValue, setValue];
}