import type {
  DefaultError,
  FetchQueryOptions,
  GetNextPageParamFunction,
  InitialPageParam,
  QueryFunction,
} from '@tanstack/react-query';
import type { AreAllOptional } from './AreAllOptional.js';
import type { DeepReadonly } from './DeepReadonly.js';
import type { OperationInfiniteData } from './OperationInfiniteData.js';
import type { PartialParameters } from './PartialParameters.js';
import type { RequestFn } from './RequestFn.js';
import type { ServiceOperationInfiniteQueryKey } from './ServiceOperationKey.js';

export type ServiceOperationFetchInfiniteQueryOptions<
  TSchema extends { url: string; method: string },
  TQueryFnData,
  TParams,
  TPageParam,
  TError = DefaultError,
> =
  | (FetchInfiniteQueryOptionsByQueryKey<
      TSchema,
      TQueryFnData,
      TParams,
      TPageParam,
      TError
    > &
      FetchInfiniteQueryOptionsQueryFn<TSchema, TQueryFnData, TParams, TError>)
  | (FetchInfiniteQueryOptionsByParameters<
      TSchema,
      TQueryFnData,
      TParams,
      TPageParam,
      TError
    > &
      FetchInfiniteQueryOptionsQueryFn<TSchema, TQueryFnData, TParams, TError>);

export type ServiceOperationEnsureInfiniteQueryDataOptions<
  TSchema extends { url: string; method: string },
  TQueryFnData,
  TParams,
  TPageParam,
  TError = DefaultError,
> =
  | ((FetchInfiniteQueryOptionsByQueryKey<
      TSchema,
      TQueryFnData,
      TParams,
      TPageParam,
      TError
    > &
      FetchInfiniteQueryOptionsQueryFn<
        TSchema,
        TQueryFnData,
        TParams,
        TError
      >) & { revalidateIfStale?: boolean })
  | ((FetchInfiniteQueryOptionsByParameters<
      TSchema,
      TQueryFnData,
      TParams,
      TPageParam,
      TError
    > &
      FetchInfiniteQueryOptionsQueryFn<
        TSchema,
        TQueryFnData,
        TParams,
        TError
      >) & { revalidateIfStale?: boolean });

type FetchInfiniteQueryPages<TQueryFnData = unknown, TPageParam = unknown> =
  | {
      pages?: never;
    }
  | {
      pages: number;
      getNextPageParam: GetNextPageParamFunction<
        PartialParameters<DeepReadonly<TPageParam>>,
        TQueryFnData
      >;
    };

type FetchInfiniteQueryOptionsBase<
  TSchema extends { url: string; method: string },
  TQueryFnData,
  TParams,
  TPageParam = unknown,
  TError = DefaultError,
> = Omit<
  FetchQueryOptions<
    TQueryFnData,
    TError,
    OperationInfiniteData<TQueryFnData, TParams>,
    ServiceOperationInfiniteQueryKey<TSchema, TParams>,
    TPageParam
  >,
  'queryKey' | 'initialPageParam'
> &
  (AreAllOptional<TParams> extends true
    ? Partial<InitialPageParam<PartialParameters<TPageParam>>>
    : InitialPageParam<PartialParameters<TPageParam>>) &
  FetchInfiniteQueryPages<TQueryFnData, TPageParam>;

type FetchInfiniteQueryOptionsByQueryKey<
  TSchema extends { url: string; method: string },
  TQueryFnData,
  TParams,
  TPageParam,
  TError = DefaultError,
> = FetchInfiniteQueryOptionsBase<
  TSchema,
  TQueryFnData,
  TParams,
  TPageParam,
  TError
> &
  (AreAllOptional<TParams> extends true
    ? Partial<FetchInfiniteQueryOptionsQueryKey<TSchema, TParams>>
    : FetchInfiniteQueryOptionsQueryKey<TSchema, TParams>);

type FetchInfiniteQueryOptionsQueryKey<
  TSchema extends { url: string; method: string },
  TParams,
> = {
  /**
   * Fetch Queries by query key
   */
  queryKey: ServiceOperationInfiniteQueryKey<TSchema, TParams>;
  parameters?: never;
};

type FetchInfiniteQueryOptionsByParameters<
  TSchema extends { url: string; method: string },
  TQueryFnData,
  TParams,
  TPageParam,
  TError = DefaultError,
> = FetchInfiniteQueryOptionsBase<
  TSchema,
  TQueryFnData,
  TParams,
  TPageParam,
  TError
> &
  (AreAllOptional<TParams> extends true
    ? Partial<FetchInfiniteQueryOptionsParameters<TParams>>
    : FetchInfiniteQueryOptionsParameters<TParams>);

type FetchInfiniteQueryOptionsParameters<TParams> = {
  /**
   * Fetch Queries by parameters
   */
  parameters: DeepReadonly<TParams>;
  queryKey?: never;
};

type FetchInfiniteQueryOptionsQueryFn<
  TSchema extends { url: string; method: string },
  TQueryFnData,
  TParams,
  TError,
> =
  | {
      queryFn: QueryFunction<
        TQueryFnData,
        ServiceOperationInfiniteQueryKey<TSchema, TParams>
      >;
    }
  | {
      /**
       * Custom request function to use for the query
       */
      requestFn?: RequestFn<TQueryFnData, TError>;
      /**
       * Base URL to use for the request (used in the `queryFn`)
       * @example 'https://api.example.com'
       */
      baseUrl?: string | undefined;
      queryFn?: never; // Workaround to fix union type error
    };