type KeyGetterFn = (index: number, previousPageData: Steps | null) => string;

const getKey = (queryParams: {
  limit?: number;
  offset?: number;
  organisationId?: string;
  search?: string;
  stepStatuses?: StepStatusEnum[];
}): KeyGetterFn => {
  return (index: number, previousPageData: Steps | null): string => {
    const offset = index * (queryParams?.limit ?? 10);

    const encodedQueryParams = encodeQueryParameters({
      ...(queryParams ?? {}),
      offset,
    });

    return `${baseUrl}/steps?${encodedQueryParams}`;
  };
};

export const useStepsGetInfinite = (
  {
    queryParams,
  }: {
    queryParams: {
      limit?: number;
      offset?: number;
      organisationId?: string;
      search?: string;
      stepStatuses?: StepStatusEnum[];
    };
  },
  requestInitArgs?: RequestInit,
  swrOptions?: SWRInfiniteConfiguration,
) => {
  const router = useRouter();

  const { data, error, isLoading, mutate, isValidating, size, setSize } = useSWRInfinite<Steps>(
    getKey(queryParams),
    () =>
      stepsGetFetch({ queryParams, ...requestInitArgs }).then(async (response) => {
        if (response.status === 401) {
          router.push(`${process.env.NEXT_PUBLIC_FE_API_BASE_URL}/auth/sign-in` as Route);
          return;
        }

        const resBody = await response.json();

        if (!response.ok) {
          console.log({
            resBody,
            stringified: JSON.stringify(resBody),
            errorJson: new Error(resBody),
            errorString: JSON.stringify(resBody),
          });
          throw new SwrError(resBody.message, resBody.statusCode);
        }

        return resBody;
      }),
    swrOptions,
  );

  return {
    data,
    isLoading,
    isError: !!error,
    error,
    mutate,
    isValidating,
    size,
    setSize,
  };
};