antd
Version:
An enterprise-class UI design language and React components implementation
222 lines (221 loc) • 8.04 kB
JavaScript
;
var _interopRequireWildcard = require("@babel/runtime/helpers/interopRequireWildcard")["default"];
var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault")["default"];
Object.defineProperty(exports, "__esModule", {
value: true
});
exports.attachTypeApi = attachTypeApi;
exports.getInstance = exports["default"] = void 0;
exports.getKeyThenIncreaseKey = getKeyThenIncreaseKey;
exports.typeList = void 0;
var _extends2 = _interopRequireDefault(require("@babel/runtime/helpers/extends"));
var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty"));
var _CheckCircleFilled = _interopRequireDefault(require("@ant-design/icons/CheckCircleFilled"));
var _CloseCircleFilled = _interopRequireDefault(require("@ant-design/icons/CloseCircleFilled"));
var _ExclamationCircleFilled = _interopRequireDefault(require("@ant-design/icons/ExclamationCircleFilled"));
var _InfoCircleFilled = _interopRequireDefault(require("@ant-design/icons/InfoCircleFilled"));
var _LoadingOutlined = _interopRequireDefault(require("@ant-design/icons/LoadingOutlined"));
var _classnames = _interopRequireDefault(require("classnames"));
var _rcNotification = _interopRequireDefault(require("rc-notification"));
var React = _interopRequireWildcard(require("react"));
var _configProvider = _interopRequireWildcard(require("../config-provider"));
var _useMessage = _interopRequireDefault(require("./hooks/useMessage"));
var messageInstance;
var defaultDuration = 3;
var defaultTop;
var key = 1;
var localPrefixCls = '';
var transitionName = 'move-up';
var hasTransitionName = false;
var getContainer;
var maxCount;
var rtl = false;
function getKeyThenIncreaseKey() {
return key++;
}
function setMessageConfig(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) {
localPrefixCls = options.prefixCls;
}
if (options.getContainer !== undefined) {
getContainer = options.getContainer;
messageInstance = null; // delete messageInstance for new getContainer
}
if (options.transitionName !== undefined) {
transitionName = options.transitionName;
messageInstance = null; // delete messageInstance for new transitionName
hasTransitionName = true;
}
if (options.maxCount !== undefined) {
maxCount = options.maxCount;
messageInstance = null;
}
if (options.rtl !== undefined) {
rtl = options.rtl;
}
}
function getRCNotificationInstance(args, callback) {
var customizePrefixCls = args.prefixCls,
getContextPopupContainer = args.getPopupContainer;
var _globalConfig = (0, _configProvider.globalConfig)(),
getPrefixCls = _globalConfig.getPrefixCls,
getRootPrefixCls = _globalConfig.getRootPrefixCls,
getIconPrefixCls = _globalConfig.getIconPrefixCls;
var prefixCls = getPrefixCls('message', customizePrefixCls || localPrefixCls);
var rootPrefixCls = getRootPrefixCls(args.rootPrefixCls, prefixCls);
var iconPrefixCls = getIconPrefixCls();
if (messageInstance) {
callback({
prefixCls: prefixCls,
rootPrefixCls: rootPrefixCls,
iconPrefixCls: iconPrefixCls,
instance: messageInstance
});
return;
}
var instanceConfig = {
prefixCls: prefixCls,
transitionName: hasTransitionName ? transitionName : "".concat(rootPrefixCls, "-").concat(transitionName),
style: {
top: defaultTop
},
getContainer: getContainer || getContextPopupContainer,
maxCount: maxCount
};
_rcNotification["default"].newInstance(instanceConfig, function (instance) {
if (messageInstance) {
callback({
prefixCls: prefixCls,
rootPrefixCls: rootPrefixCls,
iconPrefixCls: iconPrefixCls,
instance: messageInstance
});
return;
}
messageInstance = instance;
if (process.env.NODE_ENV === 'test') {
messageInstance.config = instanceConfig;
}
callback({
prefixCls: prefixCls,
rootPrefixCls: rootPrefixCls,
iconPrefixCls: iconPrefixCls,
instance: instance
});
});
}
var typeToIcon = {
info: _InfoCircleFilled["default"],
success: _CheckCircleFilled["default"],
error: _CloseCircleFilled["default"],
warning: _ExclamationCircleFilled["default"],
loading: _LoadingOutlined["default"]
};
var typeList = exports.typeList = Object.keys(typeToIcon);
function getRCNoticeProps(args, prefixCls, iconPrefixCls) {
var duration = args.duration !== undefined ? args.duration : defaultDuration;
var IconComponent = typeToIcon[args.type];
var messageClass = (0, _classnames["default"])("".concat(prefixCls, "-custom-content"), (0, _defineProperty2["default"])((0, _defineProperty2["default"])({}, "".concat(prefixCls, "-").concat(args.type), args.type), "".concat(prefixCls, "-rtl"), rtl === true));
return {
key: args.key,
duration: duration,
style: args.style || {},
className: args.className,
content: ( /*#__PURE__*/React.createElement(_configProvider["default"], {
iconPrefixCls: iconPrefixCls
}, /*#__PURE__*/React.createElement("div", {
className: messageClass
}, args.icon || IconComponent && /*#__PURE__*/React.createElement(IconComponent, null), /*#__PURE__*/React.createElement("span", null, args.content)))),
onClose: args.onClose,
onClick: args.onClick
};
}
function notice(args) {
var target = args.key || getKeyThenIncreaseKey();
var closePromise = new Promise(function (resolve) {
var callback = function callback() {
if (typeof args.onClose === 'function') {
args.onClose();
}
return resolve(true);
};
getRCNotificationInstance(args, function (_ref) {
var prefixCls = _ref.prefixCls,
iconPrefixCls = _ref.iconPrefixCls,
instance = _ref.instance;
instance.notice(getRCNoticeProps((0, _extends2["default"])((0, _extends2["default"])({}, args), {
key: target,
onClose: callback
}), prefixCls, iconPrefixCls));
});
});
var result = function result() {
var _a;
if (messageInstance) {
messageInstance.removeNotice(target);
(_a = args.onClose) === null || _a === void 0 ? void 0 : _a.call(args);
}
};
result.then = function (filled, rejected) {
return closePromise.then(filled, rejected);
};
result.promise = closePromise;
return result;
}
function isArgsProps(content) {
return Object.prototype.toString.call(content) === '[object Object]' && !!content.content;
}
var api = {
open: notice,
config: setMessageConfig,
destroy: function destroy(messageKey) {
if (messageInstance) {
if (messageKey) {
var _messageInstance = messageInstance,
removeNotice = _messageInstance.removeNotice;
removeNotice(messageKey);
} else {
var _messageInstance2 = messageInstance,
destroy = _messageInstance2.destroy;
destroy();
messageInstance = null;
}
}
}
};
function attachTypeApi(originalApi, type) {
originalApi[type] = function (content, duration, onClose) {
if (isArgsProps(content)) {
return originalApi.open((0, _extends2["default"])((0, _extends2["default"])({}, content), {
type: type
}));
}
if (typeof duration === 'function') {
onClose = duration;
duration = undefined;
}
return originalApi.open({
content: content,
duration: duration,
type: type,
onClose: onClose
});
};
}
typeList.forEach(function (type) {
return attachTypeApi(api, type);
});
api.warn = api.warning;
api.useMessage = (0, _useMessage["default"])(getRCNotificationInstance, getRCNoticeProps);
/** @internal test Only function. Not work on production */
var getInstance = exports.getInstance = function getInstance() {
return process.env.NODE_ENV === 'test' ? messageInstance : null;
};
var _default = exports["default"] = api;