From 471d1d6b28f8a6b62680408756e389e8a0b71be0 Mon Sep 17 00:00:00 2001
From: Flam3rboy <34555296+Flam3rboy@users.noreply.github.com>
Date: Sat, 6 Feb 2021 17:08:08 +0100
Subject: :sparkles: Base Server

---
 src/events/Message.ts | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 src/events/Message.ts

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

diff --git a/src/events/Message.ts b/src/events/Message.ts
new file mode 100644
index 00000000..bc497e94
--- /dev/null
+++ b/src/events/Message.ts
@@ -0,0 +1,36 @@
+import WebSocket, { Data } from "../util/WebSocket";
+import erlpack from "erlpack";
+import OPCodeHandlers from "../opcodes";
+import { Payload, CLOSECODES } from "../util/Constants";
+import { instanceOf } from "lambert-server";
+
+const PayloadSchema = {
+	op: Number,
+	$d: Object,
+	$s: Number,
+	$t: String,
+};
+
+export async function Message(this: WebSocket, buffer: Data) {
+	// TODO: compression
+	var data: Payload;
+
+	try {
+		if (this.encoding === "etf" && buffer instanceof Buffer) data = erlpack.unpack(buffer);
+		else if (this.encoding === "json" && typeof buffer === "string") data = JSON.parse(buffer);
+		if (!instanceOf(PayloadSchema, data)) throw "invalid data";
+	} catch (error) {
+		return this.close(CLOSECODES.Decode_error);
+	}
+
+	// @ts-ignore
+	const OPCodeHandler = OPCodeHandlers[data.op];
+	if (!OPCodeHandler) return this.close(CLOSECODES.Unknown_opcode);
+
+	try {
+		return await OPCodeHandler.call(this, data);
+	} catch (error) {
+		console.error(error);
+		return this.close(CLOSECODES.Unknown_error);
+	}
+}
-- 
cgit 1.5.1