import { UseInfiniteQueryOptions, useInfiniteQuery } from 'react-query';

import { QUERY_KEYS } from 'consts';
import { usePillarContext } from 'context';
import { GetDRepListArguments, getDRepList } from 'services';
import { DRepData, Infinite } from 'types';

export const useGetDRepListInfiniteQuery = (
  {
    filters = [],
    pageSize = 10,
    searchPhrase,
    sorting,
    status,
  }: GetDRepListArguments,
  options?: UseInfiniteQueryOptions<Infinite<DRepData>>
) => {
  const { apiUrl, validationApiUrl, pendingTransaction } = usePillarContext();

  const {
    data,
    isLoading,
    fetchNextPage,
    hasNextPage,
    isFetching,
    isFetchingNextPage,
    isPreviousData,
  } = useInfiniteQuery(
    [
      QUERY_KEYS.useGetDRepListInfiniteKey,
      (
        pendingTransaction.registerAsDirectVoter ||
        pendingTransaction.registerAsDrep ||
        pendingTransaction.retireAsDirectVoter ||
        pendingTransaction.retireAsDrep
      )?.transactionHash ?? 'noPendingTransaction',
      filters.length ? filters : '',
      searchPhrase ?? '',
      sorting ?? '',
      status?.length ? status : '',
    ],
    async ({ pageParam = 0 }) =>
      getDRepList({
        apiUrl,
        validationApiUrl,
        page: pageParam,
        pageSize,
        filters,
        searchPhrase,
        sorting,
        status,
      }),
    {
      getNextPageParam: (lastPage) => {
        if (lastPage.elements.length === 0) {
          return undefined;
        }

        return lastPage.page + 1;
      },
      enabled: options?.enabled,
      keepPreviousData: options?.keepPreviousData,
    }
  );

  return {
    dRepListFetchNextPage: fetchNextPage,
    dRepListHasNextPage: hasNextPage,
    isDRepListFetching: isFetching,
    isDRepListFetchingNextPage: isFetchingNextPage,
    isDRepListLoading: isLoading,
    dRepData: data?.pages.flatMap((page) => page.elements),
    isPreviousData,
  };
};