mirror of
https://github.com/terribleplan/next.js.git
synced 2024-01-19 02:48:18 +00:00
25fb3f9c2e
Fixes parts of #5716. I had some issues with the test suite but I'm fairly certain that I got it working correctly.
26 lines
667 B
JavaScript
26 lines
667 B
JavaScript
import React from 'react'
|
|
import hoistStatics from 'hoist-non-react-statics'
|
|
import { getDisplayName } from 'next-server/dist/lib/utils'
|
|
import { RouterContext } from './router'
|
|
|
|
export default function withRouter (ComposedComponent) {
|
|
const displayName = getDisplayName(ComposedComponent)
|
|
|
|
function WithRouteWrapper (props) {
|
|
return (
|
|
<RouterContext.Consumer>
|
|
{router => (
|
|
<ComposedComponent
|
|
router={router}
|
|
{...props}
|
|
/>
|
|
)}
|
|
</RouterContext.Consumer>
|
|
)
|
|
}
|
|
|
|
WithRouteWrapper.displayName = `withRouter(${displayName})`
|
|
|
|
return hoistStatics(WithRouteWrapper, ComposedComponent)
|
|
}
|