mirror of
https://github.com/ant-design/ant-design.git
synced 2024-12-17 02:18:07 +08:00
304 lines
8.0 KiB
TypeScript
304 lines
8.0 KiB
TypeScript
import IconContext from '@ant-design/icons/lib/components/Context';
|
|
import { FormProvider as RcFormProvider } from 'rc-field-form';
|
|
import type { ValidateMessages } from 'rc-field-form/lib/interface';
|
|
import useMemo from 'rc-util/lib/hooks/useMemo';
|
|
import * as React from 'react';
|
|
import type { RequiredMark } from '../form/Form';
|
|
import type { Locale } from '../locale-provider';
|
|
import LocaleProvider, { ANT_MARK } from '../locale-provider';
|
|
import LocaleReceiver from '../locale-provider/LocaleReceiver';
|
|
import defaultLocale from '../locale/default';
|
|
import message from '../message';
|
|
import notification from '../notification';
|
|
import type { Theme } from './context';
|
|
import {
|
|
ConfigConsumer,
|
|
ConfigConsumerProps,
|
|
ConfigContext,
|
|
CSPConfig,
|
|
DirectionType,
|
|
} from './context';
|
|
import { registerTheme } from './cssVariables';
|
|
import { RenderEmptyHandler } from './defaultRenderEmpty';
|
|
import { DisabledContextProvider } from './DisabledContext';
|
|
import type { SizeType } from './SizeContext';
|
|
import SizeContext, { SizeContextProvider } from './SizeContext';
|
|
|
|
export {
|
|
RenderEmptyHandler,
|
|
ConfigContext,
|
|
ConfigConsumer,
|
|
CSPConfig,
|
|
DirectionType,
|
|
ConfigConsumerProps,
|
|
};
|
|
|
|
export const configConsumerProps = [
|
|
'getTargetContainer',
|
|
'getPopupContainer',
|
|
'rootPrefixCls',
|
|
'getPrefixCls',
|
|
'renderEmpty',
|
|
'csp',
|
|
'autoInsertSpaceInButton',
|
|
'locale',
|
|
'pageHeader',
|
|
];
|
|
|
|
// These props is used by `useContext` directly in sub component
|
|
const PASSED_PROPS: Exclude<keyof ConfigConsumerProps, 'rootPrefixCls' | 'getPrefixCls'>[] = [
|
|
'getTargetContainer',
|
|
'getPopupContainer',
|
|
'renderEmpty',
|
|
'pageHeader',
|
|
'input',
|
|
'pagination',
|
|
'form',
|
|
];
|
|
|
|
export interface ConfigProviderProps {
|
|
getTargetContainer?: () => HTMLElement | Window;
|
|
getPopupContainer?: (triggerNode?: HTMLElement) => HTMLElement;
|
|
prefixCls?: string;
|
|
iconPrefixCls?: string;
|
|
children?: React.ReactNode;
|
|
renderEmpty?: RenderEmptyHandler;
|
|
csp?: CSPConfig;
|
|
autoInsertSpaceInButton?: boolean;
|
|
form?: {
|
|
validateMessages?: ValidateMessages;
|
|
requiredMark?: RequiredMark;
|
|
colon?: boolean;
|
|
};
|
|
input?: {
|
|
autoComplete?: string;
|
|
};
|
|
pagination?: {
|
|
showSizeChanger?: boolean;
|
|
};
|
|
locale?: Locale;
|
|
pageHeader?: {
|
|
ghost: boolean;
|
|
};
|
|
componentSize?: SizeType;
|
|
componentDisabled?: boolean;
|
|
direction?: DirectionType;
|
|
space?: {
|
|
size?: SizeType | number;
|
|
};
|
|
virtual?: boolean;
|
|
dropdownMatchSelectWidth?: boolean;
|
|
}
|
|
|
|
interface ProviderChildrenProps extends ConfigProviderProps {
|
|
parentContext: ConfigConsumerProps;
|
|
legacyLocale: Locale;
|
|
}
|
|
|
|
export const defaultPrefixCls = 'ant';
|
|
export const defaultIconPrefixCls = 'anticon';
|
|
let globalPrefixCls: string;
|
|
let globalIconPrefixCls: string;
|
|
|
|
function getGlobalPrefixCls() {
|
|
return globalPrefixCls || defaultPrefixCls;
|
|
}
|
|
|
|
function getGlobalIconPrefixCls() {
|
|
return globalIconPrefixCls || defaultIconPrefixCls;
|
|
}
|
|
|
|
const setGlobalConfig = ({
|
|
prefixCls,
|
|
iconPrefixCls,
|
|
theme,
|
|
}: Pick<ConfigProviderProps, 'prefixCls' | 'iconPrefixCls'> & { theme?: Theme }) => {
|
|
if (prefixCls !== undefined) {
|
|
globalPrefixCls = prefixCls;
|
|
}
|
|
if (iconPrefixCls !== undefined) {
|
|
globalIconPrefixCls = iconPrefixCls;
|
|
}
|
|
|
|
if (theme) {
|
|
registerTheme(getGlobalPrefixCls(), theme);
|
|
}
|
|
};
|
|
|
|
export const globalConfig = () => ({
|
|
getPrefixCls: (suffixCls?: string, customizePrefixCls?: string) => {
|
|
if (customizePrefixCls) return customizePrefixCls;
|
|
return suffixCls ? `${getGlobalPrefixCls()}-${suffixCls}` : getGlobalPrefixCls();
|
|
},
|
|
getIconPrefixCls: getGlobalIconPrefixCls,
|
|
getRootPrefixCls: (rootPrefixCls?: string, customizePrefixCls?: string) => {
|
|
// Customize rootPrefixCls is first priority
|
|
if (rootPrefixCls) {
|
|
return rootPrefixCls;
|
|
}
|
|
|
|
// If Global prefixCls provided, use this
|
|
if (globalPrefixCls) {
|
|
return globalPrefixCls;
|
|
}
|
|
|
|
// [Legacy] If customize prefixCls provided, we cut it to get the prefixCls
|
|
if (customizePrefixCls && customizePrefixCls.includes('-')) {
|
|
return customizePrefixCls.replace(/^(.*)-[^-]*$/, '$1');
|
|
}
|
|
|
|
// Fallback to default prefixCls
|
|
return getGlobalPrefixCls();
|
|
},
|
|
});
|
|
|
|
const ProviderChildren: React.FC<ProviderChildrenProps> = props => {
|
|
const {
|
|
children,
|
|
csp,
|
|
autoInsertSpaceInButton,
|
|
form,
|
|
locale,
|
|
componentSize,
|
|
direction,
|
|
space,
|
|
virtual,
|
|
dropdownMatchSelectWidth,
|
|
legacyLocale,
|
|
parentContext,
|
|
iconPrefixCls,
|
|
componentDisabled,
|
|
} = props;
|
|
|
|
const getPrefixCls = React.useCallback(
|
|
(suffixCls: string, customizePrefixCls?: string) => {
|
|
const { prefixCls } = props;
|
|
|
|
if (customizePrefixCls) return customizePrefixCls;
|
|
|
|
const mergedPrefixCls = prefixCls || parentContext.getPrefixCls('');
|
|
|
|
return suffixCls ? `${mergedPrefixCls}-${suffixCls}` : mergedPrefixCls;
|
|
},
|
|
[parentContext.getPrefixCls, props.prefixCls],
|
|
);
|
|
|
|
const config = {
|
|
...parentContext,
|
|
csp,
|
|
autoInsertSpaceInButton,
|
|
locale: locale || legacyLocale,
|
|
direction,
|
|
space,
|
|
virtual,
|
|
dropdownMatchSelectWidth,
|
|
getPrefixCls,
|
|
};
|
|
|
|
// Pass the props used by `useContext` directly with child component.
|
|
// These props should merged into `config`.
|
|
PASSED_PROPS.forEach(propName => {
|
|
const propValue = props[propName];
|
|
if (propValue) {
|
|
(config as any)[propName] = propValue;
|
|
}
|
|
});
|
|
|
|
// https://github.com/ant-design/ant-design/issues/27617
|
|
const memoedConfig = useMemo(
|
|
() => config,
|
|
config,
|
|
(prevConfig, currentConfig) => {
|
|
const prevKeys = Object.keys(prevConfig) as Array<keyof typeof config>;
|
|
const currentKeys = Object.keys(currentConfig) as Array<keyof typeof config>;
|
|
return (
|
|
prevKeys.length !== currentKeys.length ||
|
|
prevKeys.some(key => prevConfig[key] !== currentConfig[key])
|
|
);
|
|
},
|
|
);
|
|
|
|
const memoIconContextValue = React.useMemo(
|
|
() => ({ prefixCls: iconPrefixCls, csp }),
|
|
[iconPrefixCls, csp],
|
|
);
|
|
|
|
let childNode = children;
|
|
// Additional Form provider
|
|
let validateMessages: ValidateMessages = {};
|
|
|
|
if (locale) {
|
|
validateMessages =
|
|
locale.Form?.defaultValidateMessages || defaultLocale.Form?.defaultValidateMessages || {};
|
|
}
|
|
if (form && form.validateMessages) {
|
|
validateMessages = { ...validateMessages, ...form.validateMessages };
|
|
}
|
|
|
|
if (Object.keys(validateMessages).length > 0) {
|
|
childNode = <RcFormProvider validateMessages={validateMessages}>{children}</RcFormProvider>;
|
|
}
|
|
|
|
if (locale) {
|
|
childNode = (
|
|
<LocaleProvider locale={locale} _ANT_MARK__={ANT_MARK}>
|
|
{childNode}
|
|
</LocaleProvider>
|
|
);
|
|
}
|
|
|
|
if (iconPrefixCls || csp) {
|
|
childNode = (
|
|
<IconContext.Provider value={memoIconContextValue}>{childNode}</IconContext.Provider>
|
|
);
|
|
}
|
|
|
|
if (componentSize) {
|
|
childNode = <SizeContextProvider size={componentSize}>{childNode}</SizeContextProvider>;
|
|
}
|
|
|
|
if (componentDisabled !== undefined) {
|
|
childNode = (
|
|
<DisabledContextProvider disabled={componentDisabled}>{childNode}</DisabledContextProvider>
|
|
);
|
|
}
|
|
|
|
return <ConfigContext.Provider value={memoedConfig}>{childNode}</ConfigContext.Provider>;
|
|
};
|
|
|
|
const ConfigProvider: React.FC<ConfigProviderProps> & {
|
|
ConfigContext: typeof ConfigContext;
|
|
SizeContext: typeof SizeContext;
|
|
config: typeof setGlobalConfig;
|
|
} = props => {
|
|
React.useEffect(() => {
|
|
if (props.direction) {
|
|
message.config({
|
|
rtl: props.direction === 'rtl',
|
|
});
|
|
notification.config({
|
|
rtl: props.direction === 'rtl',
|
|
});
|
|
}
|
|
}, [props.direction]);
|
|
|
|
return (
|
|
<LocaleReceiver>
|
|
{(_, __, legacyLocale) => (
|
|
<ConfigConsumer>
|
|
{context => (
|
|
<ProviderChildren parentContext={context} legacyLocale={legacyLocale} {...props} />
|
|
)}
|
|
</ConfigConsumer>
|
|
)}
|
|
</LocaleReceiver>
|
|
);
|
|
};
|
|
|
|
/** @private internal Usage. do not use in your production */
|
|
ConfigProvider.ConfigContext = ConfigContext;
|
|
ConfigProvider.SizeContext = SizeContext;
|
|
ConfigProvider.config = setGlobalConfig;
|
|
|
|
export default ConfigProvider;
|