import { useConfig } from '@openmrs/esm-framework';
import { useMemo } from 'react';
import { ExpressWorkflowConfig } from '../../config-schema';
import { useQueueEntries } from '../../hooks/useServiceQueues';
import { Queue } from '../../types';

export const useTriageQueuesMetrics = (queue?: Queue) => {
  const { queueServiceConceptUuids, queueStatusConceptUuids } = useConfig<ExpressWorkflowConfig>();
  const {
    queueEntries: waitingEntries,
    isLoading: isLoadingWaiting,
    error: waitingError,
  } = useQueueEntries({
    service: [queueServiceConceptUuids.triageService],
    statuses: [queueStatusConceptUuids.waitingStatus],
    location: queue?.location?.uuid ? [queue.location.uuid] : undefined,
  });
  const {
    queueEntries: inServiceEntries,
    isLoading: isLoadingInService,
    error: inServiceError,
  } = useQueueEntries({
    service: [queueServiceConceptUuids.triageService],
    statuses: [queueStatusConceptUuids.inServiceStatus],
    location: queue?.location?.uuid ? [queue.location.uuid] : undefined,
  });
  const {
    queueEntries: finishedEntries,
    isLoading: isLoadingFinished,
    error: finishedError,
  } = useQueueEntries({
    service: [queueServiceConceptUuids.triageService],
    statuses: [queueStatusConceptUuids.finishedStatus],
    location: queue?.location?.uuid ? [queue.location.uuid] : undefined,
  });
  const { queueEntries: attendedtoEntries } = useQueueEntries({
    service: [queueServiceConceptUuids.triageService],
    statuses: [queueStatusConceptUuids.inServiceStatus, queueStatusConceptUuids.finishedStatus],
    location: queue?.location?.uuid ? [queue.location.uuid] : undefined,
  });

  return {
    isLoading: isLoadingFinished || isLoadingInService || isLoadingWaiting,
    waitingEntries: useMemo(
      () => waitingEntries.filter((entry) => entry?.queue?.uuid === queue?.uuid),
      [waitingEntries, queue],
    ),
    inServiceEntries: useMemo(
      () => inServiceEntries.filter((entry) => entry?.queue?.uuid === queue?.uuid),
      [inServiceEntries, queue],
    ),
    finishedEntries: useMemo(
      () => finishedEntries.filter((entry) => entry?.queue?.uuid === queue?.uuid),
      [finishedEntries, queue],
    ),
    attendedtoEntries: useMemo(
      () => attendedtoEntries.filter((entry) => entry?.queue?.uuid === queue?.uuid),
      [attendedtoEntries, queue],
    ),
    error: finishedError ?? inServiceError ?? waitingError,
  };
};