import React from 'react'; import Notification from 'rc-notification'; import Icon from '../icon'; import assign from 'object-assign'; const notificationInstance = {}; let defaultDuration = 4.5; let defaultTop = 24; let defaultBottom = 24; let defaultPlacement = 'topRight'; let defaultGetContainer; export type notificationPlacement = 'topLeft' | 'topRight' | 'bottomLeft' | 'bottomRight'; function getPlacementStyle(placement) { let style; switch (placement) { case 'topLeft': style = { left: 0, top: defaultTop, bottom: 'auto', }; break; case 'bottomLeft': style = { left: 0, top: 'auto', bottom: defaultBottom, }; break; case 'bottomRight': style = { right: 0, top: 'auto', bottom: defaultBottom, }; break; default: style = { right: 0, top: defaultTop, bottom: 'auto', }; } return style; } export interface ArgsProps { message: React.ReactNode; description: React.ReactNode; btn?: React.ReactNode; key?: string; onClose?: () => void; duration?: number; icon?: React.ReactNode; placement?: notificationPlacement; style?: string; className?: string; } export interface ConfigProps { top?: number; bottom?: number; duration?: number; placement?: notificationPlacement; getContainer?: () => HTMLElement; } function getNotificationInstance(prefixCls) { if (notificationInstance[defaultPlacement]) { return notificationInstance[defaultPlacement]; } notificationInstance[defaultPlacement] = (Notification as any).newInstance({ prefixCls: prefixCls, className: `${prefixCls}-${defaultPlacement}`, style: getPlacementStyle(defaultPlacement), getContainer: defaultGetContainer, }); return notificationInstance[defaultPlacement]; } function notice(args) { const outerPrefixCls = args.prefixCls || 'ant-notification'; const prefixCls = `${outerPrefixCls}-notice`; if (args.placement !== undefined) { defaultPlacement = args.placement; } let duration; if (args.duration === undefined) { duration = defaultDuration; } else { duration = args.duration; } let iconType = ''; switch (args.type) { case 'success': iconType = 'check-circle-o'; break; case 'info': iconType = 'info-circle-o'; break; case 'error': iconType = 'cross-circle-o'; break; case 'warning': iconType = 'exclamation-circle-o'; break; default: iconType = 'info-circle'; } let iconNode; if (args.icon) { iconNode = ( {args.icon} ); } else if (args.type) { iconNode = ; } const autoMarginTag = (!args.description && iconNode) ? : null; const { style, className } = args; getNotificationInstance(outerPrefixCls).notice({ content: (
{iconNode}
{autoMarginTag} {args.message}
{args.description}
{args.btn ? {args.btn} : null}
), duration, closable: true, onClose: args.onClose, key: args.key, style: assign({}, style), className, }); } const api: { success?(args: ArgsProps): void; error?(args: ArgsProps): void; info?(args: ArgsProps): void; warn?(args: ArgsProps): void; warning?(args: ArgsProps): void; open(args: ArgsProps): void; close(key: string): void; config(options: ConfigProps): void; destroy(): void; } = { open(args: ArgsProps) { notice(args); }, close(key) { if (notificationInstance[defaultPlacement]) { notificationInstance[defaultPlacement].removeNotice(key); } }, config(options: ConfigProps) { const { duration, placement, bottom, top, getContainer } = options; if (placement !== undefined) { defaultPlacement = placement; } if (bottom !== undefined) { defaultBottom = bottom; } if (top !== undefined) { defaultTop = top; } if (getContainer !== undefined) { defaultGetContainer = getContainer; } // delete notificationInstance if (placement !== undefined || bottom !== undefined || top !== undefined) { const notify = notificationInstance[defaultPlacement]; if (notify) { notify.destroy(); } delete notificationInstance[defaultPlacement]; } if (duration !== undefined) { defaultDuration = duration; } }, destroy() { Object.keys(notificationInstance).forEach(key => { notificationInstance[key].destroy(); delete notificationInstance[key]; }); }, }; ['success', 'info', 'warning', 'error'].forEach((type) => { api[type] = (args: ArgsProps) => api.open(assign({}, args, { type })); }); (api as any).warn = (api as any).warning; export default api;