51 lines
1.2 KiB
JavaScript
51 lines
1.2 KiB
JavaScript
import React from "react";
|
|
import PropTypes from "prop-types";
|
|
import RouterContext from "./RouterContext";
|
|
import hoistStatics from "hoist-non-react-statics";
|
|
import invariant from "tiny-invariant";
|
|
|
|
/**
|
|
* A public higher-order component to access the imperative API
|
|
*/
|
|
function withRouter(Component) {
|
|
const displayName = `withRouter(${Component.displayName || Component.name})`;
|
|
const C = props => {
|
|
const { wrappedComponentRef, ...remainingProps } = props;
|
|
|
|
return (
|
|
<RouterContext.Consumer>
|
|
{context => {
|
|
invariant(
|
|
context,
|
|
`You should not use <${displayName} /> outside a <Router>`
|
|
);
|
|
return (
|
|
<Component
|
|
{...remainingProps}
|
|
{...context}
|
|
ref={wrappedComponentRef}
|
|
/>
|
|
);
|
|
}}
|
|
</RouterContext.Consumer>
|
|
);
|
|
};
|
|
|
|
C.displayName = displayName;
|
|
C.WrappedComponent = Component;
|
|
|
|
if (__DEV__) {
|
|
C.propTypes = {
|
|
wrappedComponentRef: PropTypes.oneOfType([
|
|
PropTypes.string,
|
|
PropTypes.func,
|
|
PropTypes.object
|
|
])
|
|
};
|
|
}
|
|
|
|
return hoistStatics(C, Component);
|
|
}
|
|
|
|
export default withRouter;
|