Fix types for PatchRoutesOnNavigationFunction (#11976)

Author: brophdawg11

packages/react-router/index.ts

@@ -78,6 +78,8 @@ export type {
   NavigateOptions,
   Navigator,
   NonIndexRouteObject,
+  PatchRoutesOnNavigationFunction,
+  PatchRoutesOnNavigationFunctionArgs,
   RouteMatch,
   RouteObject,
 } from "./lib/context";
@@ -93,7 +95,6 @@ export type {
   RouterProps,
   RouterProviderProps,
   RoutesProps,
-  PatchRoutesOnNavigationFunction as PatchRoutesOnNavigationFunction,
 } from "./lib/components";
 export type { NavigateFunction } from "./lib/hooks";
 export {

packages/react-router/lib/components.tsx

@@ -23,7 +23,6 @@ import type {
 } from "./router/router";
 import { createRouter } from "./router/router";
 import type {
-  AgnosticPatchRoutesOnNavigationFunction,
   DataStrategyFunction,
   LazyRouteFunction,
   TrackedPromise,
@@ -35,6 +34,7 @@ import type {
   IndexRouteObject,
   Navigator,
   NonIndexRouteObject,
+  PatchRoutesOnNavigationFunction,
   RouteMatch,
   RouteObject,
   ViewTransitionContextObject,
@@ -130,9 +130,6 @@ export function mapRouteProperties(route: RouteObject) {
   return updates;
 }
 
-export interface PatchRoutesOnNavigationFunction
-  extends AgnosticPatchRoutesOnNavigationFunction<RouteMatch> {}
-
 /**
  * @category Routers
  */

packages/react-router/lib/context.ts

@@ -13,6 +13,8 @@ import type {
 import type {
   AgnosticIndexRouteObject,
   AgnosticNonIndexRouteObject,
+  AgnosticPatchRoutesOnNavigationFunction,
+  AgnosticPatchRoutesOnNavigationFunctionArgs,
   AgnosticRouteMatch,
   LazyRouteFunction,
   TrackedPromise,
@@ -74,6 +76,12 @@ export interface RouteMatch<
 
 export interface DataRouteMatch extends RouteMatch<string, DataRouteObject> {}
 
+export type PatchRoutesOnNavigationFunctionArgs =
+  AgnosticPatchRoutesOnNavigationFunctionArgs<RouteObject, RouteMatch>;
+
+export type PatchRoutesOnNavigationFunction =
+  AgnosticPatchRoutesOnNavigationFunction<RouteObject, RouteMatch>;
+
 export interface DataRouterContextObject
   // Omit `future` since those can be pulled from the `router`
   // `NavigationContext` needs future since it doesn't have a `router` in all cases

packages/react-router/lib/dom/lib.tsx

@@ -64,9 +64,12 @@ import {
   mergeRefs,
   usePrefetchBehavior,
 } from "./ssr/components";
-import type { PatchRoutesOnNavigationFunction } from "../components";
 import { Router, mapRouteProperties } from "../components";
-import type { RouteObject, NavigateOptions } from "../context";
+import type {
+  RouteObject,
+  NavigateOptions,
+  PatchRoutesOnNavigationFunction,
+} from "../context";
 import {
   DataRouterContext,
   DataRouterStateContext,

packages/react-router/lib/dom/ssr/fog-of-war.ts

@@ -1,5 +1,5 @@
 import * as React from "react";
-import type { PatchRoutesOnNavigationFunction } from "../../components";
+import type { PatchRoutesOnNavigationFunction } from "../../context";
 import type { Router as RemixRouter } from "../../router/router";
 import { matchRoutes } from "../../router/utils";
 import type { AssetsManifest } from "./entry";

packages/react-router/lib/router/utils.ts

@@ -223,15 +223,21 @@ export interface DataStrategyFunction {
   (args: DataStrategyFunctionArgs): Promise<Record<string, DataStrategyResult>>;
 }
 
-export interface AgnosticPatchRoutesOnNavigationFunction<
+export type AgnosticPatchRoutesOnNavigationFunctionArgs<
+  O extends AgnosticRouteObject = AgnosticRouteObject,
   M extends AgnosticRouteMatch = AgnosticRouteMatch
-> {
-  (opts: {
-    path: string;
-    matches: M[];
-    patch: (routeId: string | null, children: AgnosticRouteObject[]) => void;
-  }): void | Promise<void>;
-}
+> = {
+  path: string;
+  matches: M[];
+  patch: (routeId: string | null, children: O[]) => void;
+};
+
+export type AgnosticPatchRoutesOnNavigationFunction<
+  O extends AgnosticRouteObject = AgnosticRouteObject,
+  M extends AgnosticRouteMatch = AgnosticRouteMatch
+> = (
+  opts: AgnosticPatchRoutesOnNavigationFunctionArgs<O, M>
+) => void | Promise<void>;
 
 /**
  * Function provided by the framework-aware layers to set any framework-specific