'use client' import { skipToken } from '@tanstack/query-core' 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 // Widen the type of the symbol to enable type inference even if skipToken is not immutable. type SkipTokenForUseQueries = symbol type GetUseSuspenseQueryOptions = // 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 | SkipTokenForUseQueries select?: (data: any) => infer TData throwOnError?: ThrowOnError } ? UseSuspenseQueryOptions< TQueryFnData, TError, TData, TQueryKey > : T extends { queryFn?: | QueryFunction | SkipTokenForUseQueries throwOnError?: ThrowOnError } ? UseSuspenseQueryOptions< TQueryFnData, TError, TQueryFnData, TQueryKey > : // Fallback UseSuspenseQueryOptions type GetUseSuspenseQueryResult = // 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 | SkipTokenForUseQueries select?: (data: any) => infer TData throwOnError?: ThrowOnError } ? UseSuspenseQueryResult< unknown extends TData ? TQueryFnData : TData, unknown extends TError ? DefaultError : TError > : T extends { queryFn?: | QueryFunction | SkipTokenForUseQueries 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, TResults extends Array = [], TDepth extends ReadonlyArray = [], > = TDepth['length'] extends MAXIMUM_DEPTH ? Array : T extends [] ? [] : T extends [infer Head] ? [...TResults, GetUseSuspenseQueryOptions] : T extends [infer Head, ...infer Tails] ? SuspenseQueriesOptions< [...Tails], [...TResults, GetUseSuspenseQueryOptions], [...TDepth, 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< UseSuspenseQueryOptions > : // Fallback Array /** * SuspenseQueriesResults reducer recursively maps type param to results */ export type SuspenseQueriesResults< T extends Array, TResults extends Array = [], TDepth extends ReadonlyArray = [], > = TDepth['length'] extends MAXIMUM_DEPTH ? Array : T extends [] ? [] : T extends [infer Head] ? [...TResults, GetUseSuspenseQueryResult] : T extends [infer Head, ...infer Tails] ? SuspenseQueriesResults< [...Tails], [...TResults, GetUseSuspenseQueryResult], [...TDepth, 1] > : { [K in keyof T]: GetUseSuspenseQueryResult } export function useSuspenseQueries< T extends Array, TCombinedResult = SuspenseQueriesResults, >( options: { queries: | readonly [...SuspenseQueriesOptions] | readonly [...{ [K in keyof T]: GetUseSuspenseQueryOptions }] combine?: (result: SuspenseQueriesResults) => TCombinedResult }, queryClient?: QueryClient, ): TCombinedResult export function useSuspenseQueries< T extends Array, TCombinedResult = SuspenseQueriesResults, >( options: { queries: readonly [...SuspenseQueriesOptions] combine?: (result: SuspenseQueriesResults) => TCombinedResult }, queryClient?: QueryClient, ): TCombinedResult export function useSuspenseQueries(options: any, queryClient?: QueryClient) { return useQueries( { ...options, queries: options.queries.map((query: any) => { if (process.env.NODE_ENV !== 'production') { if (query.queryFn === skipToken) { console.error('skipToken is not allowed for useSuspenseQueries') } } return { ...query, suspense: true, throwOnError: defaultThrowOnError, enabled: true, placeholderData: undefined, } }), }, queryClient, ) }