Expose all websocket options

Signed-off-by: Chris Bygrave <[email protected]>

Author: chrisbygrave

lib/firefly.ts

@@ -1,5 +1,7 @@
 import { Stream, Readable } from 'stream';
+import * as http from 'http';
 import * as FormData from 'form-data';
+import * as WebSocket from 'ws';
 import {
   FireFlyStatusResponse,
   FireFlyPrivateSendOptions,
@@ -549,7 +551,7 @@ export default class FireFly extends HttpBase {
   listen(
     subscriptions: string | string[] | FireFlySubscriptionBase,
     callback: FireFlyWebSocketCallback,
-    headers?: Record<string, string>,
+    socketOptions?: WebSocket.ClientOptions | http.ClientRequestArgs,
   ): FireFlyWebSocket {
     const options: FireFlyWebSocketOptions = {
       host: this.options.websocket.host,
@@ -560,7 +562,7 @@ export default class FireFly extends HttpBase {
       autoack: false,
       reconnectDelay: this.options.websocket.reconnectDelay,
       heartbeatInterval: this.options.websocket.heartbeatInterval,
-      headers: headers,
+      socketOptions: socketOptions,
     };
 
     const handler: FireFlyWebSocketCallback = (socket, event) => {

lib/interfaces.ts

@@ -1,4 +1,6 @@
+import * as http from 'http';
 import { AxiosRequestConfig } from 'axios';
+import * as WebSocket from 'ws';
 import { operations } from './schema';
 
 /**
@@ -61,7 +63,7 @@ export interface FireFlyWebSocketOptions {
   autoack: boolean;
   reconnectDelay: number;
   heartbeatInterval: number;
-  headers?: Record<string, string>;
+  socketOptions?: WebSocket.ClientOptions | http.ClientRequestArgs;
 }
 
 // Namespace

lib/websocket.ts

@@ -57,9 +57,9 @@ export class FireFlyWebSocket {
         ? `${this.options.username}:${this.options.password}`
         : undefined;
     const socket = (this.socket = new WebSocket(url, {
+      ...this.options.socketOptions,
       auth,
       handshakeTimeout: this.options.heartbeatInterval,
-      headers: this.options.headers,
     }));
     this.closed = false;