with-context.js

import React, { Component } from 'react'

function withContext(key, type) {
  return WrappedComponent => class ContextComponent extends Component {
    static contextTypes = {
      ...WrappedComponent.contextTypes,
      [key]: type
    }

    render() {
      return (
        <WrappedComponent {...this.context} {...this.props} />
      )
    }
  }
}

export default withContext