ant-design/components/modal/useModal/index.tsx

112 lines
3.0 KiB
TypeScript
Raw Normal View History

import * as React from 'react';
2020-08-19 09:50:04 +08:00
import usePatchElement from '../../_util/hooks/usePatchElement';
2022-06-22 14:57:09 +08:00
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] {
2022-11-09 17:36:49 +08:00
const holderRef = React.useRef<ElementsHolderRef>(null);
// ========================== Effect ==========================
const [actionQueue, setActionQueue] = React.useState<(() => void)[]>([]);
React.useEffect(() => {
if (actionQueue.length) {
const cloneQueue = [...actionQueue];
2022-11-13 14:33:07 +08:00
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>();
2022-11-09 17:36:49 +08:00
let closeFunc: Function | undefined;
const modal = (
<HookModal
key={`modal-${uuid}`}
config={withFunc(config)}
ref={modalRef}
afterClose={() => {
2022-11-09 17:36:49 +08:00
closeFunc?.();
}}
/>
);
closeFunc = holderRef.current?.patchElement(modal);
return {
destroy: () => {
function destroyAction() {
modalRef.current?.destroy();
}
if (modalRef.current) {
destroyAction();
} else {
2022-11-13 14:33:07 +08:00
setActionQueue((prev) => [...prev, destroyAction]);
}
},
update: (newConfig: ModalFuncProps) => {
function updateAction() {
modalRef.current?.update(newConfig);
}
if (modalRef.current) {
updateAction();
} else {
2022-11-13 14:33:07 +08:00
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;