feat: InnerWrap option

TanStack · Dec 14, 2023 · 8b631d9 · 8b631d9
1 parent 369bcb1
commit 8b631d9
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/packages/react-router/src/RouterProvider.tsx b/packages/react-router/src/RouterProvider.tsx
@@ -74,18 +74,26 @@ export function RouterProvider<
     },
   } as any)
 
-  const inner = (
-    <routerContext.Provider value={router}>
+  const matches = router.options.InnerWrap ? (
+    <router.options.InnerWrap>
       <Matches />
+    </router.options.InnerWrap>
+  ) : (
+    <Matches />
+  )
+
+  const provider = (
+    <routerContext.Provider value={router}>
+      {matches}
       <Transitioner />
     </routerContext.Provider>
   )
 
   if (router.options.Wrap) {
-    return <router.options.Wrap>{inner}</router.options.Wrap>
+    return <router.options.Wrap>{provider}</router.options.Wrap>
   }
 
-  return inner
+  return provider
 }
 
 function Transitioner() {

diff --git a/packages/react-router/src/router.ts b/packages/react-router/src/router.ts
@@ -119,6 +119,7 @@ export interface RouterOptions<
   routeMasks?: RouteMask<TRouteTree>[]
   unmaskOnReload?: boolean
   Wrap?: (props: { children: any }) => JSX.Element
+  InnerWrap?: (props: { children: any }) => JSX.Element
   notFoundRoute?: AnyRoute
 }