import React from 'react';
import Button from 'antd/es/button';
import classnames from 'classnames';

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

export default props => {
  const {
    size = '',
    className = '',
    prefixCls: customizePrefixCls,
    ...restProps
  } = props;
  return (
    <ConfigConsumer>
      {({ getPrefixCls }) => {
        const prefixCls = getPrefixCls('btn', customizePrefixCls);
        let cls = className;
        if (size === 'xsmall') {
          cls = classnames(className, `${prefixCls}-xs`);
        }
        return <Button {...restProps} prefixCls={prefixCls} className={cls} />;
      }}
    </ConfigConsumer>
  );
};