withWindowSize

import React                from 'react'

function withWindowSize(WrappedComponent) {
  return class WindowSizeProvider extends React.Component {
    constructor(props) {
      super(props)

      this.state = {
        innerWidth: 1280,
        innerHeight: 800,
        show: false,
      }
    }

    onWindowResize = () => {
      if (!this.mounted) return
      this.setState({
        innerWidth: window.innerWidth,
        innerHeight: window.innerHeight,
      })
    }

    componentDidMount() {
      this.mounted = true
      window.addEventListener('resize', this.onWindowResize)
      this.onWindowResize()
      this.setState({ show: true })
    }

    componentWillUnmount() {
      this.mounted = false
      window.removeEventListener('resize', this.onWindowResize);
    }

    render() {
      const { show } = this.state

      if (!show) return null

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

export default withWindowSize