[node] Add global WebSocket

Semigradsky · Semigradsky · commit 785e8a17c7e8 · 2024-07-29T20:05:52.000+03:00
diff --git a/types/node/globals.d.ts b/types/node/globals.d.ts
@@ -12,6 +12,7 @@ type _RequestInit = typeof globalThis extends { onmessage: any } ? {}
 type _ResponseInit = typeof globalThis extends { onmessage: any } ? {}
     : import("undici-types").ResponseInit;
 type _File = typeof globalThis extends { onmessage: any } ? {} : import("node:buffer").File;
+type _WebSocket = typeof globalThis extends { onmessage: any } ? {} : import("undici-types").WebSocket;
 // #endregion Fetch and friends
 
 declare global {
@@ -409,4 +410,8 @@ declare global {
         File: infer T;
     } ? T
         : typeof import("node:buffer").File;
+
+    interface WebSocket extends _WebSocket {}
+    var WebSocket: typeof globalThis extends { onmessage: any; WebSocket: infer T } ? T
+        : typeof import("undici-types").WebSocket;
 }
diff --git a/types/node/test/globals-dom.ts b/types/node/test/globals-dom.ts
@@ -48,3 +48,11 @@
         },
     );
 }
+
+{
+    const server = new WebSocket("htps://example.com");
+    server.addEventListener("message", (event) => {
+        console.log(event.data);
+    });
+    server.send("some data");
+}
diff --git a/types/node/test/globals-non-dom.ts b/types/node/test/globals-non-dom.ts
@@ -43,3 +43,11 @@
     const abort = new AbortController();
     AbortSignal.any([abort.signal]); // $ExpectType AbortSignal
 }
+
+{
+    const server = new WebSocket("htps://example.com");
+    server.addEventListener("message", (event) => {
+        console.log(event.data);
+    });
+    server.send("some data");
+}
