import { useMemo } from 'react';
import { useFilterId } from '../useFilterId/useFilterId';
import { useFiltersQuery } from '@/domains/filters/hooks/queries/useFiltersQuery/useFiltersQuery';

export type TEntityType = 'individuals' | 'business';

export const useEntityType = (defaultEntityType: TEntityType = 'individuals'): TEntityType => {
  const filterId = useFilterId();
  const { data: filters, isLoading } = useFiltersQuery();

  const entityType = useMemo(() => {
    if (isLoading || !Array.isArray(filters)) return null;

    const matchedFilter = filters.find(filter => filter.id === filterId);

    return matchedFilter ? (matchedFilter.entity as TEntityType) : null;
  }, [filterId, filters, isLoading]);

  return entityType ? entityType : defaultEntityType;
};