ant-design/components/rate/index.tsx
2022-05-23 14:37:16 +08:00

43 lines
1.1 KiB
TypeScript

import StarFilled from '@ant-design/icons/StarFilled';
import RcRate from 'rc-rate';
import type { RateProps as RcRateProps } from 'rc-rate/lib/Rate';
import * as React from 'react';
import { ConfigContext } from '../config-provider';
import Tooltip from '../tooltip';
export interface RateProps extends RcRateProps {
tooltips?: Array<string>;
}
interface RateNodeProps {
index: number;
}
const Rate = React.forwardRef<unknown, RateProps>(({ prefixCls, tooltips, ...props }, ref) => {
const characterRender = (node: React.ReactElement, { index }: RateNodeProps) => {
if (!tooltips) return node;
return <Tooltip title={tooltips[index]}>{node}</Tooltip>;
};
const { getPrefixCls, direction } = React.useContext(ConfigContext);
const ratePrefixCls = getPrefixCls('rate', prefixCls);
return (
<RcRate
ref={ref}
characterRender={characterRender}
{...props}
prefixCls={ratePrefixCls}
direction={direction}
/>
);
});
Rate.displayName = 'Rate';
Rate.defaultProps = {
character: <StarFilled />,
};
export default Rate;