withFlexibleRender.js

import React from 'react'
import hoistNonReactStatics from 'hoist-non-react-statics'

export default function withFlexibleRender (OriginalComponent) {
  class ComponentWithFlexibleRender extends OriginalComponent {
    render () {
      const { children, render } = this.props
      
      const args = {
        props: this.props,
        state: this.state,
        context: this.context,
      }

      return children
        ? (typeof children === 'function'
          ? children(args)
          : children
        )
        : (render 
          ? render(args)
          : null
        )
    }
  }

  return hoistNonReactStatics(ComponentWithFlexibleRender, OriginalComponent)
}