mirror of
https://github.com/ant-design/ant-design.git
synced 2024-12-05 09:49:57 +08:00
e613861e63
* test driven * use ref for the same content
74 lines
2.2 KiB
TypeScript
74 lines
2.2 KiB
TypeScript
import * as React from 'react';
|
|
import useRCNotification from 'rc-notification/lib/useNotification';
|
|
import {
|
|
NotificationInstance as RCNotificationInstance,
|
|
NoticeContent as RCNoticeContent,
|
|
HolderReadyCallback as RCHolderReadyCallback,
|
|
} from 'rc-notification/lib/Notification';
|
|
import { ConfigConsumer, ConfigConsumerProps } from '../../config-provider';
|
|
import { NotificationInstance, ArgsProps } from '..';
|
|
|
|
export default function createUseNotification(
|
|
getNotificationInstance: (
|
|
args: ArgsProps,
|
|
callback: (info: { prefixCls: string; instance: RCNotificationInstance }) => void,
|
|
) => void,
|
|
getRCNoticeProps: (args: ArgsProps, prefixCls: string) => RCNoticeContent,
|
|
) {
|
|
const useNotification = (): [NotificationInstance, React.ReactElement] => {
|
|
// We can only get content by render
|
|
let getPrefixCls: ConfigConsumerProps['getPrefixCls'];
|
|
|
|
// We create a proxy to handle delay created instance
|
|
let innerInstance: RCNotificationInstance | null = null;
|
|
const proxy = {
|
|
add: (noticeProps: RCNoticeContent, holderCallback?: RCHolderReadyCallback) => {
|
|
innerInstance?.component.add(noticeProps, holderCallback);
|
|
},
|
|
} as any;
|
|
|
|
const [hookNotify, holder] = useRCNotification(proxy);
|
|
|
|
function notify(args: ArgsProps) {
|
|
const { prefixCls: customizePrefixCls } = args;
|
|
const mergedPrefixCls = getPrefixCls('notification', customizePrefixCls);
|
|
|
|
getNotificationInstance(
|
|
{
|
|
...args,
|
|
prefixCls: mergedPrefixCls,
|
|
},
|
|
({ prefixCls, instance }) => {
|
|
innerInstance = instance;
|
|
hookNotify(getRCNoticeProps(args, prefixCls));
|
|
},
|
|
);
|
|
}
|
|
|
|
// Fill functions
|
|
const hookApiRef = React.useRef<any>({});
|
|
|
|
hookApiRef.current.open = notify;
|
|
|
|
['success', 'info', 'warning', 'error'].forEach(type => {
|
|
hookApiRef.current[type] = (args: ArgsProps) =>
|
|
hookApiRef.current.open({
|
|
...args,
|
|
type,
|
|
});
|
|
});
|
|
|
|
return [
|
|
hookApiRef.current,
|
|
<ConfigConsumer key="holder">
|
|
{(context: ConfigConsumerProps) => {
|
|
({ getPrefixCls } = context);
|
|
return holder;
|
|
}}
|
|
</ConfigConsumer>,
|
|
];
|
|
};
|
|
|
|
return useNotification;
|
|
}
|