antd
Version:
An enterprise-class UI design language and React components implementation
189 lines (188 loc) • 7.32 kB
JavaScript
"use client";
var __rest = this && this.__rest || function (s, e) {
var t = {};
for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0) t[p] = s[p];
if (s != null && typeof Object.getOwnPropertySymbols === "function") for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i])) t[p[i]] = s[p[i]];
}
return t;
};
import React, { useContext } from 'react';
import classNames from 'classnames';
import { NotificationProvider, useNotification as useRcNotification } from 'rc-notification';
import { devUseWarning } from '../_util/warning';
import { ConfigContext } from '../config-provider';
import useCSSVarCls from '../config-provider/hooks/useCSSVarCls';
import { useToken } from '../theme/internal';
import { getCloseIcon, PureContent } from './PurePanel';
import useStyle from './style';
import { getMotion, getPlacementStyle } from './util';
const DEFAULT_OFFSET = 24;
const DEFAULT_DURATION = 4.5;
const DEFAULT_PLACEMENT = 'topRight';
const Wrapper = _ref => {
let {
children,
prefixCls
} = _ref;
const rootCls = useCSSVarCls(prefixCls);
const [wrapCSSVar, hashId, cssVarCls] = useStyle(prefixCls, rootCls);
return wrapCSSVar(/*#__PURE__*/React.createElement(NotificationProvider, {
classNames: {
list: classNames(hashId, cssVarCls, rootCls)
}
}, children));
};
const renderNotifications = (node, _ref2) => {
let {
prefixCls,
key
} = _ref2;
return /*#__PURE__*/React.createElement(Wrapper, {
prefixCls: prefixCls,
key: key
}, node);
};
const Holder = /*#__PURE__*/React.forwardRef((props, ref) => {
const {
top,
bottom,
prefixCls: staticPrefixCls,
getContainer: staticGetContainer,
maxCount,
rtl,
onAllRemoved,
stack,
duration,
pauseOnHover = true,
showProgress
} = props;
const {
getPrefixCls,
getPopupContainer,
notification,
direction
} = useContext(ConfigContext);
const [, token] = useToken();
const prefixCls = staticPrefixCls || getPrefixCls('notification');
// =============================== Style ===============================
const getStyle = placement => getPlacementStyle(placement, top !== null && top !== void 0 ? top : DEFAULT_OFFSET, bottom !== null && bottom !== void 0 ? bottom : DEFAULT_OFFSET);
const getClassName = () => classNames({
[`${prefixCls}-rtl`]: rtl !== null && rtl !== void 0 ? rtl : direction === 'rtl'
});
// ============================== Motion ===============================
const getNotificationMotion = () => getMotion(prefixCls);
// ============================== Origin ===============================
const [api, holder] = useRcNotification({
prefixCls,
style: getStyle,
className: getClassName,
motion: getNotificationMotion,
closable: true,
closeIcon: getCloseIcon(prefixCls),
duration: duration !== null && duration !== void 0 ? duration : DEFAULT_DURATION,
getContainer: () => (staticGetContainer === null || staticGetContainer === void 0 ? void 0 : staticGetContainer()) || (getPopupContainer === null || getPopupContainer === void 0 ? void 0 : getPopupContainer()) || document.body,
maxCount,
pauseOnHover,
showProgress,
onAllRemoved,
renderNotifications,
stack: stack === false ? false : {
threshold: typeof stack === 'object' ? stack === null || stack === void 0 ? void 0 : stack.threshold : undefined,
offset: 8,
gap: token.margin
}
});
// ================================ Ref ================================
React.useImperativeHandle(ref, () => Object.assign(Object.assign({}, api), {
prefixCls,
notification
}));
return holder;
});
// ==============================================================================
// == Hook ==
// ==============================================================================
export function useInternalNotification(notificationConfig) {
const holderRef = React.useRef(null);
const warning = devUseWarning('Notification');
// ================================ API ================================
const wrapAPI = React.useMemo(() => {
// Wrap with notification content
// >>> Open
const open = config => {
var _a;
if (!holderRef.current) {
process.env.NODE_ENV !== "production" ? warning(false, 'usage', 'You are calling notice in render which will break in React 18 concurrent mode. Please trigger in effect instead.') : void 0;
return;
}
const {
open: originOpen,
prefixCls,
notification
} = holderRef.current;
const noticePrefixCls = `${prefixCls}-notice`;
const {
message,
description,
icon,
type,
btn,
className,
style,
role = 'alert',
closeIcon,
closable
} = config,
restConfig = __rest(config, ["message", "description", "icon", "type", "btn", "className", "style", "role", "closeIcon", "closable"]);
const realCloseIcon = getCloseIcon(noticePrefixCls, typeof closeIcon !== 'undefined' ? closeIcon : notification === null || notification === void 0 ? void 0 : notification.closeIcon);
return originOpen(Object.assign(Object.assign({
// use placement from props instead of hard-coding "topRight"
placement: (_a = notificationConfig === null || notificationConfig === void 0 ? void 0 : notificationConfig.placement) !== null && _a !== void 0 ? _a : DEFAULT_PLACEMENT
}, restConfig), {
content: (/*#__PURE__*/React.createElement(PureContent, {
prefixCls: noticePrefixCls,
icon: icon,
type: type,
message: message,
description: description,
btn: btn,
role: role
})),
className: classNames(type && `${noticePrefixCls}-${type}`, className, notification === null || notification === void 0 ? void 0 : notification.className),
style: Object.assign(Object.assign({}, notification === null || notification === void 0 ? void 0 : notification.style), style),
closeIcon: realCloseIcon,
closable: closable !== null && closable !== void 0 ? closable : !!realCloseIcon
}));
};
// >>> destroy
const destroy = key => {
var _a, _b;
if (key !== undefined) {
(_a = holderRef.current) === null || _a === void 0 ? void 0 : _a.close(key);
} else {
(_b = holderRef.current) === null || _b === void 0 ? void 0 : _b.destroy();
}
};
const clone = {
open,
destroy
};
const keys = ['success', 'info', 'warning', 'error'];
keys.forEach(type => {
clone[type] = config => open(Object.assign(Object.assign({}, config), {
type
}));
});
return clone;
}, []);
// ============================== Return ===============================
return [wrapAPI, /*#__PURE__*/React.createElement(Holder, Object.assign({
key: "notification-holder"
}, notificationConfig, {
ref: holderRef
}))];
}
export default function useNotification(notificationConfig) {
return useInternalNotification(notificationConfig);
}