import { useQuery } from '@tanstack/react-query';
import type { GrpcConnection } from '@yaakapp-internal/models';
import { invokeCmd } from '../lib/tauri';

export function grpcConnectionsQueryKey({ requestId }: { requestId: string }) {
  return ['grpc_connections', { requestId }];
}

export function useGrpcConnections(requestId: string | null) {
  return (
    useQuery<GrpcConnection[]>({
      enabled: requestId !== null,
      initialData: [],
      queryKey: grpcConnectionsQueryKey({ requestId: requestId ?? 'n/a' }),
      queryFn: async () => {
        if (requestId == null) return [];
        return (await invokeCmd('cmd_list_grpc_connections', {
          requestId,
          limit: 200,
        })) as GrpcConnection[];
      },
    }).data ?? []
  );
}