mirror of
https://gitee.com/ant-design/ant-design.git
synced 2024-12-14 08:51:26 +08:00
164ea65250
* 🐞 FIX: destroyAll not working with useModal * 🧪 TEST: add test * Update components/modal/useModal/index.tsx Co-authored-by: Amumu <yoyo837@hotmail.com> Co-authored-by: Amumu <yoyo837@hotmail.com>
120 lines
3.2 KiB
TypeScript
120 lines
3.2 KiB
TypeScript
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';
|
|
import destroyFns from '../destroyFns';
|
|
|
|
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 [
|
|
instance: Omit<ModalStaticFunctions, 'warn'>,
|
|
contextHolder: 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);
|
|
|
|
if (closeFunc) {
|
|
destroyFns.push(closeFunc);
|
|
}
|
|
|
|
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;
|