gistfile1.txt

import {
  useQuery,
  QueryKey,
  QueryFunction,
  UseQueryResult
} from 'react-query';

export function useAdaptedQuery<ApiReturnType, ReturnType = ApiReturnType>(
  queryKey: QueryKey,
  fn: QueryFunction<ApiReturnType>,
  options: {
    keepPreviousData?: boolean;
    onSuccess?: (data: ReturnType) => void;
    select?: (data: ApiReturnType) => ReturnType;
    staleTime?: number;
  } = {}
):
  | [undefined, true, Error | null, UseQueryResult<ReturnType, Error>]
  | [undefined, false, Error, UseQueryResult<ReturnType, Error>]
  | [ReturnType, false, null, UseQueryResult<ReturnType, Error>] {
  const queryResult = useQuery<ApiReturnType, Error, ReturnType>(queryKey, fn, {
    ...options,
    staleTime: options.staleTime || 5 * 60 * 1000, // 5 min by default
  });

  if (queryResult.isLoading)
    return [
      queryResult.data,
      queryResult.isLoading,
      queryResult.error,
      queryResult,
    ] as [undefined, true, Error | null, UseQueryResult<ReturnType, Error>];

  if (queryResult.error)
    return [
      queryResult.data,
      queryResult.isLoading,
      queryResult.error,
      queryResult,
    ] as [undefined, false, Error, UseQueryResult<ReturnType, Error>];

  return [
    queryResult.data,
    queryResult.isLoading,
    queryResult.error,
    queryResult,
  ] as [ReturnType, false, null, UseQueryResult<ReturnType, Error>];
}