"use client"; import _toConsumableArray from "@babel/runtime/helpers/esm/toConsumableArray"; import * as React from 'react'; import { render } from "rc-util/es/React/render"; import ConfigProvider, { globalConfig, warnContext } from '../config-provider'; import PurePanel from './PurePanel'; import useMessage, { useInternalMessage } from './useMessage'; import { wrapPromiseFn } from './util'; let message = null; let act = callback => callback(); let taskQueue = []; let defaultGlobalConfig = {}; function getGlobalContext() { const { prefixCls: globalPrefixCls, getContainer: globalGetContainer, duration, rtl, maxCount, top } = defaultGlobalConfig; const mergedPrefixCls = globalPrefixCls !== null && globalPrefixCls !== void 0 ? globalPrefixCls : globalConfig().getPrefixCls('message'); const mergedContainer = (globalGetContainer === null || globalGetContainer === void 0 ? void 0 : globalGetContainer()) || document.body; return { prefixCls: mergedPrefixCls, getContainer: () => mergedContainer, duration, rtl, maxCount, top }; } const GlobalHolder = /*#__PURE__*/React.forwardRef((_, ref) => { const [messageConfig, setMessageConfig] = React.useState(getGlobalContext); const [api, holder] = useInternalMessage(messageConfig); const global = globalConfig(); const rootPrefixCls = global.getRootPrefixCls(); const rootIconPrefixCls = global.getIconPrefixCls(); const theme = global.getTheme(); const sync = () => { setMessageConfig(getGlobalContext); }; React.useEffect(sync, []); React.useImperativeHandle(ref, () => { const instance = Object.assign({}, api); Object.keys(instance).forEach(method => { instance[method] = function () { sync(); return api[method].apply(api, arguments); }; }); return { instance, sync }; }); return /*#__PURE__*/React.createElement(ConfigProvider, { prefixCls: rootPrefixCls, iconPrefixCls: rootIconPrefixCls, theme: theme }, holder); }); function flushNotice() { if (!message) { const holderFragment = document.createDocumentFragment(); const newMessage = { fragment: holderFragment }; message = newMessage; // Delay render to avoid sync issue act(() => { render( /*#__PURE__*/React.createElement(GlobalHolder, { ref: node => { const { instance, sync } = node || {}; // React 18 test env will throw if call immediately in ref Promise.resolve().then(() => { if (!newMessage.instance && instance) { newMessage.instance = instance; newMessage.sync = sync; flushNotice(); } }); } }), holderFragment); }); return; } // Notification not ready if (!message.instance) { return; } // >>> Execute task taskQueue.forEach(task => { const { type, skipped } = task; // Only `skipped` when user call notice but cancel it immediately // and instance not ready if (!skipped) { switch (type) { case 'open': { act(() => { const closeFn = message.instance.open(Object.assign(Object.assign({}, defaultGlobalConfig), task.config)); closeFn === null || closeFn === void 0 ? void 0 : closeFn.then(task.resolve); task.setCloseFn(closeFn); }); break; } case 'destroy': act(() => { message === null || message === void 0 ? void 0 : message.instance.destroy(task.key); }); break; // Other type open default: { act(() => { var _message$instance; const closeFn = (_message$instance = message.instance)[type].apply(_message$instance, _toConsumableArray(task.args)); closeFn === null || closeFn === void 0 ? void 0 : closeFn.then(task.resolve); task.setCloseFn(closeFn); }); } } } }); // Clean up taskQueue = []; } // ============================================================================== // == Export == // ============================================================================== function setMessageGlobalConfig(config) { defaultGlobalConfig = Object.assign(Object.assign({}, defaultGlobalConfig), config); // Trigger sync for it act(() => { var _a; (_a = message === null || message === void 0 ? void 0 : message.sync) === null || _a === void 0 ? void 0 : _a.call(message); }); } function open(config) { const result = wrapPromiseFn(resolve => { let closeFn; const task = { type: 'open', config, resolve, setCloseFn: fn => { closeFn = fn; } }; taskQueue.push(task); return () => { if (closeFn) { act(() => { closeFn(); }); } else { task.skipped = true; } }; }); flushNotice(); return result; } function typeOpen(type, args) { // Warning if exist theme if (process.env.NODE_ENV !== 'production') { warnContext('message'); } const result = wrapPromiseFn(resolve => { let closeFn; const task = { type, args, resolve, setCloseFn: fn => { closeFn = fn; } }; taskQueue.push(task); return () => { if (closeFn) { act(() => { closeFn(); }); } else { task.skipped = true; } }; }); flushNotice(); return result; } function destroy(key) { taskQueue.push({ type: 'destroy', key }); flushNotice(); } const methods = ['success', 'info', 'warning', 'error', 'loading']; const baseStaticMethods = { open, destroy, config: setMessageGlobalConfig, useMessage, _InternalPanelDoNotUseOrYouWillBeFired: PurePanel }; const staticMethods = baseStaticMethods; methods.forEach(type => { staticMethods[type] = function () { for (var _len = arguments.length, args = new Array(_len), _key = 0; _key < _len; _key++) { args[_key] = arguments[_key]; } return typeOpen(type, args); }; }); // ============================================================================== // == Test == // ============================================================================== const noop = () => {}; /** @internal Only Work in test env */ // eslint-disable-next-line import/no-mutable-exports export let actWrapper = noop; if (process.env.NODE_ENV === 'test') { actWrapper = wrapper => { act = wrapper; }; } /** @internal Only Work in test env */ // eslint-disable-next-line import/no-mutable-exports export let actDestroy = noop; if (process.env.NODE_ENV === 'test') { actDestroy = () => { message = null; }; } export default staticMethods;