import React from 'react';
import AntdCheckbox from 'antd/es/checkbox';

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

const CheckboxGroup = props => {
  const { getPrefixCls } = React.useContext(ConfigContext);
  const { prefixCls: customizePrefixCls, ...restProps } = props;
  const prefixCls = getPrefixCls('checkbox', customizePrefixCls);
  return <AntdCheckbox.Group {...restProps} prefixCls={prefixCls} />;
};

const Checkbox = props => {
  const { getPrefixCls } = React.useContext(ConfigContext);
  const { prefixCls: customizePrefixCls, ...restProps } = props;
  const prefixCls = getPrefixCls('checkbox', customizePrefixCls);
  return <AntdCheckbox {...restProps} prefixCls={prefixCls} />;
};

Checkbox.Group = CheckboxGroup;

export default Checkbox;