import ConfigProvider from '../config-provider'; import React, { Component } from 'react'; import cls from 'classnames'; import { Badge as NextBadge } from '@alifd/next'; import { BadgeProps as NextBadgeProps } from '@alifd/next/types/badge'; interface BadgeAddProps { size?: 'medium' | 'small'; status?: 'success' | 'warning' | 'error'; } interface BadgeProps extends NextBadgeProps, BadgeAddProps {} class Badge extends Component { render() { const { prefix = 'next-', size = 'small', count, className, overflowCount, dot, status, ...otherProps } = this.props; let content: React.ReactNode; if (status && !dot) { if (status === 'warning') { content = ; } else if (status === 'error') { content = ; } else { content = ; } return ( ); } return ( 9, [`${prefix}badge-success`]: status === 'success', [`${prefix}badge-warning`]: status === 'warning', [`${prefix}badge-error`]: status === 'error', })} /> ); } } export default ConfigProvider.config(Badge);