From d90ad5c955a16a4d4359f287dca0aa6f4b1fdec7 Mon Sep 17 00:00:00 2001
From: PalmDevs <git@palmdevs.me>
Date: Sun, 4 Aug 2024 02:47:00 +0700
Subject: [PATCH] fix(packages/api): handle close event as a disconnect

---
 packages/api/src/classes/ClientWebSocket.ts | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/packages/api/src/classes/ClientWebSocket.ts b/packages/api/src/classes/ClientWebSocket.ts
index 620d068..fd6e9d0 100755
--- a/packages/api/src/classes/ClientWebSocket.ts
+++ b/packages/api/src/classes/ClientWebSocket.ts
@@ -77,9 +77,13 @@ export class ClientWebSocketManager {
             } catch (e) {
                 rj(e)
             }
-        }).finally(() => {
-            this.connecting = false
         })
+            .then(() => {
+                this.#socket.on('close', (code, reason) => this._handleDisconnect(code, reason.toString()))
+            })
+            .finally(() => {
+                this.connecting = false
+            })
     }
 
     /**