import React from 'react';
import PropTypes from 'prop-types';
import { forbidExtraProps } from 'airbnb-prop-types';

export const CustomShape = PropTypes.shape({});

export default function withWrapper(Component) {
  class Wrapper extends React.Component {
  }

  Wrapper.propTypes = forbidExtraProps({
    ...Component.propTypes,
    phrases: CustomShape.isRequired,
  });

  Wrapper.childContextTypes = {
    ...Component.contextTypes,
    phrases: CustomShape.isRequired,
  };

  Wrapper.displayName = `withWrapper(${Component.displayName || Component.name})`;

  return Wrapper;
}