feat: implement reconnection state management in LavalinkNode

Author: ductridev

src/structures/Node.ts

@@ -20,6 +20,7 @@ import type { NodeManager } from "./NodeManager";
 import type {
     BaseNodeStats, LavalinkInfo, LavalinkNodeOptions, LyricsResult, ModifyRequest, NodeStats, SponsorBlockSegment
 } from "./Types/Node";
+import { ReconnectionState } from "./Types/Node";
 /**
  * Lavalink Node creator class
  */
@@ -70,6 +71,8 @@ export class LavalinkNode {
     private reconnectTimeout?: NodeJS.Timeout = undefined;
     /** The Reconnection Attempt counter */
     private reconnectAttempts = 1;
+    /** Reconnection current state */
+    private reconnectionState: ReconnectionState = ReconnectionState.IDLE;
     /** The Socket of the Lavalink */
     private socket: WebSocket | null = null;
     /** Version of what the Lavalink Server should be */
@@ -1025,16 +1028,25 @@ export class LavalinkNode {
      * ```
      */
     private reconnect(instaReconnect = false): void {
+        // If already trying to reconnect or pending, return
+        if (this.reconnectionState != ReconnectionState.IDLE) {
+            return;
+        }
+
+        // Set reconnection state to pending
+        this.reconnectionState = ReconnectionState.PENDING;
         this.NodeManager.emit("reconnectinprogress", this);
         if (instaReconnect) {
             if (this.reconnectAttempts >= this.options.retryAmount) {
                 const error = new Error(`Unable to connect after ${this.options.retryAmount} attempts.`)
 
                 this.NodeManager.emit("error", this, error);
+                this.reconnectionState = ReconnectionState.IDLE;
                 return this.destroy(DestroyReasons.NodeReconnectFail);
             }
 
             this.NodeManager.emit("reconnecting", this);
+            this.reconnectionState = ReconnectionState.RECONNECTING;
             this.connect();
             this.reconnectAttempts++;
             return;
@@ -1044,10 +1056,12 @@ export class LavalinkNode {
             if (this.reconnectAttempts >= this.options.retryAmount) {
                 const error = new Error(`Unable to connect after ${this.options.retryAmount} attempts.`)
                 this.NodeManager.emit("error", this, error);
+                this.reconnectionState = ReconnectionState.IDLE;
                 return this.destroy(DestroyReasons.NodeReconnectFail);
             }
 
             this.NodeManager.emit("reconnecting", this);
+            this.reconnectionState = ReconnectionState.RECONNECTING;
             this.connect();
             this.reconnectAttempts++;
         }, this.options.retryDelay || 1000);
@@ -1057,6 +1071,9 @@ export class LavalinkNode {
     private async open(): Promise<void> {
         this.isAlive = true;
 
+        // Reset reconnection state on successful connection
+        this.reconnectionState = ReconnectionState.IDLE;
+
         this.reconnectAttempts = 1;
         if (this.reconnectTimeout) clearTimeout(this.reconnectTimeout);
 
@@ -1122,7 +1139,10 @@ export class LavalinkNode {
 
         if (code !== 1000 || reason !== "Node-Destroy") {
             if (this.NodeManager.nodes.has(this.id)) { // try to reconnect only when the node is still in the nodeManager.nodes list
-                this.reconnect();
+                // Only reconnect if not already in progress
+                if (this.reconnectionState === ReconnectionState.IDLE) {
+                    this.reconnect();
+                }
             }
         }