Extract anyclip
This commit is contained in:
41
client/with-router.tsx
Normal file
41
client/with-router.tsx
Normal file
@@ -0,0 +1,41 @@
|
||||
import React, { type JSX } from 'react'
|
||||
import type {
|
||||
BaseContext,
|
||||
NextComponentType,
|
||||
NextPageContext,
|
||||
} from '../shared/lib/utils'
|
||||
import type { NextRouter } from './router'
|
||||
import { useRouter } from './router'
|
||||
|
||||
export type WithRouterProps = {
|
||||
router: NextRouter
|
||||
}
|
||||
|
||||
export type ExcludeRouterProps<P> = Pick<
|
||||
P,
|
||||
Exclude<keyof P, keyof WithRouterProps>
|
||||
>
|
||||
|
||||
export default function withRouter<
|
||||
P extends WithRouterProps,
|
||||
C extends BaseContext = NextPageContext,
|
||||
>(
|
||||
ComposedComponent: NextComponentType<C, any, P>
|
||||
): React.ComponentType<ExcludeRouterProps<P>> {
|
||||
function WithRouterWrapper(props: any): JSX.Element {
|
||||
return <ComposedComponent router={useRouter()} {...props} />
|
||||
}
|
||||
|
||||
WithRouterWrapper.getInitialProps = ComposedComponent.getInitialProps
|
||||
// This is needed to allow checking for custom getInitialProps in _app
|
||||
;(WithRouterWrapper as any).origGetInitialProps = (
|
||||
ComposedComponent as any
|
||||
).origGetInitialProps
|
||||
if (process.env.NODE_ENV !== 'production') {
|
||||
const name =
|
||||
ComposedComponent.displayName || ComposedComponent.name || 'Unknown'
|
||||
WithRouterWrapper.displayName = `withRouter(${name})`
|
||||
}
|
||||
|
||||
return WithRouterWrapper
|
||||
}
|
||||
Reference in New Issue
Block a user