feat: support headers options with submit cookie

starknt · Jul 7, 2024 · 3aaec14 · 3aaec14
1 parent f47fc53
commit 3aaec14
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/base/types.ts b/src/base/types.ts
@@ -153,6 +153,8 @@ interface CustomWebSocket {
 }
 
 export interface WSOptions extends BaseOptions, AuthOptions, ConnectionOptions {
+  headers?: { [key: string]: string } | undefined
+
   ssl?: boolean
   platform?: 'web' | string
   protover?: 1 | 2 | 3

diff --git a/src/node.ts b/src/node.ts
@@ -276,7 +276,9 @@ export class KeepLiveWS<E extends Record<EventKey, any> = object> extends LiveCl
         }
       }
       else {
-        socket = new WebSocket(_url)
+        socket = new WebSocket(_url, {
+          headers: options.headers,
+        })
         socket.binaryType = 'arraybuffer'
       }
       if (socket) {
@@ -304,7 +306,9 @@ export class KeepLiveWS<E extends Record<EventKey, any> = object> extends LiveCl
       }
     }
     else {
-      socket = new WebSocket(url)
+      socket = new WebSocket(url, {
+        headers: options.headers,
+      })
       socket.binaryType = 'arraybuffer'
     }