From e427758e5a3804bf5fb6df449fb4fca6dad33f05 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=A3o=20Seixas?= <joaoseixas88@gmail.com>
Date: Thu, 5 Jun 2025 19:47:14 -0300
Subject: [PATCH] chore: add generic param to useWebSocket for custom event map

---
 src/lib/use-websocket.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/lib/use-websocket.ts b/src/lib/use-websocket.ts
index 82f60e5..976c2a3 100644
--- a/src/lib/use-websocket.ts
+++ b/src/lib/use-websocket.ts
@@ -15,11 +15,11 @@ import {
 } from './types';
 import { assertIsWebSocket } from './util';
 
-export const useWebSocket = <T = unknown>(
+export const useWebSocket = <T = unknown, P = WebSocketEventMap['message'] | null>(
   url: string | (() => string | Promise<string>) | null,
   options: Options = DEFAULT_OPTIONS,
   connect: boolean = true,
-): WebSocketHook<T> => {
+): WebSocketHook<T, P> => {
   const [lastMessage, setLastMessage] = useState<WebSocketEventMap['message'] | null>(null);
   const [readyState, setReadyState] = useState<ReadyStateState>({});
   const lastJsonMessage: T = useMemo(() => {