import Notification from '../vc-notification'; import Icon from '../icon'; let defaultDuration = 3; let defaultTop; let messageInstance; let key = 1; let prefixCls = 'ant-message'; let transitionName = 'move-up'; let getContainer = () => document.body; let maxCount; function getMessageInstance(callback) { if (messageInstance) { callback(messageInstance); return; } Notification.newInstance( { prefixCls, transitionName, style: { top: defaultTop }, // 覆盖原来的样式 getContainer, maxCount, }, instance => { if (messageInstance) { callback(messageInstance); return; } messageInstance = instance; callback(instance); }, ); } // type NoticeType = 'info' | 'success' | 'error' | 'warning' | 'loading'; function notice(args) { const duration = args.duration !== undefined ? args.duration : defaultDuration; const iconType = { info: 'info-circle', success: 'check-circle', error: 'close-circle', warning: 'exclamation-circle', loading: 'loading', }[args.type]; const target = key++; const closePromise = new Promise(resolve => { const callback = () => { if (typeof args.onClose === 'function') { args.onClose(); } return resolve(true); }; getMessageInstance(instance => { instance.notice({ key: target, duration, style: {}, content: h => (
{args.icon ? ( typeof args.icon === 'function' ? ( args.icon(h) ) : ( args.icon ) ) : iconType ? ( ) : ( '' )} {typeof content === 'function' ? args.content(h) : args.content}
), onClose: callback, }); }); }); const result = () => { if (messageInstance) { messageInstance.removeNotice(target); } }; result.then = (filled, rejected) => closePromise.then(filled, rejected); result.promise = closePromise; return result; } // type ConfigContent = React.ReactNode | string; // type ConfigDuration = number | (() => void); // export type ConfigOnClose = () => void; // export interface ConfigOptions { // top?: number; // duration?: number; // prefixCls?: string; // getContainer?: () => HTMLElement; // transitionName?: string; // } const api = { open: notice, config(options) { if (options.top !== undefined) { defaultTop = options.top; messageInstance = null; // delete messageInstance for new defaultTop } if (options.duration !== undefined) { defaultDuration = options.duration; } if (options.prefixCls !== undefined) { prefixCls = options.prefixCls; } if (options.getContainer !== undefined) { getContainer = options.getContainer; } if (options.transitionName !== undefined) { transitionName = options.transitionName; messageInstance = null; // delete messageInstance for new transitionName } if (options.maxCount !== undefined) { maxCount = options.maxCount; messageInstance = null; } }, destroy() { if (messageInstance) { messageInstance.destroy(); messageInstance = null; } }, }; ['success', 'info', 'warning', 'error', 'loading'].forEach(type => { api[type] = (content, duration, onClose) => { if (typeof duration === 'function') { onClose = duration; duration = undefined; } return api.open({ content, duration: duration, type, onClose }); }; }); api.warn = api.warning; export default api;