import * as React from 'react'; import Notification from 'rc-notification'; import { NotificationInstance as RCNotificationInstance } from 'rc-notification/lib/Notification'; import CloseOutlined from '@ant-design/icons/CloseOutlined'; import classNames from 'classnames'; import CheckCircleOutlined from '@ant-design/icons/CheckCircleOutlined'; import CloseCircleOutlined from '@ant-design/icons/CloseCircleOutlined'; import ExclamationCircleOutlined from '@ant-design/icons/ExclamationCircleOutlined'; import InfoCircleOutlined from '@ant-design/icons/InfoCircleOutlined'; import createUseNotification from './hooks/useNotification'; export type NotificationPlacement = 'topLeft' | 'topRight' | 'bottomLeft' | 'bottomRight'; export type IconType = 'success' | 'info' | 'error' | 'warning'; const notificationInstance: { [key: string]: Promise; } = {}; let defaultDuration = 4.5; let defaultTop = 24; let defaultBottom = 24; let defaultPrefixCls = 'ant-notification'; let defaultPlacement: NotificationPlacement = 'topRight'; let defaultGetContainer: () => HTMLElement; let defaultCloseIcon: React.ReactNode; export interface ConfigProps { top?: number; bottom?: number; duration?: number; prefixCls?: string; placement?: NotificationPlacement; getContainer?: () => HTMLElement; closeIcon?: React.ReactNode; rtl?: boolean; } let rtl = false; function setNotificationConfig(options: ConfigProps) { const { duration, placement, bottom, top, getContainer, closeIcon, prefixCls } = options; if (prefixCls !== undefined) { defaultPrefixCls = prefixCls; } if (duration !== undefined) { defaultDuration = duration; } if (placement !== undefined) { defaultPlacement = placement; } if (bottom !== undefined) { defaultBottom = bottom; } if (top !== undefined) { defaultTop = top; } if (getContainer !== undefined) { defaultGetContainer = getContainer; } if (closeIcon !== undefined) { defaultCloseIcon = closeIcon; } if (options.rtl !== undefined) { rtl = options.rtl; } } function getPlacementStyle( placement: NotificationPlacement, top: number = defaultTop, bottom: number = defaultBottom, ) { let style; switch (placement) { case 'topLeft': style = { left: 0, top, bottom: 'auto', }; break; case 'topRight': style = { right: 0, top, bottom: 'auto', }; break; case 'bottomLeft': style = { left: 0, top: 'auto', bottom, }; break; default: style = { right: 0, top: 'auto', bottom, }; break; } return style; } function getNotificationInstance( args: ArgsProps, callback: (info: { prefixCls: string; instance: RCNotificationInstance }) => void, ) { const { placement = defaultPlacement, top, bottom, getContainer = defaultGetContainer, closeIcon = defaultCloseIcon, } = args; const outerPrefixCls = args.prefixCls || defaultPrefixCls; const prefixCls = `${outerPrefixCls}-notice`; const cacheKey = `${outerPrefixCls}-${placement}`; const cacheInstance = notificationInstance[cacheKey]; if (cacheInstance) { Promise.resolve(cacheInstance).then(instance => { callback({ prefixCls, instance }); }); return; } const closeIconToRender = ( {closeIcon || } ); const notificationClass = classNames(`${outerPrefixCls}-${placement}`, { [`${outerPrefixCls}-rtl`]: rtl === true, }); notificationInstance[cacheKey] = new Promise(resolve => { Notification.newInstance( { prefixCls: outerPrefixCls, className: notificationClass, style: getPlacementStyle(placement, top, bottom), getContainer, closeIcon: closeIconToRender, }, notification => { resolve(notification); callback({ prefixCls, instance: notification, }); }, ); }); } const typeToIcon = { success: CheckCircleOutlined, info: InfoCircleOutlined, error: CloseCircleOutlined, warning: ExclamationCircleOutlined, }; export interface ArgsProps { message: React.ReactNode; description?: React.ReactNode; btn?: React.ReactNode; key?: string; onClose?: () => void; duration?: number | null; icon?: React.ReactNode; placement?: NotificationPlacement; style?: React.CSSProperties; prefixCls?: string; className?: string; readonly type?: IconType; onClick?: () => void; top?: number; bottom?: number; getContainer?: () => HTMLElement; closeIcon?: React.ReactNode; } function getRCNoticeProps(args: ArgsProps, prefixCls: string) { const duration = args.duration === undefined ? defaultDuration : args.duration; let iconNode: React.ReactNode = null; if (args.icon) { iconNode = {args.icon}; } else if (args.type) { iconNode = React.createElement(typeToIcon[args.type] || null, { className: `${prefixCls}-icon ${prefixCls}-icon-${args.type}`, }); } const autoMarginTag = !args.description && iconNode ? ( ) : null; return { content: (
{iconNode}
{autoMarginTag} {args.message}
{args.description}
{args.btn ? {args.btn} : null}
), duration, closable: true, onClose: args.onClose, onClick: args.onClick, key: args.key, style: args.style || {}, className: args.className, }; } const api: any = { open: (args: ArgsProps) => { getNotificationInstance(args, ({ prefixCls, instance }) => { instance.notice(getRCNoticeProps(args, prefixCls)); }); }, close(key: string) { Object.keys(notificationInstance).forEach(cacheKey => Promise.resolve(notificationInstance[cacheKey]).then(instance => { instance.removeNotice(key); }), ); }, config: setNotificationConfig, destroy() { Object.keys(notificationInstance).forEach(cacheKey => { Promise.resolve(notificationInstance[cacheKey]).then(instance => { instance.destroy(); }); delete notificationInstance[cacheKey]; // lgtm[js/missing-await] }); }, }; ['success', 'info', 'warning', 'error'].forEach(type => { api[type] = (args: ArgsProps) => api.open({ ...args, type, }); }); api.warn = api.warning; api.useNotification = createUseNotification(getNotificationInstance, getRCNoticeProps); export interface NotificationInstance { success(args: ArgsProps): void; error(args: ArgsProps): void; info(args: ArgsProps): void; warning(args: ArgsProps): void; open(args: ArgsProps): void; } export interface NotificationApi extends NotificationInstance { warn(args: ArgsProps): void; close(key: string): void; config(options: ConfigProps): void; destroy(): void; // Hooks useNotification: () => [NotificationInstance, React.ReactElement]; } export default api as NotificationApi;