mirror of
https://github.com/ant-design/ant-design.git
synced 2024-11-29 21:59:41 +08:00
47 lines
1.0 KiB
JavaScript
47 lines
1.0 KiB
JavaScript
import React, { cloneElement } from 'react';
|
|
const prefixCls = 'ant-badge';
|
|
|
|
class AntBadge extends React.Component {
|
|
constructor(props) {
|
|
super(props);
|
|
}
|
|
|
|
render() {
|
|
if (this.props.dot) {
|
|
return <span className={prefixCls} {...this.props}>
|
|
{this.props.children}
|
|
<sup className={prefixCls + '-dot'}></sup>
|
|
</span>;
|
|
}
|
|
let count = this.props.count;
|
|
// null undefined "" "0" 0
|
|
if (!count || count === '0') {
|
|
return cloneElement(this.props.children);
|
|
} else {
|
|
count = count >= 100 ? '99+' : count;
|
|
return (
|
|
<span className={prefixCls} title={count} {...this.props}>
|
|
{this.props.children}
|
|
<sup className={prefixCls + '-count'}>{count}</sup>
|
|
</span>
|
|
);
|
|
}
|
|
}
|
|
}
|
|
|
|
AntBadge.defaultProps = {
|
|
prefixCls: prefixCls,
|
|
count: null,
|
|
dot: false
|
|
};
|
|
|
|
AntBadge.propTypes = {
|
|
count: React.PropTypes.oneOfType([
|
|
React.PropTypes.string,
|
|
React.PropTypes.number
|
|
]),
|
|
dot: React.PropTypes.bool
|
|
};
|
|
|
|
export default AntBadge;
|