diff options
Diffstat (limited to 'gateway/src/events/Message.ts')
-rw-r--r-- | gateway/src/events/Message.ts | 19 |
1 files changed, 15 insertions, 4 deletions
diff --git a/gateway/src/events/Message.ts b/gateway/src/events/Message.ts index b675afcd..83917690 100644 --- a/gateway/src/events/Message.ts +++ b/gateway/src/events/Message.ts @@ -3,7 +3,7 @@ import { WebSocket, Payload } from "@fosscord/gateway"; var erlpack: any; try { erlpack = require("@yukikaze-bot/erlpack"); -} catch (error) {} +} catch (error) { } import OPCodeHandlers from "../opcodes"; import { Tuple } from "lambert-server"; import { check } from "../opcodes/instanceOf"; @@ -22,8 +22,19 @@ export async function Message(this: WebSocket, buffer: WS.Data) { if (this.encoding === "etf" && buffer instanceof Buffer) data = erlpack.unpack(buffer); - else if (this.encoding === "json" && typeof buffer === "string") - data = JSON.parse(buffer); + else if (this.encoding === "json" && buffer instanceof Buffer) { + if (this.inflate) { + try { + buffer = this.inflate.process(buffer) as any; + } catch { + buffer = buffer.toString() as any; + } + } + data = JSON.parse(buffer as string); + } + else if (typeof buffer == "string") { + data = JSON.parse(buffer as string); + } else return; check.call(this, PayloadSchema, data); @@ -42,6 +53,6 @@ export async function Message(this: WebSocket, buffer: WS.Data) { } catch (error) { console.error(error); // if (!this.CLOSED && this.CLOSING) - return this.close(CLOSECODES.Unknown_error); + return this.close(CLOSECODES.Unknown_error); } } |