'use client' import { useQueries } from './useQueries' import { defaultThrowOnError } from './suspense' import type { UseSuspenseQueryOptions, UseSuspenseQueryResult } from './types' import type { DefaultError, QueryClient, QueryFunction, ThrowOnError, } from '@tanstack/query-core' // Avoid TS depth-limit error in case of large array literal type MAXIMUM_DEPTH = 20 type GetSuspenseOptions = // Part 1: responsible for applying explicit type parameter to function arguments, if object { queryFnData: TQueryFnData, error: TError, data: TData } T extends { queryFnData: infer TQueryFnData error?: infer TError data: infer TData } ? UseSuspenseQueryOptions : T extends { queryFnData: infer TQueryFnData; error?: infer TError } ? UseSuspenseQueryOptions : T extends { data: infer TData; error?: infer TError } ? UseSuspenseQueryOptions : // Part 2: responsible for applying explicit type parameter to function arguments, if tuple [TQueryFnData, TError, TData] T extends [infer TQueryFnData, infer TError, infer TData] ? UseSuspenseQueryOptions : T extends [infer TQueryFnData, infer TError] ? UseSuspenseQueryOptions : T extends [infer TQueryFnData] ? UseSuspenseQueryOptions : // Part 3: responsible for inferring and enforcing type if no explicit parameter was provided T extends { queryFn?: QueryFunction select: (data: any) => infer TData throwOnError?: ThrowOnError } ? UseSuspenseQueryOptions : T extends { queryFn?: QueryFunction throwOnError?: ThrowOnError } ? UseSuspenseQueryOptions : // Fallback UseSuspenseQueryOptions type GetSuspenseResults = // Part 1: responsible for mapping explicit type parameter to function result, if object T extends { queryFnData: any; error?: infer TError; data: infer TData } ? UseSuspenseQueryResult : T extends { queryFnData: infer TQueryFnData; error?: infer TError } ? UseSuspenseQueryResult : T extends { data: infer TData; error?: infer TError } ? UseSuspenseQueryResult : // Part 2: responsible for mapping explicit type parameter to function result, if tuple T extends [any, infer TError, infer TData] ? UseSuspenseQueryResult : T extends [infer TQueryFnData, infer TError] ? UseSuspenseQueryResult : T extends [infer TQueryFnData] ? UseSuspenseQueryResult : // Part 3: responsible for mapping inferred type to results, if no explicit parameter was provided T extends { queryFn?: QueryFunction select: (data: any) => infer TData throwOnError?: ThrowOnError } ? UseSuspenseQueryResult< TData, unknown extends TError ? DefaultError : TError > : T extends { queryFn?: QueryFunction throwOnError?: ThrowOnError } ? UseSuspenseQueryResult< TQueryFnData, unknown extends TError ? DefaultError : TError > : // Fallback UseSuspenseQueryResult /** * SuspenseQueriesOptions reducer recursively unwraps function arguments to infer/enforce type param */ export type SuspenseQueriesOptions< T extends Array, Result extends Array = [], Depth extends ReadonlyArray = [], > = Depth['length'] extends MAXIMUM_DEPTH ? Array : T extends [] ? [] : T extends [infer Head] ? [...Result, GetSuspenseOptions] : T extends [infer Head, ...infer Tail] ? SuspenseQueriesOptions< [...Tail], [...Result, GetSuspenseOptions], [...Depth, 1] > : Array extends T ? T : // If T is *some* array but we couldn't assign unknown[] to it, then it must hold some known/homogenous type! // use this to infer the param types in the case of Array.map() argument T extends Array< UseSuspenseQueryOptions< infer TQueryFnData, infer TError, infer TData, infer TQueryKey > > ? Array> : // Fallback Array /** * SuspenseQueriesResults reducer recursively maps type param to results */ export type SuspenseQueriesResults< T extends Array, Result extends Array = [], Depth extends ReadonlyArray = [], > = Depth['length'] extends MAXIMUM_DEPTH ? Array : T extends [] ? [] : T extends [infer Head] ? [...Result, GetSuspenseResults] : T extends [infer Head, ...infer Tail] ? SuspenseQueriesResults< [...Tail], [...Result, GetSuspenseResults], [...Depth, 1] > : T extends Array< UseSuspenseQueryOptions< infer TQueryFnData, infer TError, infer TData, any > > ? // Dynamic-size (homogenous) UseQueryOptions array: map directly to array of results Array< UseSuspenseQueryResult< unknown extends TData ? TQueryFnData : TData, unknown extends TError ? DefaultError : TError > > : // Fallback Array export function useSuspenseQueries< T extends Array, TCombinedResult = SuspenseQueriesResults, >( options: { queries: readonly [...SuspenseQueriesOptions] combine?: (result: SuspenseQueriesResults) => TCombinedResult }, queryClient?: QueryClient, ): TCombinedResult { return useQueries( { ...options, queries: options.queries.map((query) => ({ ...query, suspense: true, throwOnError: defaultThrowOnError, enabled: true, })), } as any, queryClient, ) }