import React from 'react';
import AntTimeline from 'antd/es/timeline';

import {
  ConfigConsumer,
  getPrefixCls as getPrefixCls2,
} from '../config-provider';
import './index.less';

const Timeline = props => (
  <ConfigConsumer>
    {({ getPrefixCls }) => {
      const { prefixCls: customizePrefixCls, ...restProps } = props;
      const prefixCls = getPrefixCls('timeline', customizePrefixCls);
      return <AntTimeline prefixCls={prefixCls} {...restProps} />;
    }}
  </ConfigConsumer>
);

const TimeItem = props => {
  const { prefixCls: customizePrefixCls, ...restProps } = props;
  const prefixCls = getPrefixCls2('timeline', customizePrefixCls);
  return <AntTimeline.Item prefixCls={prefixCls} {...props} />;
};

Timeline.Item = TimeItem;
Timeline.defaultProps = {
  reverse: false,
  mode: '',
};

export default Timeline;