-
-
Notifications
You must be signed in to change notification settings - Fork 46.7k
/
index.tsx
111 lines (94 loc) · 3.03 KB
/
index.tsx
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
import * as React from 'react';
import usePatchElement from '../../_util/hooks/usePatchElement';
import type { ModalStaticFunctions } from '../confirm';
import { withConfirm, withError, withInfo, withSuccess, withWarn } from '../confirm';
import type { ModalFuncProps } from '../Modal';
import type { HookModalRef } from './HookModal';
import HookModal from './HookModal';
let uuid = 0;
interface ElementsHolderRef {
patchElement: ReturnType<typeof usePatchElement>[1];
}
const ElementsHolder = React.memo(
React.forwardRef<ElementsHolderRef>((_props, ref) => {
const [elements, patchElement] = usePatchElement();
React.useImperativeHandle(
ref,
() => ({
patchElement,
}),
[],
);
// eslint-disable-next-line react/jsx-no-useless-fragment
return <>{elements}</>;
}),
);
function useModal(): readonly [Omit<ModalStaticFunctions, 'warn'>, React.ReactElement] {
const holderRef = React.useRef<ElementsHolderRef>(null);
// ========================== Effect ==========================
const [actionQueue, setActionQueue] = React.useState<(() => void)[]>([]);
React.useEffect(() => {
if (actionQueue.length) {
const cloneQueue = [...actionQueue];
cloneQueue.forEach((action) => {
action();
});
setActionQueue([]);
}
}, [actionQueue]);
// =========================== Hook ===========================
const getConfirmFunc = React.useCallback(
(withFunc: (config: ModalFuncProps) => ModalFuncProps) =>
function hookConfirm(config: ModalFuncProps) {
uuid += 1;
const modalRef = React.createRef<HookModalRef>();
let closeFunc: Function | undefined;
const modal = (
<HookModal
key={`modal-${uuid}`}
config={withFunc(config)}
ref={modalRef}
afterClose={() => {
closeFunc?.();
}}
/>
);
closeFunc = holderRef.current?.patchElement(modal);
return {
destroy: () => {
function destroyAction() {
modalRef.current?.destroy();
}
if (modalRef.current) {
destroyAction();
} else {
setActionQueue((prev) => [...prev, destroyAction]);
}
},
update: (newConfig: ModalFuncProps) => {
function updateAction() {
modalRef.current?.update(newConfig);
}
if (modalRef.current) {
updateAction();
} else {
setActionQueue((prev) => [...prev, updateAction]);
}
},
};
},
[],
);
const fns = React.useMemo<Omit<ModalStaticFunctions, 'warn'>>(
() => ({
info: getConfirmFunc(withInfo),
success: getConfirmFunc(withSuccess),
error: getConfirmFunc(withError),
warning: getConfirmFunc(withWarn),
confirm: getConfirmFunc(withConfirm),
}),
[],
);
return [fns, <ElementsHolder key="modal-holder" ref={holderRef} />] as const;
}
export default useModal;