import { type UseMutationOptions, useMutation } from '@tanstack/react-query';
import type { RpcSchema } from '@yaakapp-internal/proxy-lib';
import { minPromiseMillis } from '@yaakapp-internal/ui';
import type { Req, Res } from '../lib/rpc';
import { rpc } from '../lib/rpc';

/**
 * React Query mutation wrapper for RPC commands.
 */
export function useRpcMutation<K extends keyof RpcSchema>(
  cmd: K,
  opts?: Omit<UseMutationOptions<Res<K>, Error, Req<K>>, 'mutationFn'>,
) {
  return useMutation<Res<K>, Error, Req<K>>({
    mutationFn: (payload) => minPromiseMillis(rpc(cmd, payload)),
    ...opts,
  });
}