mirror of
https://github.com/ant-design/ant-design.git
synced 2024-12-13 15:49:10 +08:00
313 lines
8.3 KiB
TypeScript
313 lines
8.3 KiB
TypeScript
import { createTheme } from '@ant-design/cssinjs';
|
|
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 { DesignTokenContext } from '../theme';
|
|
import defaultSeedToken from '../theme/themes/seed';
|
|
import type { ConfigConsumerProps, CSPConfig, DirectionType, Theme, ThemeConfig } from './context';
|
|
import { ConfigConsumer, ConfigContext, defaultIconPrefixCls } from './context';
|
|
import { registerTheme } from './cssVariables';
|
|
import { RenderEmptyHandler } from './defaultRenderEmpty';
|
|
import { DisabledContextProvider } from './DisabledContext';
|
|
import useTheme from './hooks/useTheme';
|
|
import type { SizeType } from './SizeContext';
|
|
import SizeContext, { SizeContextProvider } from './SizeContext';
|
|
|
|
export {
|
|
RenderEmptyHandler,
|
|
ConfigContext,
|
|
ConfigConsumer,
|
|
CSPConfig,
|
|
DirectionType,
|
|
ConfigConsumerProps,
|
|
};
|
|
export { defaultIconPrefixCls };
|
|
|
|
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;
|
|
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;
|
|
theme?: ThemeConfig;
|
|
}
|
|
|
|
interface ProviderChildrenProps extends ConfigProviderProps {
|
|
parentContext: ConfigConsumerProps;
|
|
legacyLocale: Locale;
|
|
}
|
|
|
|
export const defaultPrefixCls = 'ant';
|
|
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: () => {
|
|
// If Global prefixCls provided, use this
|
|
if (globalPrefixCls) {
|
|
return globalPrefixCls;
|
|
}
|
|
|
|
// 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,
|
|
theme,
|
|
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 mergedTheme = useTheme(theme, parentContext.theme);
|
|
|
|
const config = {
|
|
...parentContext,
|
|
csp,
|
|
autoInsertSpaceInButton,
|
|
locale: locale || legacyLocale,
|
|
direction,
|
|
space,
|
|
virtual,
|
|
dropdownMatchSelectWidth,
|
|
getPrefixCls,
|
|
iconPrefixCls: iconPrefixCls ?? defaultIconPrefixCls,
|
|
theme: mergedTheme,
|
|
};
|
|
|
|
// Pass the props used by `useContext` directly with child component.
|
|
// These props should merged into `config`.
|
|
PASSED_PROPS.forEach(propName => {
|
|
const propValue: any = props[propName];
|
|
if (propValue) {
|
|
(config as any)[propName] = propValue;
|
|
}
|
|
});
|
|
|
|
// https://github.com/ant-design/ant-design/issues/27617
|
|
const memoedConfig = useMemo(
|
|
() => config,
|
|
config,
|
|
(prevConfig: Record<string, any>, currentConfig) => {
|
|
const prevKeys = Object.keys(prevConfig);
|
|
const currentKeys = Object.keys(currentConfig);
|
|
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>;
|
|
}
|
|
|
|
// ================================ Dynamic theme ================================
|
|
const memoTheme = React.useMemo(() => {
|
|
const { algorithm, token, ...rest } = mergedTheme || {};
|
|
const themeObj =
|
|
algorithm && (!Array.isArray(algorithm) || algorithm.length > 0)
|
|
? createTheme(algorithm)
|
|
: undefined;
|
|
|
|
return {
|
|
...rest,
|
|
theme: themeObj,
|
|
|
|
token: {
|
|
...defaultSeedToken,
|
|
...token,
|
|
},
|
|
};
|
|
}, [mergedTheme]);
|
|
|
|
if (theme) {
|
|
childNode = (
|
|
<DesignTokenContext.Provider value={memoTheme}>{childNode}</DesignTokenContext.Provider>
|
|
);
|
|
}
|
|
|
|
// =================================== Render ===================================
|
|
if (componentDisabled !== undefined) {
|
|
childNode = (
|
|
<DisabledContextProvider disabled={componentDisabled}>{childNode}</DisabledContextProvider>
|
|
);
|
|
}
|
|
|
|
return <ConfigContext.Provider value={memoedConfig}>{childNode}</ConfigContext.Provider>;
|
|
};
|
|
|
|
const ConfigProvider: React.FC<ConfigProviderProps> & {
|
|
/** @private internal Usage. do not use in your production */
|
|
ConfigContext: typeof ConfigContext;
|
|
SizeContext: typeof SizeContext;
|
|
config: typeof setGlobalConfig;
|
|
} = props => (
|
|
<LocaleReceiver>
|
|
{(_, __, legacyLocale) => (
|
|
<ConfigConsumer>
|
|
{context => (
|
|
<ProviderChildren
|
|
parentContext={context}
|
|
legacyLocale={legacyLocale as Locale}
|
|
{...props}
|
|
/>
|
|
)}
|
|
</ConfigConsumer>
|
|
)}
|
|
</LocaleReceiver>
|
|
);
|
|
|
|
ConfigProvider.ConfigContext = ConfigContext;
|
|
ConfigProvider.SizeContext = SizeContext;
|
|
ConfigProvider.config = setGlobalConfig;
|
|
|
|
export default ConfigProvider;
|