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 (
content,
duration,
type,
onClose,
) {
const iconType = ({
info: 'info-circle',
success: 'check-circle',
error: 'cross-circle',
warning: 'exclamation-circle',
loading: 'loading',
})[type]
if (typeof duration === 'function') {
onClose = duration
duration = defaultDuration
}
const target = key++
const closePromise = new Promise((resolve) => {
const callback = () => {
if (typeof onClose === 'function') {
onClose()
}
return resolve(true)
}
getMessageInstance((instance) => {
instance.notice({
key: target,
duration,
style: {},
content: (h) => (
{typeof content === 'function' ? content(h) : 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;
// }
export default {
info (content, duration, onClose) {
return notice(content, duration, 'info', onClose)
},
success (content, duration, onClose) {
return notice(content, duration, 'success', onClose)
},
error (content, duration, onClose) {
return notice(content, duration, 'error', onClose)
},
// Departed usage, please use warning()
warn (content, duration, onClose) {
return notice(content, duration, 'warning', onClose)
},
warning (content, duration, onClose) {
return notice(content, duration, 'warning', onClose)
},
loading (content, duration, onClose) {
return notice(content, duration, 'loading', onClose)
},
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
}
},
}