forked from bcgame-project/verify
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathhooks.js
More file actions
46 lines (38 loc) · 983 Bytes
/
hooks.js
File metadata and controls
46 lines (38 loc) · 983 Bytes
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
function useMountedState() {
const mountedRef = React.useRef(false);
const get = React.useCallback(() => mountedRef.current, []);
React.useEffect(() => {
mountedRef.current = true;
return () => {
mountedRef.current = false;
};
}, []);
return get;
}
function useSetState (
initialState
) {
const [state, set] = React.useState(initialState);
const isMounted = useMountedState();
const setState = React.useCallback(
(patch) => {
if (isMounted()) {
set((prevState) => {
const newState = patch instanceof Function ? patch(prevState) : patch;
const needUpdate =
Object.keys(newState).findIndex(
(it) => prevState[it] !== newState[it]
) != -1;
if (!needUpdate) return prevState;
return Object.assign({}, prevState, newState);
});
}
},
[set]
);
return [state, setState];
}
window.hooks = {
useMountedState,
useSetState
}