import { useQuery } from '@tanstack/react-query';
import { api } from '@common/services/api';
import type { SimulationStatusResponse } from '../services/predictionApi';

export const useSimulationStatus = (execSn: number | null) => {
  return useQuery<SimulationStatusResponse>({
    queryKey: ['simulationStatus', execSn],
    queryFn: () => api.get<SimulationStatusResponse>(`/simulation/status/${execSn}`).then(r => r.data),
    enabled: execSn !== null,
    refetchInterval: (query) => {
      const status = query.state.data?.status;
      if (status === 'DONE' || status === 'ERROR') return false;
      return 3000;
    },
  });
};