umami/src/components/hooks/queries/useWebsiteCohortsQuery.ts

import { useApi } from '../useApi';
import { useModified } from '../useModified';
import { keepPreviousData } from '@tanstack/react-query';
import { ReactQueryOptions } from '@/lib/types';
import { useFilterParameters } from '@/components/hooks/useFilterParameters';

export function useWebsiteCohortsQuery(
  websiteId: string,
  params?: Record<string, string>,
  options?: ReactQueryOptions,
) {
  const { get, useQuery } = useApi();
  const { modified } = useModified(`cohorts`);
  const filters = useFilterParameters();

  return useQuery({
    queryKey: ['website:cohorts', { websiteId, modified, ...filters, ...params }],
    queryFn: pageParams => {
      return get(`/websites/${websiteId}/segments`, { ...pageParams, ...filters, ...params });
    },
    enabled: !!websiteId,
    placeholderData: keepPreviousData,
    ...options,
  });
}