import React from 'react'; import Notification from 'rc-notification'; import Icon from '../icon'; import assign from 'object-assign'; let notificationInstance; let defaultDuration = 4.5; let defaultTop = 24; let defaultBottom = 24; let defaultPlacement = 'topRight'; 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; getContainer?: () => HTMLElement; } export interface ConfigProps { top?: number; bottom?: number; duration?: number; placement?: notificationPlacement; } function getNotificationInstance(prefixCls) { if (notificationInstance) { return notificationInstance; } notificationInstance = (Notification as any).newInstance({ prefixCls: prefixCls, className: `${prefixCls}-${defaultPlacement}`, style: getPlacementStyle(defaultPlacement), }); return notificationInstance; } function notice(args) { const outerPrefixCls = args.prefixCls || 'ant-notification'; const prefixCls = `${outerPrefixCls}-notice`; if (args.placement !== undefined) { defaultPlacement = args.placement; notificationInstance = null; // delete notificationInstance for new defaultPlacement } 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; 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: {}, }); } 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) { notificationInstance.removeNotice(key); } }, config(options: ConfigProps) { const { duration, placement, bottom, top } = options; if (placement !== undefined) { defaultPlacement = placement; } if (bottom !== undefined) { defaultBottom = bottom; } if (top !== undefined) { defaultTop = top; } // delete notificationInstance if (placement !== undefined || bottom !== undefined || top !== undefined) { notificationInstance = null; } if (duration !== undefined) { defaultDuration = duration; } }, destroy() { if (notificationInstance) { notificationInstance.destroy(); notificationInstance = null; } }, }; ['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;