Wrapper for passing context to React component with new context API

withContext.js

import React from 'react';

export default function withContext(Context) {
  return (Component) => (props) => (
    <Context.Consumer>
      { context => <Component {...props} context={context} /> }
    </Context.Consumer>
  );
}

/*
 *  Example: 
 *  export default withContext(MyContext)(SomeCoolComponent)
 */