Connect To Context

connectToContext.jsx

// @flow

import React from 'react';

type MapValuesToProps = (values: any) => any;

export const connectToContext = (context: React$Context<any>) => (mapValuesToProps: MapValuesToProps) => (
  Comp: any
) => {
  const wrapped = (props: any) => {
    return (
      <context.Consumer>
        {values => {
          const extraProps = mapValuesToProps(values);
          return <Comp {...props} {...extraProps} />;
        }}
      </context.Consumer>
    );
  };
  wrapped.displayName = 'connectToContext';
  return wrapped;
};