243 lines
10 KiB
JavaScript
243 lines
10 KiB
JavaScript
import _typeof from "@babel/runtime/helpers/esm/typeof";
|
|
import _slicedToArray from "@babel/runtime/helpers/esm/slicedToArray";
|
|
import _toConsumableArray from "@babel/runtime/helpers/esm/toConsumableArray";
|
|
import _defineProperty from "@babel/runtime/helpers/esm/defineProperty";
|
|
import _objectSpread from "@babel/runtime/helpers/esm/objectSpread2";
|
|
import { useMemo } from 'react';
|
|
import { token2CSSVar, useCSSVarRegister, useStyleRegister } from '@ant-design/cssinjs';
|
|
import genCalc from "./calc";
|
|
import getCompVarPrefix from "./getCompVarPrefix";
|
|
import getComponentToken from "./getComponentToken";
|
|
import getDefaultComponentToken from "./getDefaultComponentToken";
|
|
import genMaxMin from "./maxmin";
|
|
import statisticToken, { merge as mergeToken } from "./statistic";
|
|
import useUniqueMemo from "../_util/hooks/useUniqueMemo";
|
|
import useDefaultCSP from "../hooks/useCSP";
|
|
function genStyleUtils(config) {
|
|
// Dependency inversion for preparing basic config.
|
|
var _config$useCSP = config.useCSP,
|
|
useCSP = _config$useCSP === void 0 ? useDefaultCSP : _config$useCSP,
|
|
useToken = config.useToken,
|
|
usePrefix = config.usePrefix,
|
|
getResetStyles = config.getResetStyles,
|
|
getCommonStyle = config.getCommonStyle,
|
|
getCompUnitless = config.getCompUnitless;
|
|
function genStyleHooks(component, styleFn, getDefaultToken, options) {
|
|
var componentName = Array.isArray(component) ? component[0] : component;
|
|
function prefixToken(key) {
|
|
return "".concat(String(componentName)).concat(key.slice(0, 1).toUpperCase()).concat(key.slice(1));
|
|
}
|
|
|
|
// Fill unitless
|
|
var originUnitless = (options === null || options === void 0 ? void 0 : options.unitless) || {};
|
|
var originCompUnitless = typeof getCompUnitless === 'function' ? getCompUnitless(component) : {};
|
|
var compUnitless = _objectSpread(_objectSpread({}, originCompUnitless), {}, _defineProperty({}, prefixToken('zIndexPopup'), true));
|
|
Object.keys(originUnitless).forEach(function (key) {
|
|
compUnitless[prefixToken(key)] = originUnitless[key];
|
|
});
|
|
|
|
// Options
|
|
var mergedOptions = _objectSpread(_objectSpread({}, options), {}, {
|
|
unitless: compUnitless,
|
|
prefixToken: prefixToken
|
|
});
|
|
|
|
// Hooks
|
|
var useStyle = genComponentStyleHook(component, styleFn, getDefaultToken, mergedOptions);
|
|
var useCSSVar = genCSSVarRegister(componentName, getDefaultToken, mergedOptions);
|
|
return function (prefixCls) {
|
|
var rootCls = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : prefixCls;
|
|
var hashId = useStyle(prefixCls, rootCls);
|
|
|
|
// Resolve function type to get dynamic extra prefix
|
|
var extraPrefixCls = options === null || options === void 0 ? void 0 : options.extraCssVarPrefixCls;
|
|
var resolvedExtraPrefixCls = typeof extraPrefixCls === 'function' ? extraPrefixCls({
|
|
prefixCls: prefixCls,
|
|
rootCls: rootCls
|
|
}) : extraPrefixCls;
|
|
var cssVarCls = useCSSVar(resolvedExtraPrefixCls !== null && resolvedExtraPrefixCls !== void 0 && resolvedExtraPrefixCls.length ? [rootCls].concat(_toConsumableArray(resolvedExtraPrefixCls)) : rootCls);
|
|
return [hashId, cssVarCls];
|
|
};
|
|
}
|
|
function genCSSVarRegister(component, getDefaultToken, options) {
|
|
var compUnitless = options.unitless,
|
|
prefixToken = options.prefixToken,
|
|
ignore = options.ignore;
|
|
return function (rootCls) {
|
|
var _useToken = useToken(),
|
|
cssVar = _useToken.cssVar,
|
|
realToken = _useToken.realToken;
|
|
var csp = useCSP();
|
|
useCSSVarRegister({
|
|
path: [component],
|
|
prefix: cssVar.prefix,
|
|
key: cssVar.key,
|
|
unitless: compUnitless,
|
|
ignore: ignore,
|
|
token: realToken,
|
|
scope: rootCls,
|
|
nonce: function nonce() {
|
|
return csp.nonce;
|
|
}
|
|
}, function () {
|
|
var defaultToken = getDefaultComponentToken(component, realToken, getDefaultToken);
|
|
var componentToken = getComponentToken(component, realToken, defaultToken, {
|
|
deprecatedTokens: options === null || options === void 0 ? void 0 : options.deprecatedTokens
|
|
});
|
|
if (defaultToken) {
|
|
Object.keys(defaultToken).forEach(function (key) {
|
|
componentToken[prefixToken(key)] = componentToken[key];
|
|
delete componentToken[key];
|
|
});
|
|
}
|
|
return componentToken;
|
|
});
|
|
return cssVar === null || cssVar === void 0 ? void 0 : cssVar.key;
|
|
};
|
|
}
|
|
function genComponentStyleHook(componentName, styleFn, getDefaultToken) {
|
|
var options = arguments.length > 3 && arguments[3] !== undefined ? arguments[3] : {};
|
|
var cells = Array.isArray(componentName) ? componentName : [componentName, componentName];
|
|
var _cells = _slicedToArray(cells, 1),
|
|
component = _cells[0];
|
|
var concatComponent = cells.join('-');
|
|
var mergedLayer = config.layer || {
|
|
name: 'antd'
|
|
};
|
|
|
|
// Return new style hook
|
|
return function (prefixCls) {
|
|
var rootCls = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : prefixCls;
|
|
var _useToken2 = useToken(),
|
|
theme = _useToken2.theme,
|
|
realToken = _useToken2.realToken,
|
|
hashId = _useToken2.hashId,
|
|
token = _useToken2.token,
|
|
cssVar = _useToken2.cssVar,
|
|
zeroRuntime = _useToken2.zeroRuntime;
|
|
|
|
// Update of `disabledRuntimeStyle` would cause React hook error, so memoized it and never update.
|
|
var memoizedZeroRuntime = useMemo(function () {
|
|
return zeroRuntime;
|
|
}, []);
|
|
if (memoizedZeroRuntime) {
|
|
return hashId;
|
|
}
|
|
var _usePrefix = usePrefix(),
|
|
rootPrefixCls = _usePrefix.rootPrefixCls,
|
|
iconPrefixCls = _usePrefix.iconPrefixCls;
|
|
var csp = useCSP();
|
|
var type = 'css';
|
|
|
|
// Use unique memo to share the result across all instances
|
|
var calc = useUniqueMemo(function () {
|
|
var unitlessCssVar = new Set();
|
|
Object.keys(options.unitless || {}).forEach(function (key) {
|
|
// Some component proxy the AliasToken (e.g. Image) and some not (e.g. Modal)
|
|
// We should both pass in `unitlessCssVar` to make sure the CSSVar can be unitless.
|
|
unitlessCssVar.add(token2CSSVar(key, cssVar.prefix));
|
|
unitlessCssVar.add(token2CSSVar(key, getCompVarPrefix(component, cssVar.prefix)));
|
|
});
|
|
return genCalc(type, unitlessCssVar);
|
|
}, [type, component, cssVar === null || cssVar === void 0 ? void 0 : cssVar.prefix]);
|
|
var _genMaxMin = genMaxMin(type),
|
|
max = _genMaxMin.max,
|
|
min = _genMaxMin.min;
|
|
|
|
// Shared config
|
|
var sharedConfig = {
|
|
theme: theme,
|
|
token: token,
|
|
hashId: hashId,
|
|
nonce: function nonce() {
|
|
return csp.nonce;
|
|
},
|
|
clientOnly: options.clientOnly,
|
|
layer: mergedLayer,
|
|
// antd is always at top of styles
|
|
order: options.order || -999
|
|
};
|
|
|
|
// This if statement is safe, as it will only be used if the generator has the function. It's not dynamic.
|
|
if (typeof getResetStyles === 'function') {
|
|
// Generate style for all need reset tags.
|
|
useStyleRegister(_objectSpread(_objectSpread({}, sharedConfig), {}, {
|
|
clientOnly: false,
|
|
path: ['Shared', rootPrefixCls]
|
|
}), function () {
|
|
return getResetStyles(token, {
|
|
prefix: {
|
|
rootPrefixCls: rootPrefixCls,
|
|
iconPrefixCls: iconPrefixCls
|
|
},
|
|
csp: csp
|
|
});
|
|
});
|
|
}
|
|
useStyleRegister(_objectSpread(_objectSpread({}, sharedConfig), {}, {
|
|
path: [concatComponent, prefixCls, iconPrefixCls]
|
|
}), function () {
|
|
if (options.injectStyle === false) {
|
|
return [];
|
|
}
|
|
var _statisticToken = statisticToken(token),
|
|
proxyToken = _statisticToken.token,
|
|
flush = _statisticToken.flush;
|
|
var defaultComponentToken = getDefaultComponentToken(component, realToken, getDefaultToken);
|
|
var componentCls = ".".concat(prefixCls);
|
|
var componentToken = getComponentToken(component, realToken, defaultComponentToken, {
|
|
deprecatedTokens: options.deprecatedTokens
|
|
});
|
|
if (defaultComponentToken && _typeof(defaultComponentToken) === 'object') {
|
|
Object.keys(defaultComponentToken).forEach(function (key) {
|
|
defaultComponentToken[key] = "var(".concat(token2CSSVar(key, getCompVarPrefix(component, cssVar.prefix)), ")");
|
|
});
|
|
}
|
|
var mergedToken = mergeToken(proxyToken, {
|
|
componentCls: componentCls,
|
|
prefixCls: prefixCls,
|
|
iconCls: ".".concat(iconPrefixCls),
|
|
antCls: ".".concat(rootPrefixCls),
|
|
calc: calc,
|
|
max: max,
|
|
min: min
|
|
}, defaultComponentToken);
|
|
var styleInterpolation = styleFn(mergedToken, {
|
|
hashId: hashId,
|
|
prefixCls: prefixCls,
|
|
rootPrefixCls: rootPrefixCls,
|
|
iconPrefixCls: iconPrefixCls
|
|
});
|
|
flush(component, componentToken);
|
|
var commonStyle = typeof getCommonStyle === 'function' ? getCommonStyle(mergedToken, prefixCls, rootCls, options.resetFont) : null;
|
|
return [options.resetStyle === false ? null : commonStyle, styleInterpolation];
|
|
});
|
|
return hashId;
|
|
};
|
|
}
|
|
function genSubStyleComponent(componentName, styleFn, getDefaultToken) {
|
|
var options = arguments.length > 3 && arguments[3] !== undefined ? arguments[3] : {};
|
|
var useStyle = genComponentStyleHook(componentName, styleFn, getDefaultToken, _objectSpread({
|
|
resetStyle: false,
|
|
// Sub Style should default after root one
|
|
order: -998
|
|
}, options));
|
|
var StyledComponent = function StyledComponent(_ref) {
|
|
var prefixCls = _ref.prefixCls,
|
|
_ref$rootCls = _ref.rootCls,
|
|
rootCls = _ref$rootCls === void 0 ? prefixCls : _ref$rootCls;
|
|
useStyle(prefixCls, rootCls);
|
|
return null;
|
|
};
|
|
if (process.env.NODE_ENV !== 'production') {
|
|
StyledComponent.displayName = "SubStyle_".concat(String(Array.isArray(componentName) ? componentName.join('.') : componentName));
|
|
}
|
|
return StyledComponent;
|
|
}
|
|
return {
|
|
genStyleHooks: genStyleHooks,
|
|
genSubStyleComponent: genSubStyleComponent,
|
|
genComponentStyleHook: genComponentStyleHook
|
|
};
|
|
}
|
|
export default genStyleUtils; |