mirror of
https://github.com/ant-design/ant-design.git
synced 2024-12-23 15:08:34 +08:00
5a2818600c
Co-authored-by: afc163 <afc163@gmail.com> Co-authored-by: liuyuan <yuanliu147@qq.com>
149 lines
3.8 KiB
TypeScript
149 lines
3.8 KiB
TypeScript
import * as React from 'react';
|
|
import classNames from 'classnames';
|
|
import useMergedState from 'rc-util/lib/hooks/useMergedState';
|
|
import pickAttrs from 'rc-util/lib/pickAttrs';
|
|
|
|
import { ConfigContext } from '../config-provider';
|
|
import useCSSVarCls from '../config-provider/hooks/useCSSVarCls';
|
|
import useSize from '../config-provider/hooks/useSize';
|
|
import { RadioGroupContextProvider } from './context';
|
|
import type {
|
|
RadioChangeEvent,
|
|
RadioGroupButtonStyle,
|
|
RadioGroupContextProps,
|
|
RadioGroupProps,
|
|
} from './interface';
|
|
import Radio from './radio';
|
|
import useStyle from './style';
|
|
|
|
const RadioGroup = React.forwardRef<HTMLDivElement, RadioGroupProps>((props, ref) => {
|
|
const { getPrefixCls, direction } = React.useContext(ConfigContext);
|
|
|
|
const {
|
|
prefixCls: customizePrefixCls,
|
|
className,
|
|
rootClassName,
|
|
options,
|
|
buttonStyle = 'outline' as RadioGroupButtonStyle,
|
|
disabled,
|
|
children,
|
|
size: customizeSize,
|
|
style,
|
|
id,
|
|
optionType,
|
|
name,
|
|
defaultValue,
|
|
value: customizedValue,
|
|
block = false,
|
|
onChange,
|
|
onMouseEnter,
|
|
onMouseLeave,
|
|
onFocus,
|
|
onBlur,
|
|
} = props;
|
|
|
|
const [value, setValue] = useMergedState(defaultValue, {
|
|
value: customizedValue,
|
|
});
|
|
|
|
const onRadioChange = React.useCallback(
|
|
(event: RadioChangeEvent) => {
|
|
const lastValue = value;
|
|
const val = event.target.value;
|
|
if (!('value' in props)) {
|
|
setValue(val);
|
|
}
|
|
if (val !== lastValue) {
|
|
onChange?.(event);
|
|
}
|
|
},
|
|
[value, setValue, onChange],
|
|
);
|
|
|
|
const prefixCls = getPrefixCls('radio', customizePrefixCls);
|
|
const groupPrefixCls = `${prefixCls}-group`;
|
|
|
|
// Style
|
|
const rootCls = useCSSVarCls(prefixCls);
|
|
const [wrapCSSVar, hashId, cssVarCls] = useStyle(prefixCls, rootCls);
|
|
|
|
let childrenToRender = children;
|
|
// 如果存在 options, 优先使用
|
|
if (options && options.length > 0) {
|
|
childrenToRender = options.map((option) => {
|
|
if (typeof option === 'string' || typeof option === 'number') {
|
|
// 此处类型自动推导为 string
|
|
return (
|
|
<Radio
|
|
key={option.toString()}
|
|
prefixCls={prefixCls}
|
|
disabled={disabled}
|
|
value={option}
|
|
checked={value === option}
|
|
>
|
|
{option}
|
|
</Radio>
|
|
);
|
|
}
|
|
// 此处类型自动推导为 { label: string value: string }
|
|
return (
|
|
<Radio
|
|
key={`radio-group-value-options-${option.value}`}
|
|
prefixCls={prefixCls}
|
|
disabled={option.disabled || disabled}
|
|
value={option.value}
|
|
checked={value === option.value}
|
|
title={option.title}
|
|
style={option.style}
|
|
id={option.id}
|
|
required={option.required}
|
|
>
|
|
{option.label}
|
|
</Radio>
|
|
);
|
|
});
|
|
}
|
|
|
|
const mergedSize = useSize(customizeSize);
|
|
|
|
const classString = classNames(
|
|
groupPrefixCls,
|
|
`${groupPrefixCls}-${buttonStyle}`,
|
|
{
|
|
[`${groupPrefixCls}-${mergedSize}`]: mergedSize,
|
|
[`${groupPrefixCls}-rtl`]: direction === 'rtl',
|
|
[`${groupPrefixCls}-block`]: block,
|
|
},
|
|
className,
|
|
rootClassName,
|
|
hashId,
|
|
cssVarCls,
|
|
rootCls,
|
|
);
|
|
|
|
const memoizedValue = React.useMemo<RadioGroupContextProps>(
|
|
() => ({ onChange: onRadioChange, value, disabled, name, optionType, block }),
|
|
[onRadioChange, value, disabled, name, optionType, block],
|
|
);
|
|
|
|
return wrapCSSVar(
|
|
<div
|
|
{...pickAttrs(props, { aria: true, data: true })}
|
|
className={classString}
|
|
style={style}
|
|
onMouseEnter={onMouseEnter}
|
|
onMouseLeave={onMouseLeave}
|
|
onFocus={onFocus}
|
|
onBlur={onBlur}
|
|
id={id}
|
|
ref={ref}
|
|
>
|
|
<RadioGroupContextProvider value={memoizedValue}>
|
|
{childrenToRender}
|
|
</RadioGroupContextProvider>
|
|
</div>,
|
|
);
|
|
});
|
|
|
|
export default React.memo(RadioGroup);
|