yaak/apps/yaak-proxy/hooks/useRpcQueryWithEvent.ts

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

/**
 * Combines useRpcQuery with an event listener that invalidates the query
 * whenever the specified event fires, keeping data fresh automatically.
 */
export function useRpcQueryWithEvent<
  K extends keyof RpcSchema,
  E extends keyof RpcEventSchema & string,
>(cmd: K, payload: Req<K>, event: E, opts?: Omit<UseQueryOptions<Res<K>>, 'queryKey' | 'queryFn'>) {
  const queryClient = useQueryClient();
  const query = useRpcQuery(cmd, payload, opts);

  useRpcEvent(event, () => {
    queryClient.invalidateQueries({ queryKey: [cmd, payload] });
  });

  return query;
}