1
0
Fork 0
mirror of https://github.com/terribleplan/next.js.git synced 2024-01-19 02:48:18 +00:00
next.js/packages/next/lib/router/with-router.js

28 lines
674 B
JavaScript
Raw Normal View History

import React, { Component } from 'react'
import PropTypes from 'prop-types'
import hoistStatics from 'hoist-non-react-statics'
import { getDisplayName } from '../utils'
export default function withRouter (ComposedComponent) {
const displayName = getDisplayName(ComposedComponent)
class WithRouteWrapper extends Component {
static contextTypes = {
router: PropTypes.object
}
2018-02-07 08:09:23 +00:00
static displayName = `withRouter(${displayName})`
render () {
const props = {
router: this.context.router,
...this.props
}
return <ComposedComponent {...props} />
}
}
return hoistStatics(WithRouteWrapper, ComposedComponent)
}