feat: reconnect ws on abnormal disconnection

package.json

@@ -1,6 +1,6 @@
 {
   "name": "3commas-typescript",
-  "version": "1.3.1",
+  "version": "1.3.2",
   "description": "TypeScript 3Commas API client",
   "repository": {
     "type": "git",
@@ -27,21 +27,21 @@
   "license": "MIT",
   "devDependencies": {
     "@types/crypto-js": "^4.0.2",
-    "@types/node": "^16.0.0",
-    "@types/qs": "^6.9.6",
-    "@types/ws": "^7.4.6",
+    "@types/node": "^16.7.4",
+    "@types/qs": "^6.9.7",
+    "@types/ws": "^7.4.7",
     "dotenv": "^10.0.0",
-    "husky": "^7.0.1",
-    "lint-staged": "^11.0.0",
+    "husky": "^7.0.2",
+    "lint-staged": "^11.1.2",
     "prettier": "^2.3.2",
     "ts-node-dev": "^1.1.8",
-    "typescript": "^4.3.5"
+    "typescript": "^4.4.2"
   },
   "dependencies": {
     "axios": "^0.21.1",
-    "crypto-js": "^4.0.0",
+    "crypto-js": "^4.1.1",
     "qs": "^6.10.1",
-    "ws": "^7.5.2"
+    "ws": "^8.2.0"
   },
   "lint-staged": {
     "*.{ts,json,md}": "prettier --write"

src/index.ts

@@ -12,6 +12,7 @@ import {
   ThreeCommasError,
   TransferHistoryParams,
   TransferParams,
+  WebsocketCallback,
 } from './types/types';
 import { sign } from './lib/crypto';
 import { Convert, Order } from './types/generated-types';
@@ -361,23 +362,36 @@ export class API {
   private subscribe(
     channel: Channel,
     url: string,
-    callback?: (data: WebSocket.Data) => void
+    callback?: WebsocketCallback
   ) {
     const payload = JSON.stringify({
       identifier: this.buildIdentifier(channel, url),
       command: 'subscribe',
     });
-
-    if (!this.ws) {
+    const setUpWebsocketListener = (callback?: WebsocketCallback) => {
+      if (callback) {
+        this.ws?.on('message', (data: Buffer, isBinary: boolean) => {
+          const message = isBinary ? data : data.toString();
+          callback(message);
+        });
+      }
+      this.ws?.on('close', (code) => {
+        if (code === 1006) {
+          setUpWebsocket(payload);
+        }
+      });
+    };
+    const setUpWebsocket = (payload: string) => {
       this.ws = new WebSocket(WS);
       this.ws.onopen = () => this.ws?.send(payload);
+      setUpWebsocketListener(callback);
+    };
+
+    if (!this.ws) {
+      setUpWebsocket(payload);
     } else {
       this.ws.send(payload);
     }
-
-    if (callback) {
-      this.ws.on('message', callback);
-    }
   }
 
   subscribeSmartTrade(callback?: (data: WebSocket.Data) => void) {