mirror of
https://gitee.com/ant-design/ant-design.git
synced 2024-12-05 05:28:20 +08:00
a3f1de3293
* refactor: hooks * fix: lint * update hooks * update ref
140 lines
3.8 KiB
TypeScript
140 lines
3.8 KiB
TypeScript
import * as React from 'react';
|
|
import classNames from 'classnames';
|
|
import RowContext from './RowContext';
|
|
import { ConfigConsumer, ConfigConsumerProps } from '../config-provider';
|
|
|
|
// https://github.com/ant-design/ant-design/issues/14324
|
|
type ColSpanType = number | string;
|
|
|
|
type FlexType = number | 'none' | 'auto' | string;
|
|
|
|
export interface ColSize {
|
|
span?: ColSpanType;
|
|
order?: ColSpanType;
|
|
offset?: ColSpanType;
|
|
push?: ColSpanType;
|
|
pull?: ColSpanType;
|
|
}
|
|
|
|
export interface ColProps extends React.HTMLAttributes<HTMLDivElement> {
|
|
span?: ColSpanType;
|
|
order?: ColSpanType;
|
|
offset?: ColSpanType;
|
|
push?: ColSpanType;
|
|
pull?: ColSpanType;
|
|
xs?: ColSpanType | ColSize;
|
|
sm?: ColSpanType | ColSize;
|
|
md?: ColSpanType | ColSize;
|
|
lg?: ColSpanType | ColSize;
|
|
xl?: ColSpanType | ColSize;
|
|
xxl?: ColSpanType | ColSize;
|
|
prefixCls?: string;
|
|
flex?: FlexType;
|
|
}
|
|
|
|
function parseFlex(flex: FlexType): string {
|
|
if (typeof flex === 'number') {
|
|
return `${flex} ${flex} auto`;
|
|
}
|
|
|
|
if (/^\d+(\.\d+)?(px|em|rem|%)$/.test(flex)) {
|
|
return `0 0 ${flex}`;
|
|
}
|
|
|
|
return flex;
|
|
}
|
|
|
|
const Col = React.forwardRef<HTMLDivElement, ColProps>((props, ref) => {
|
|
const renderCol = ({ getPrefixCls, direction }: ConfigConsumerProps) => {
|
|
const {
|
|
prefixCls: customizePrefixCls,
|
|
span,
|
|
order,
|
|
offset,
|
|
push,
|
|
pull,
|
|
className,
|
|
children,
|
|
flex,
|
|
style,
|
|
...others
|
|
} = props;
|
|
const prefixCls = getPrefixCls('col', customizePrefixCls);
|
|
let sizeClassObj = {};
|
|
['xs', 'sm', 'md', 'lg', 'xl', 'xxl'].forEach(size => {
|
|
let sizeProps: ColSize = {};
|
|
const propSize = (props as any)[size];
|
|
if (typeof propSize === 'number') {
|
|
sizeProps.span = propSize;
|
|
} else if (typeof propSize === 'object') {
|
|
sizeProps = propSize || {};
|
|
}
|
|
|
|
delete (others as any)[size];
|
|
|
|
sizeClassObj = {
|
|
...sizeClassObj,
|
|
[`${prefixCls}-${size}-${sizeProps.span}`]: sizeProps.span !== undefined,
|
|
[`${prefixCls}-${size}-order-${sizeProps.order}`]: sizeProps.order || sizeProps.order === 0,
|
|
[`${prefixCls}-${size}-offset-${sizeProps.offset}`]:
|
|
sizeProps.offset || sizeProps.offset === 0,
|
|
[`${prefixCls}-${size}-push-${sizeProps.push}`]: sizeProps.push || sizeProps.push === 0,
|
|
[`${prefixCls}-${size}-pull-${sizeProps.pull}`]: sizeProps.pull || sizeProps.pull === 0,
|
|
[`${prefixCls}-rtl`]: direction === 'rtl',
|
|
};
|
|
});
|
|
const classes = classNames(
|
|
prefixCls,
|
|
{
|
|
[`${prefixCls}-${span}`]: span !== undefined,
|
|
[`${prefixCls}-order-${order}`]: order,
|
|
[`${prefixCls}-offset-${offset}`]: offset,
|
|
[`${prefixCls}-push-${push}`]: push,
|
|
[`${prefixCls}-pull-${pull}`]: pull,
|
|
},
|
|
className,
|
|
sizeClassObj,
|
|
);
|
|
|
|
return (
|
|
<RowContext.Consumer>
|
|
{({ gutter }) => {
|
|
let mergedStyle: React.CSSProperties = { ...style };
|
|
if (gutter) {
|
|
mergedStyle = {
|
|
...(gutter[0]! > 0
|
|
? {
|
|
paddingLeft: gutter[0]! / 2,
|
|
paddingRight: gutter[0]! / 2,
|
|
}
|
|
: {}),
|
|
...(gutter[1]! > 0
|
|
? {
|
|
paddingTop: gutter[1]! / 2,
|
|
paddingBottom: gutter[1]! / 2,
|
|
}
|
|
: {}),
|
|
...mergedStyle,
|
|
};
|
|
}
|
|
if (flex) {
|
|
mergedStyle.flex = parseFlex(flex);
|
|
}
|
|
|
|
return (
|
|
<div {...others} style={mergedStyle} className={classes} ref={ref}>
|
|
{children}
|
|
</div>
|
|
);
|
|
}}
|
|
</RowContext.Consumer>
|
|
);
|
|
};
|
|
|
|
return <ConfigConsumer>{renderCol}</ConfigConsumer>;
|
|
});
|
|
|
|
Col.displayName = 'Col';
|
|
|
|
export default Col;
|