Function withCustomEventResponse

packages/client/src/ws/util/websocket.util.ts:103–152

({
  timeout = 10_000,
  addEventListener,
  removeEventListener
}: {
  timeout?: number;
  addEventListener: (event: string, handler: Function) => void;
  removeEventListener: (predicate: (listener: any) => boolean) => void;
})

}): {
  responseId: string;
  promise: Promise<T>;
  controller: AbortController;
} {
  const responseId = generateId('res');
  const controller = new AbortController();
  const signal = controller.signal;

  // Create a promise that resolves when we get a matching Custom event
  const promise = new Promise<T>((resolve, reject) => {
    // Set up temporary event listener for Custom events
    const handleCustomEvent = (eventData: any) => {
      const data = eventData?.data;
      // Check if this is our response by matching responseId
      if (data?.eventName === responseId) {
        controller.abort(); // Clean up
        resolve(data?.args as T);
      }
    };

    // Add listener with cleanup
    addEventListener('Custom.Event', handleCustomEvent);

    // Set up timeout
    const timeoutId = setTimeout(() => {
      controller.abort();
      reject(new Error('Timed out waiting for Custom Event'));
    }, timeout);

    // Cleanup function
    signal.addEventListener(
      'abort',
      () => {
        clearTimeout(timeoutId);
        // Remove the event listener
        removeEventListener(
          (listener) =>
            listener.events?.includes('Custom.Event') && listener.callback === handleCustomEvent,
        );
      },
      { once: true },
    );
  });

  return { responseId, promise, controller };
}