From e4115d13ef239c2aade5116e58ecc2693092c8e9 Mon Sep 17 00:00:00 2001
From: Flam3rboy <34555296+Flam3rboy@users.noreply.github.com>
Date: Tue, 6 Apr 2021 04:06:32 +0200
Subject: :zap: zlib compression

---
 src/events/Connection.ts | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

(limited to 'src/events/Connection.ts')

diff --git a/src/events/Connection.ts b/src/events/Connection.ts
index 0c4dd2b8..34274d9c 100644
--- a/src/events/Connection.ts
+++ b/src/events/Connection.ts
@@ -5,6 +5,7 @@ import { Message } from "./Message";
 import { setHeartbeat } from "../util/setHeartbeat";
 import { Send } from "../util/Send";
 import { CLOSECODES, OPCODES } from "../util/Constants";
+import { createDeflate } from "zlib";
 
 // TODO: check rate limit
 // TODO: specify rate limit in config
@@ -24,8 +25,14 @@ export async function Connection(this: Server, socket: WebSocket, request: Incom
 		if (socket.version != 8) return socket.close(CLOSECODES.Invalid_API_version);
 
 		// @ts-ignore
-		socket.compression = searchParams.get("compress") || "";
-		// TODO: compression
+		socket.compress = searchParams.get("compress") || "";
+		if (socket.compress) {
+			if (socket.compress !== "zlib-stream") return socket.close(CLOSECODES.Decode_error);
+			socket.deflate = createDeflate({ chunkSize: 65535 });
+			socket.deflate.on("data", (chunk) => socket.send(chunk));
+		}
+
+		socket.sequence = 0;
 
 		setHeartbeat(socket);
 
-- 
cgit 1.5.1