cameleer-server/ui/src/api/queries/agents.ts

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

export function useAgents(status?: string, application?: string, environment?: string) {
  const refetchInterval = useRefreshInterval(10_000);
  return useQuery({
    queryKey: ['agents', status, application, environment],
    queryFn: async () => {
      const token = useAuthStore.getState().accessToken;
      const params = new URLSearchParams();
      if (status) params.set('status', status);
      if (application) params.set('application', application);
      if (environment) params.set('environment', environment);
      const qs = params.toString();
      const res = await fetch(`${config.apiBaseUrl}/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();
    },
    refetchInterval,
  });
}

export function useAgentEvents(appId?: string, agentId?: string, limit = 50, toOverride?: string) {
  const refetchInterval = useRefreshInterval(15_000);
  return useQuery({
    queryKey: ['agents', 'events', 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}/agents/events-log?${params}`, {
        headers: {
          Authorization: `Bearer ${token}`,
          'X-Cameleer-Protocol-Version': '1',
        },
      });
      if (!res.ok) throw new Error('Failed to load agent events');
      return res.json();
    },
    refetchInterval,
  });
}