1 | "use strict";
|
2 |
|
3 | exports.__esModule = true;
|
4 | exports.default = useStateAsync;
|
5 |
|
6 | var _react = require("react");
|
7 |
|
8 |
|
9 |
|
10 |
|
11 |
|
12 |
|
13 |
|
14 |
|
15 |
|
16 |
|
17 |
|
18 |
|
19 |
|
20 |
|
21 |
|
22 |
|
23 |
|
24 |
|
25 | function useStateAsync(initialState) {
|
26 | var _useState = (0, _react.useState)(initialState),
|
27 | state = _useState[0],
|
28 | setState = _useState[1];
|
29 |
|
30 | var resolvers = (0, _react.useRef)([]);
|
31 | (0, _react.useEffect)(function () {
|
32 | resolvers.current.forEach(function (resolve) {
|
33 | return resolve(state);
|
34 | });
|
35 | resolvers.current.length = 0;
|
36 | }, [state]);
|
37 | var setStateAsync = (0, _react.useCallback)(function (update) {
|
38 | return new Promise(function (resolve, reject) {
|
39 | setState(function (prevState) {
|
40 | try {
|
41 | var nextState;
|
42 |
|
43 | if (update instanceof Function) {
|
44 | nextState = update(prevState);
|
45 | } else {
|
46 | nextState = update;
|
47 | }
|
48 |
|
49 |
|
50 |
|
51 |
|
52 | if (!resolvers.current.length && Object.is(nextState, prevState)) {
|
53 | resolve(nextState);
|
54 | } else {
|
55 | resolvers.current.push(resolve);
|
56 | }
|
57 |
|
58 | return nextState;
|
59 | } catch (e) {
|
60 | reject(e);
|
61 | throw e;
|
62 | }
|
63 | });
|
64 | });
|
65 | }, [setState]);
|
66 | return [state, setStateAsync];
|
67 | } |
\ | No newline at end of file |