@gssfed/vital-ui-kit-react/src/Icon/components/IconProvider.js

Vital UI Kit for React!

// @flow

import { Component, Children, type Node } from 'react';
import PropTypes from 'prop-types';

type Props = {
  icon: Object,
  children: Node,
};

class IconProvider extends Component<Props> {
  static childContextTypes = {
    // eslint-disable-next-line
    icon: PropTypes.object.isRequired,
  };

  getChildContext() {
    const { icon } = this.props;
    return { icon };
  }

  render() {
    if (!this.props.children) {
      return null;
    }
    return Children.only(this.props.children);
  }
}

export default IconProvider;