mirror of
https://github.com/ant-design/ant-design.git
synced 2024-12-11 13:59:11 +08:00
9d1854763d
* refactor: add useSize * fix * fix * fix * fix * fix
23 lines
603 B
TypeScript
23 lines
603 B
TypeScript
import React from 'react';
|
|
import type { SizeType } from '../SizeContext';
|
|
import SizeContext from '../SizeContext';
|
|
|
|
const useSize = <T>(customSize?: T | ((ctxSize: SizeType) => T)): T => {
|
|
const size = React.useContext<SizeType>(SizeContext);
|
|
const mergedSize = React.useMemo<T>(() => {
|
|
if (!customSize) {
|
|
return size as T;
|
|
}
|
|
if (typeof customSize === 'string') {
|
|
return customSize ?? size;
|
|
}
|
|
if (customSize instanceof Function) {
|
|
return customSize(size);
|
|
}
|
|
return size as T;
|
|
}, [customSize, size]);
|
|
return mergedSize;
|
|
};
|
|
|
|
export default useSize;
|