import { useQuery } from '@tanstack/react-query';
import { config } from '../../config';
import { useAuthStore } from '../../auth/auth-store';
import { useEnvironmentStore } from '../environment-store';
import { useRefreshInterval } from './use-refresh-interval';

export function useAgents(status?: string, application?: string) {
  const environment = useEnvironmentStore((s) => s.environment);
  const refetchInterval = useRefreshInterval(10_000);
  return useQuery({
    queryKey: ['agents', environment, status, application],
    queryFn: async () => {
      const token = useAuthStore.getState().accessToken;
      const params = new URLSearchParams();
      if (status) params.set('status', status);
      if (application) params.set('application', application);
      const qs = params.toString();
      const res = await fetch(
        `${config.apiBaseUrl}/environments/${encodeURIComponent(environment!)}/agents${qs ? `?${qs}` : ''}`,
        {
          headers: {
            Authorization: `Bearer ${token}`,
            'X-Cameleer-Protocol-Version': '1',
          },
        });
      if (!res.ok) throw new Error('Failed to load agents');
      return res.json();
    },
    enabled: !!environment,
    refetchInterval,
  });
}

export function useAgentEvents(appId?: string, agentId?: string, limit = 50, toOverride?: string) {
  const environment = useEnvironmentStore((s) => s.environment);
  const refetchInterval = useRefreshInterval(15_000);
  return useQuery({
    queryKey: ['agents', 'events', environment, appId, agentId, limit, toOverride],
    queryFn: async () => {
      const token = useAuthStore.getState().accessToken;
      const params = new URLSearchParams();
      if (appId) params.set('appId', appId);
      if (agentId) params.set('agentId', agentId);
      if (toOverride) params.set('to', toOverride);
      params.set('limit', String(limit));
      const res = await fetch(
        `${config.apiBaseUrl}/environments/${encodeURIComponent(environment!)}/agents/events?${params}`,
        {
          headers: {
            Authorization: `Bearer ${token}`,
            'X-Cameleer-Protocol-Version': '1',
          },
        });
      if (!res.ok) throw new Error('Failed to load agent events');
      return res.json();
    },
    enabled: !!environment,
    refetchInterval,
  });
}