import { VNodeTypes, CSSProperties } from 'vue'; import Notification from '../vc-notification'; import CheckCircleOutlined from '@ant-design/icons-vue/CheckCircleOutlined'; import InfoCircleOutlined from '@ant-design/icons-vue/InfoCircleOutlined'; import CloseCircleOutlined from '@ant-design/icons-vue/CloseCircleOutlined'; import ExclamationCircleOutlined from '@ant-design/icons-vue/ExclamationCircleOutlined'; import CloseOutlined from '@ant-design/icons-vue/CloseOutlined'; export type NotificationPlacement = 'topLeft' | 'topRight' | 'bottomLeft' | 'bottomRight'; export type IconType = 'success' | 'info' | 'error' | 'warning'; export interface ConfigProps { top?: string | number; bottom?: string | number; duration?: number; placement?: NotificationPlacement; getContainer?: () => HTMLElement; closeIcon?: VNodeTypes; } const notificationInstance: { [key: string]: any } = {}; let defaultDuration = 4.5; let defaultTop = '24px'; let defaultBottom = '24px'; let defaultPlacement: NotificationPlacement = 'topRight'; let defaultGetContainer = () => document.body; let defaultCloseIcon = null; function setNotificationConfig(options: ConfigProps) { const { duration, placement, bottom, top, getContainer, closeIcon } = options; if (duration !== undefined) { defaultDuration = duration; } if (placement !== undefined) { defaultPlacement = placement; } if (bottom !== undefined) { defaultBottom = typeof bottom === 'number' ? `${bottom}px` : bottom; } if (top !== undefined) { defaultTop = typeof top === 'number' ? `${top}px` : top; } if (getContainer !== undefined) { defaultGetContainer = getContainer; } if (closeIcon !== undefined) { defaultCloseIcon = closeIcon; } } function getPlacementStyle( placement: NotificationPlacement, top: string = defaultTop, bottom: string = defaultBottom, ) { let style; switch (placement) { case 'topLeft': style = { left: '0px', top, bottom: 'auto', }; break; case 'topRight': style = { right: '0px', top, bottom: 'auto', }; break; case 'bottomLeft': style = { left: '0px', top: 'auto', bottom, }; break; default: style = { right: '0px', top: 'auto', bottom, }; break; } return style; } type NotificationInstanceProps = { prefixCls: string; placement?: NotificationPlacement; getContainer?: () => HTMLElement; top?: string; bottom?: string; closeIcon?: VNodeTypes; }; function getNotificationInstance( { prefixCls, placement = defaultPlacement, getContainer = defaultGetContainer, top, bottom, closeIcon = defaultCloseIcon, }: NotificationInstanceProps, callback: (n: any) => void, ) { const cacheKey = `${prefixCls}-${placement}`; if (notificationInstance[cacheKey]) { callback(notificationInstance[cacheKey]); return; } Notification.newInstance( { prefixCls, class: `${prefixCls}-${placement}`, style: getPlacementStyle(placement, top, bottom), getContainer, closeIcon: () => { const closeIconToRender = ( {closeIcon || } ); return closeIconToRender; }, }, (notification: any) => { notificationInstance[cacheKey] = notification; callback(notification); }, ); } const typeToIcon = { success: CheckCircleOutlined, info: InfoCircleOutlined, error: CloseCircleOutlined, warning: ExclamationCircleOutlined, }; export interface ArgsProps { message: VNodeTypes; description?: VNodeTypes; btn?: VNodeTypes; key?: string; onClose?: () => void; duration?: number | null; icon?: VNodeTypes; placement?: NotificationPlacement; style?: CSSProperties; prefixCls?: string; class?: string; readonly type?: IconType; onClick?: () => void; top?: string; bottom?: string; getContainer?: () => HTMLElement; closeIcon?: VNodeTypes; } function notice(args: ArgsProps) { const { icon, type, description, message, btn } = args; const outerPrefixCls = args.prefixCls || 'ant-notification'; const prefixCls = `${outerPrefixCls}-notice`; const duration = args.duration === undefined ? defaultDuration : args.duration; let iconNode = null; if (icon) { iconNode = () => {icon}; } else if (type) { const Icon = typeToIcon[type]; iconNode = () => ; } const { placement, top, bottom, getContainer, closeIcon } = args; getNotificationInstance( { prefixCls: outerPrefixCls, placement, top, bottom, getContainer, closeIcon, }, notification => { notification.notice({ content: () => (
{iconNode && iconNode()}
{!description && iconNode ? ( ) : null} {message}
{description}
{btn ? {btn} : null}
), duration, closable: true, onClose: args.onClose, onClick: args.onClick, key: args.key, style: args.style || {}, class: args.class, }); }, ); } const api: any = { open: notice, close(key: string) { Object.keys(notificationInstance).forEach(cacheKey => notificationInstance[cacheKey].removeNotice(key), ); }, config: setNotificationConfig, destroy() { Object.keys(notificationInstance).forEach(cacheKey => { notificationInstance[cacheKey].destroy(); delete notificationInstance[cacheKey]; }); }, }; ['success', 'info', 'warning', 'error'].forEach(type => { api[type] = args => api.open({ ...args, type, }); }); api.warn = api.warning; export default api;