augh

1 files changed, 13 insertions, 8 deletions

diff --git a/webrtc/src/util/Heartbeat.ts b/webrtc/src/util/Heartbeat.ts

index 7b5ed9cd..8c5e3a7a 100644
--- a/webrtc/src/util/Heartbeat.ts
+++ b/webrtc/src/util/Heartbeat.ts
@@ -1,18 +1,23 @@
 import { WebSocket, CLOSECODES } from "@fosscord/gateway";
 import { VoiceOPCodes } from "@fosscord/util";
 
-export async function setHeartbeat(socket: WebSocket) {
+export async function setHeartbeat(socket: WebSocket, nonce?: Number) {
 	if (socket.heartbeatTimeout) clearTimeout(socket.heartbeatTimeout);
 
 	socket.heartbeatTimeout = setTimeout(() => {
 		return socket.close(CLOSECODES.Session_timed_out);
 	}, 1000 * 45);
 
-	socket.send(JSON.stringify({
-		op: VoiceOPCodes.HEARTBEAT_ACK,
-		d: {
-			v: 6,
-			heartbeat_interval: 13750,
-		}
-	}));
+	if (!nonce) {
+		socket.send(JSON.stringify({
+			op: VoiceOPCodes.HELLO,
+			d: {
+				v: 5,
+				heartbeat_interval: 13750,
+			}
+		}));
+	}
+	else {
+		socket.send(JSON.stringify({ op: VoiceOPCodes.HEARTBEAT_ACK, d: nonce }));
+	}
 }
\ No newline at end of file