index.js 6.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190
  1. "use strict";
  2. var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
  3. Object.defineProperty(exports, "__esModule", {
  4. value: true
  5. });
  6. exports.attachTypeApi = attachTypeApi;
  7. exports.getInstance = exports.default = void 0;
  8. exports.getKeyThenIncreaseKey = getKeyThenIncreaseKey;
  9. exports.typeList = void 0;
  10. var _vue = require("vue");
  11. var _extends2 = _interopRequireDefault(require("@babel/runtime/helpers/extends"));
  12. var _vcNotification = _interopRequireDefault(require("../vc-notification"));
  13. var _LoadingOutlined = _interopRequireDefault(require("@ant-design/icons-vue/lib/icons/LoadingOutlined"));
  14. var _ExclamationCircleFilled = _interopRequireDefault(require("@ant-design/icons-vue/lib/icons/ExclamationCircleFilled"));
  15. var _CloseCircleFilled = _interopRequireDefault(require("@ant-design/icons-vue/lib/icons/CloseCircleFilled"));
  16. var _CheckCircleFilled = _interopRequireDefault(require("@ant-design/icons-vue/lib/icons/CheckCircleFilled"));
  17. var _InfoCircleFilled = _interopRequireDefault(require("@ant-design/icons-vue/lib/icons/InfoCircleFilled"));
  18. var _classNames = _interopRequireDefault(require("../_util/classNames"));
  19. var _style = _interopRequireDefault(require("./style"));
  20. var _useMessage = _interopRequireDefault(require("./useMessage"));
  21. let defaultDuration = 3;
  22. let defaultTop;
  23. let messageInstance;
  24. let key = 1;
  25. let localPrefixCls = '';
  26. let transitionName = 'move-up';
  27. let hasTransitionName = false;
  28. let getContainer = () => document.body;
  29. let maxCount;
  30. let rtl = false;
  31. function getKeyThenIncreaseKey() {
  32. return key++;
  33. }
  34. function setMessageConfig(options) {
  35. if (options.top !== undefined) {
  36. defaultTop = options.top;
  37. messageInstance = null; // delete messageInstance for new defaultTop
  38. }
  39. if (options.duration !== undefined) {
  40. defaultDuration = options.duration;
  41. }
  42. if (options.prefixCls !== undefined) {
  43. localPrefixCls = options.prefixCls;
  44. }
  45. if (options.getContainer !== undefined) {
  46. getContainer = options.getContainer;
  47. messageInstance = null; // delete messageInstance for new getContainer
  48. }
  49. if (options.transitionName !== undefined) {
  50. transitionName = options.transitionName;
  51. messageInstance = null; // delete messageInstance for new transitionName
  52. hasTransitionName = true;
  53. }
  54. if (options.maxCount !== undefined) {
  55. maxCount = options.maxCount;
  56. messageInstance = null;
  57. }
  58. if (options.rtl !== undefined) {
  59. rtl = options.rtl;
  60. }
  61. }
  62. function getMessageInstance(args, callback) {
  63. if (messageInstance) {
  64. callback(messageInstance);
  65. return;
  66. }
  67. _vcNotification.default.newInstance({
  68. appContext: args.appContext,
  69. prefixCls: args.prefixCls || localPrefixCls,
  70. rootPrefixCls: args.rootPrefixCls,
  71. transitionName,
  72. hasTransitionName,
  73. style: {
  74. top: defaultTop
  75. },
  76. getContainer: getContainer || args.getPopupContainer,
  77. maxCount,
  78. name: 'message',
  79. useStyle: _style.default
  80. }, instance => {
  81. if (messageInstance) {
  82. callback(messageInstance);
  83. return;
  84. }
  85. messageInstance = instance;
  86. callback(instance);
  87. });
  88. }
  89. const typeToIcon = {
  90. info: _InfoCircleFilled.default,
  91. success: _CheckCircleFilled.default,
  92. error: _CloseCircleFilled.default,
  93. warning: _ExclamationCircleFilled.default,
  94. loading: _LoadingOutlined.default
  95. };
  96. const typeList = exports.typeList = Object.keys(typeToIcon);
  97. function notice(args) {
  98. const duration = args.duration !== undefined ? args.duration : defaultDuration;
  99. const target = args.key || getKeyThenIncreaseKey();
  100. const closePromise = new Promise(resolve => {
  101. const callback = () => {
  102. if (typeof args.onClose === 'function') {
  103. args.onClose();
  104. }
  105. return resolve(true);
  106. };
  107. getMessageInstance(args, instance => {
  108. instance.notice({
  109. key: target,
  110. duration,
  111. style: args.style || {},
  112. class: args.class,
  113. content: _ref => {
  114. let {
  115. prefixCls
  116. } = _ref;
  117. const Icon = typeToIcon[args.type];
  118. const iconNode = Icon ? (0, _vue.createVNode)(Icon, null, null) : '';
  119. const messageClass = (0, _classNames.default)(`${prefixCls}-custom-content`, {
  120. [`${prefixCls}-${args.type}`]: args.type,
  121. [`${prefixCls}-rtl`]: rtl === true
  122. });
  123. return (0, _vue.createVNode)("div", {
  124. "class": messageClass
  125. }, [typeof args.icon === 'function' ? args.icon() : args.icon || iconNode, (0, _vue.createVNode)("span", null, [typeof args.content === 'function' ? args.content() : args.content])]);
  126. },
  127. onClose: callback,
  128. onClick: args.onClick
  129. });
  130. });
  131. });
  132. const result = () => {
  133. if (messageInstance) {
  134. messageInstance.removeNotice(target);
  135. }
  136. };
  137. result.then = (filled, rejected) => closePromise.then(filled, rejected);
  138. result.promise = closePromise;
  139. return result;
  140. }
  141. function isArgsProps(content) {
  142. return Object.prototype.toString.call(content) === '[object Object]' && !!content.content;
  143. }
  144. const api = {
  145. open: notice,
  146. config: setMessageConfig,
  147. destroy(messageKey) {
  148. if (messageInstance) {
  149. if (messageKey) {
  150. const {
  151. removeNotice
  152. } = messageInstance;
  153. removeNotice(messageKey);
  154. } else {
  155. const {
  156. destroy
  157. } = messageInstance;
  158. destroy();
  159. messageInstance = null;
  160. }
  161. }
  162. }
  163. };
  164. function attachTypeApi(originalApi, type) {
  165. originalApi[type] = (content, duration, onClose) => {
  166. if (isArgsProps(content)) {
  167. return originalApi.open((0, _extends2.default)((0, _extends2.default)({}, content), {
  168. type
  169. }));
  170. }
  171. if (typeof duration === 'function') {
  172. onClose = duration;
  173. duration = undefined;
  174. }
  175. return originalApi.open({
  176. content,
  177. duration,
  178. type,
  179. onClose
  180. });
  181. };
  182. }
  183. typeList.forEach(type => attachTypeApi(api, type));
  184. api.warn = api.warning;
  185. api.useMessage = _useMessage.default;
  186. /** @private test Only function. Not work on production */
  187. const getInstance = () => process.env.NODE_ENV === 'test' ? messageInstance : null;
  188. exports.getInstance = getInstance;
  189. var _default = exports.default = api;