import { type UseQueryOptions, useQuery } from '@tanstack/react-query';
import type { RpcSchema } from '@yaakapp-internal/proxy-lib';
import type { Req, Res } from '../lib/rpc';
import { rpc } from '../lib/rpc';

/**
 * React Query wrapper for RPC commands.
 * Automatically caches by [cmd, payload] and supports all useQuery options.
 */
export function useRpcQuery<K extends keyof RpcSchema>(
  cmd: K,
  payload: Req<K>,
  opts?: Omit<UseQueryOptions<Res<K>>, 'queryKey' | 'queryFn'>,
) {
  return useQuery<Res<K>>({
    queryKey: [cmd, payload],
    queryFn: () => rpc(cmd, payload),
    ...opts,
  });
}