Update DiscordBot.js

Mistium · Jan 8, 2025 · 71385c8 · 71385c8
1 parent 5926ffd
commit 71385c8
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/featured/DiscordBot.js b/featured/DiscordBot.js
@@ -1,7 +1,7 @@
 (function (Scratch) {
 
   let bot_data = '';
-  
+
   class DiscordBot {
     getInfo() {
       return {
@@ -275,6 +275,16 @@
         }
 
         if (data.op === 10) {
+          // Start heartbeat
+          const heartbeatInterval = data.d.heartbeat_interval;
+          this.heartbeat = setInterval(() => {
+            this.client.send(JSON.stringify({
+              "op": 1,
+              "d": null
+            }));
+          }, heartbeatInterval);
+
+          // Send initial heartbeat
           this.client.send(JSON.stringify({
             "op": 1,
             "d": null
@@ -295,6 +305,13 @@
       this.client.onerror = (error) => {
         console.error('WebSocket error:', error);
       };
+
+      this.client.onclose = () => {
+        if (this.heartbeat) {
+          clearInterval(this.heartbeat);
+          this.heartbeat = null;
+        }
+      };
     }
 
     disconnectFromDiscord() {