@tanstack/react-router 0.0.1-beta.235 → 0.0.1-beta.236

package/src/utils.ts

@@ -18,8 +18,8 @@ export type PickExtra<T, K> = {
   [TKey in keyof K as string extends TKey
     ? never
     : TKey extends keyof T
-      ? never
-      : TKey]: K[TKey]
+    ? never
+    : TKey]: K[TKey]
 }
 
 export type PickRequired<T> = {
@@ -105,15 +105,15 @@ export type ValueKeys<O> = Extract<keyof O, PropertyKey>
 export type DeepAwaited<T> = T extends Promise<infer A>
   ? DeepAwaited<A>
   : T extends Record<infer A, Promise<infer B>>
-    ? { [K in A]: DeepAwaited<B> }
-    : T
+  ? { [K in A]: DeepAwaited<B> }
+  : T
 
 export type PathParamMask<TRoutePath extends string> =
   TRoutePath extends `${infer L}/$${infer C}/${infer R}`
     ? PathParamMask<`${L}/${string}/${R}`>
     : TRoutePath extends `${infer L}/$${infer C}`
-      ? PathParamMask<`${L}/${string}`>
-      : TRoutePath
+    ? PathParamMask<`${L}/${string}`>
+    : TRoutePath
 
 export type Timeout = ReturnType<typeof setTimeout>
 
@@ -314,10 +314,10 @@ export type RouteFromIdOrRoute<
 > = T extends ParseRoute<TRouteTree>
   ? T
   : T extends RouteIds<TRouteTree>
-    ? RoutesById<TRouteTree>[T]
-    : T extends string
-      ? RouteIds<TRouteTree>
-      : never
+  ? RoutesById<TRouteTree>[T]
+  : T extends string
+  ? RouteIds<TRouteTree>
+  : never
 
 export function useRouteContext<
   TRouteTree extends AnyRoute = RegisteredRouter['routeTree'],
@@ -326,9 +326,17 @@ export function useRouteContext<
   TRouteContext = RouteById<TRouteTree, TFrom>['types']['allContext'],
   TSelected = TRouteContext,
 >(
-  opts: StrictOrFrom<TFrom>,
+  opts: StrictOrFrom<TFrom> & {
+    select?: (search: TRouteContext) => TSelected
+  },
 ): TStrict extends true ? TSelected : TSelected | undefined {
-  return useMatch(opts).context
+  return useMatch({
+    ...(opts as any),
+    select: (match: RouteMatch) =>
+      opts?.select
+        ? opts.select(match.context as TRouteContext)
+        : match.context,
+  })
 }
 
 export const useLayoutEffect =