mirror of
https://github.com/ant-design/ant-design.git
synced 2024-12-11 23:00:20 +08:00
76 lines
2.0 KiB
TypeScript
76 lines
2.0 KiB
TypeScript
import * as React from 'react';
|
|
import { ModalFuncProps } from '../Modal';
|
|
import ConfirmDialog from '../ConfirmDialog';
|
|
import defaultLocale from '../../locale/default';
|
|
import LocaleReceiver from '../../locale-provider/LocaleReceiver';
|
|
import { ConfigContext } from '../../config-provider';
|
|
|
|
export interface HookModalProps {
|
|
afterClose: () => void;
|
|
config: ModalFuncProps;
|
|
}
|
|
|
|
export interface HookModalRef {
|
|
destroy: () => void;
|
|
update: (config: ModalFuncProps) => void;
|
|
}
|
|
|
|
interface ModalLocale {
|
|
okText: string;
|
|
cancelText: string;
|
|
justOkText: string;
|
|
}
|
|
|
|
const HookModal: React.ForwardRefRenderFunction<HookModalRef, HookModalProps> = (
|
|
{ afterClose, config },
|
|
ref,
|
|
) => {
|
|
const [visible, setVisible] = React.useState(true);
|
|
const [innerConfig, setInnerConfig] = React.useState(config);
|
|
const { direction, getPrefixCls } = React.useContext(ConfigContext);
|
|
|
|
const prefixCls = getPrefixCls('modal');
|
|
const rootPrefixCls = getPrefixCls();
|
|
|
|
function close(...args: any[]) {
|
|
setVisible(false);
|
|
const triggerCancel = args.some(param => param && param.triggerCancel);
|
|
if (innerConfig.onCancel && triggerCancel) {
|
|
innerConfig.onCancel();
|
|
}
|
|
}
|
|
|
|
React.useImperativeHandle(ref, () => ({
|
|
destroy: close,
|
|
update: (newConfig: ModalFuncProps) => {
|
|
setInnerConfig(originConfig => ({
|
|
...originConfig,
|
|
...newConfig,
|
|
}));
|
|
},
|
|
}));
|
|
|
|
return (
|
|
<LocaleReceiver componentName="Modal" defaultLocale={defaultLocale.Modal}>
|
|
{(modalLocale: ModalLocale) => (
|
|
<ConfirmDialog
|
|
prefixCls={prefixCls}
|
|
rootPrefixCls={rootPrefixCls}
|
|
{...innerConfig}
|
|
close={close}
|
|
visible={visible}
|
|
afterClose={afterClose}
|
|
okText={
|
|
innerConfig.okText ||
|
|
(innerConfig.okCancel ? modalLocale.okText : modalLocale.justOkText)
|
|
}
|
|
direction={direction}
|
|
cancelText={innerConfig.cancelText || modalLocale.cancelText}
|
|
/>
|
|
)}
|
|
</LocaleReceiver>
|
|
);
|
|
};
|
|
|
|
export default React.forwardRef(HookModal);
|