1 | function _extends() { _extends = Object.assign || function (target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i]; for (var key in source) { if (Object.prototype.hasOwnProperty.call(source, key)) { target[key] = source[key]; } } } return target; }; return _extends.apply(this, arguments); }
|
2 |
|
3 | import { useCallback, useState } from 'react';
|
4 |
|
5 |
|
6 |
|
7 |
|
8 |
|
9 |
|
10 |
|
11 |
|
12 |
|
13 |
|
14 |
|
15 |
|
16 |
|
17 |
|
18 |
|
19 |
|
20 | export default function useMergeState(initialState) {
|
21 | var _useState = useState(initialState),
|
22 | state = _useState[0],
|
23 | setState = _useState[1];
|
24 |
|
25 | var updater = useCallback(function (update) {
|
26 | if (update === null) return;
|
27 |
|
28 | if (typeof update === 'function') {
|
29 | setState(function (state) {
|
30 | var nextState = update(state);
|
31 | return nextState == null ? state : _extends({}, state, nextState);
|
32 | });
|
33 | } else {
|
34 | setState(function (state) {
|
35 | return _extends({}, state, update);
|
36 | });
|
37 | }
|
38 | }, [setState]);
|
39 | return [state, updater];
|
40 | } |
\ | No newline at end of file |