2020-03-02 12:09:38 +08:00
|
|
|
import StarFilled from '@ant-design/icons/StarFilled';
|
2022-05-23 14:37:16 +08:00
|
|
|
import RcRate from 'rc-rate';
|
2022-05-19 09:46:26 +08:00
|
|
|
import type { RateProps as RcRateProps } from 'rc-rate/lib/Rate';
|
2022-05-23 14:37:16 +08:00
|
|
|
import * as React from 'react';
|
2020-04-27 13:42:04 +08:00
|
|
|
import { ConfigContext } from '../config-provider';
|
2022-05-23 14:37:16 +08:00
|
|
|
import Tooltip from '../tooltip';
|
2016-03-30 13:20:03 +08:00
|
|
|
|
2022-05-19 09:46:26 +08:00
|
|
|
export interface RateProps extends RcRateProps {
|
2018-12-28 16:04:05 +08:00
|
|
|
tooltips?: Array<string>;
|
2016-08-10 09:46:56 +08:00
|
|
|
}
|
|
|
|
|
2018-12-28 16:04:05 +08:00
|
|
|
interface RateNodeProps {
|
|
|
|
index: number;
|
|
|
|
}
|
|
|
|
|
2022-09-30 17:44:24 +08:00
|
|
|
const Rate = React.forwardRef<unknown, RateProps>((props, ref) => {
|
|
|
|
const { prefixCls, tooltips, character = <StarFilled />, ...rest } = props;
|
2020-04-27 13:42:04 +08:00
|
|
|
const characterRender = (node: React.ReactElement, { index }: RateNodeProps) => {
|
2022-09-30 17:44:24 +08:00
|
|
|
if (!tooltips) {
|
|
|
|
return node;
|
|
|
|
}
|
2018-12-28 16:04:05 +08:00
|
|
|
return <Tooltip title={tooltips[index]}>{node}</Tooltip>;
|
|
|
|
};
|
|
|
|
|
2020-04-27 13:42:04 +08:00
|
|
|
const { getPrefixCls, direction } = React.useContext(ConfigContext);
|
|
|
|
const ratePrefixCls = getPrefixCls('rate', prefixCls);
|
|
|
|
|
|
|
|
return (
|
|
|
|
<RcRate
|
|
|
|
ref={ref}
|
2022-09-30 17:44:24 +08:00
|
|
|
character={character}
|
2020-04-27 13:42:04 +08:00
|
|
|
characterRender={characterRender}
|
2022-09-30 17:44:24 +08:00
|
|
|
{...rest}
|
2020-04-27 13:42:04 +08:00
|
|
|
prefixCls={ratePrefixCls}
|
2020-04-27 21:02:16 +08:00
|
|
|
direction={direction}
|
2020-04-27 13:42:04 +08:00
|
|
|
/>
|
|
|
|
);
|
|
|
|
});
|
|
|
|
|
2022-06-21 10:24:52 +08:00
|
|
|
if (process.env.NODE_ENV !== 'production') {
|
|
|
|
Rate.displayName = 'Rate';
|
|
|
|
}
|
2020-04-27 13:42:04 +08:00
|
|
|
|
|
|
|
export default Rate;
|