2017-08-30 14:07:12 +00:00
|
|
|
import React, { Component } from 'react'
|
|
|
|
import PropTypes from 'prop-types'
|
|
|
|
import hoistStatics from 'hoist-non-react-statics'
|
|
|
|
import { getDisplayName } from '../utils'
|
|
|
|
|
2017-08-31 19:37:20 +00:00
|
|
|
export default function withRouter (ComposedComponent) {
|
2017-08-30 14:07:12 +00:00
|
|
|
const displayName = getDisplayName(ComposedComponent)
|
|
|
|
|
|
|
|
class WithRouteWrapper extends Component {
|
|
|
|
static contextTypes = {
|
|
|
|
router: PropTypes.object
|
|
|
|
}
|
|
|
|
|
|
|
|
static displayName = `withRoute(${displayName})`
|
|
|
|
|
|
|
|
render () {
|
|
|
|
const props = {
|
|
|
|
router: this.context.router,
|
|
|
|
...this.props
|
|
|
|
}
|
|
|
|
|
|
|
|
return <ComposedComponent {...props} />
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return hoistStatics(WithRouteWrapper, ComposedComponent)
|
|
|
|
}
|