mirror of
https://gitee.com/ant-design/ant-design.git
synced 2024-12-05 21:48:19 +08:00
e985f8e845
Signed-off-by: Rainey <xuebing_orz@icloud.com>
84 lines
2.5 KiB
TypeScript
84 lines
2.5 KiB
TypeScript
import classNames from 'classnames';
|
|
import type { ReactNode } from 'react';
|
|
import React, { useContext } from 'react';
|
|
import type { ConfigConsumerProps } from '../config-provider';
|
|
import { ConfigContext } from '../config-provider';
|
|
import useMessage from '../message/useMessage';
|
|
import useModal from '../modal/useModal';
|
|
import useNotification from '../notification/useNotification';
|
|
import type { AppConfig, useAppProps } from './context';
|
|
import AppContext, { AppConfigContext } from './context';
|
|
import useStyle from './style';
|
|
|
|
export interface AppProps extends AppConfig {
|
|
style?: React.CSSProperties;
|
|
className?: string;
|
|
rootClassName?: string;
|
|
prefixCls?: string;
|
|
children?: ReactNode;
|
|
}
|
|
|
|
const useApp = () => React.useContext<useAppProps>(AppContext);
|
|
|
|
const App: React.FC<AppProps> & { useApp: typeof useApp } = (props) => {
|
|
const {
|
|
prefixCls: customizePrefixCls,
|
|
children,
|
|
className,
|
|
rootClassName,
|
|
message,
|
|
notification,
|
|
style,
|
|
} = props;
|
|
const { getPrefixCls } = useContext<ConfigConsumerProps>(ConfigContext);
|
|
const prefixCls = getPrefixCls('app', customizePrefixCls);
|
|
const [wrapSSR, hashId] = useStyle(prefixCls);
|
|
const customClassName = classNames(hashId, prefixCls, className, rootClassName);
|
|
|
|
const appConfig = useContext<AppConfig>(AppConfigContext);
|
|
|
|
const mergedAppConfig = React.useMemo<AppConfig>(
|
|
() => ({
|
|
message: { ...appConfig.message, ...message },
|
|
notification: { ...appConfig.notification, ...notification },
|
|
}),
|
|
[message, notification, appConfig.message, appConfig.notification],
|
|
);
|
|
|
|
const [messageApi, messageContextHolder] = useMessage(mergedAppConfig.message);
|
|
const [notificationApi, notificationContextHolder] = useNotification(
|
|
mergedAppConfig.notification,
|
|
);
|
|
const [ModalApi, ModalContextHolder] = useModal();
|
|
|
|
const memoizedContextValue = React.useMemo<useAppProps>(
|
|
() => ({
|
|
message: messageApi,
|
|
notification: notificationApi,
|
|
modal: ModalApi,
|
|
}),
|
|
[messageApi, notificationApi, ModalApi],
|
|
);
|
|
|
|
return wrapSSR(
|
|
<AppContext.Provider value={memoizedContextValue}>
|
|
<AppConfigContext.Provider value={mergedAppConfig}>
|
|
<div className={customClassName} style={style}>
|
|
{ModalContextHolder}
|
|
{messageContextHolder}
|
|
{notificationContextHolder}
|
|
{children}
|
|
</div>
|
|
</AppConfigContext.Provider>
|
|
</AppContext.Provider>,
|
|
);
|
|
};
|
|
|
|
if (process.env.NODE_ENV !== 'production') {
|
|
App.displayName = 'App';
|
|
}
|
|
|
|
App.useApp = useApp;
|
|
|
|
export default App;
|