Allow monkey-patching navigate function

dmitry-vychikov · Jan 29, 2022 · 713c702 · 713c702
1 parent 5dc07e2
commit 713c702
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 4 deletions.
diff --git a/packages/react-router-dom/index.tsx b/packages/react-router-dom/index.tsx
@@ -24,7 +24,8 @@ import {
   useParams,
   useResolvedPath,
   useRoutes,
-  useOutletContext
+  useOutletContext,
+  patchNavigate
 } from "react-router";
 import type { To } from "react-router";
 
@@ -73,7 +74,8 @@ export {
   useParams,
   useResolvedPath,
   useRoutes,
-  useOutletContext
+  useOutletContext,
+  patchNavigate
 };
 
 export type {

diff --git a/packages/react-router-native/index.tsx b/packages/react-router-native/index.tsx
@@ -31,7 +31,8 @@ import {
   useParams,
   useResolvedPath,
   useRoutes,
-  useOutletContext
+  useOutletContext,
+  patchNavigate
 } from "react-router";
 import type { To } from "react-router";
 
@@ -65,7 +66,8 @@ export {
   useParams,
   useResolvedPath,
   useRoutes,
-  useOutletContext
+  useOutletContext,
+  patchNavigate
 };
 
 export type {

diff --git a/packages/react-router/index.tsx b/packages/react-router/index.tsx
@@ -497,13 +497,27 @@ export interface NavigateOptions {
   state?: any;
 }
 
+let decoratedNavigate: (() => NavigateFunction) | null = null
+
+export function patchNavigate(decorator: (original: () => NavigateFunction) => () => NavigateFunction): void {
+  decoratedNavigate = decorator(useNavigateInternal)
+}
 /**
  * Returns an imperative method for changing the location. Used by <Link>s, but
  * may also be used by other elements to change the location.
  *
  * @see https://reactrouter.com/docs/en/v6/api#usenavigate
  */
 export function useNavigate(): NavigateFunction {
+  if (decoratedNavigate) {
+    return decoratedNavigate()
+  }
+
+  // eslint-disable-next-line react-hooks/rules-of-hooks
+  return useNavigateInternal()
+}
+
+function useNavigateInternal(): NavigateFunction {
   invariant(
     useInRouterContext(),
     // TODO: This error is probably because they somehow have 2 versions of the