import React from 'react';
import AntdBadge from 'antd/es/badge';

import { ConfigConsumer } from '../config-provider';
import './index.less';

export default props => (
  <ConfigConsumer>
    {({ getPrefixCls }) => {
      const {
        prefixCls: customizePrefixCls,
        scrollNumberPrefixCls: customizeScrollNumberPrefixCls,
        ...restProps
      } = props;
      const prefixCls = getPrefixCls('badge', customizePrefixCls);
      const scrollNumberPrefixCls = getPrefixCls(
        'scroll-number',
        customizeScrollNumberPrefixCls,
      );
      return (
        <AntdBadge
          {...restProps}
          prefixCls={prefixCls}
          scrollNumberPrefixCls={scrollNumberPrefixCls}
        />
      );
    }}
  </ConfigConsumer>
);