fix(use/ws): Correct usage of WS types (#585)

enisdenjo · Nov 5, 2024 · 2c603ad · 2c603ad
1 parent 9cbeee0
commit 2c603ad
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/use/ws.ts b/src/use/ws.ts
@@ -1,5 +1,5 @@
 import type * as http from 'http';
-import type * as ws from 'ws';
+import type { WebSocket, WebSocketServer } from 'ws';
 import { handleProtocols, makeServer, ServerOptions } from '../server';
 import {
   DEPRECATED_GRAPHQL_WS_PROTOCOL,
@@ -10,8 +10,7 @@ import {
 import { limitCloseReason } from '../utils';
 
 // for nicer documentation
-type WebSocket = typeof ws.prototype;
-type WebSocketServer = ws.Server;
+export type { WebSocket, WebSocketServer };
 
 /**
  * The extra that will be put in the `Context`.