summary refs log tree commit diff
path: root/gateway/src/events/Connection.ts
diff options
context:
space:
mode:
Diffstat (limited to 'gateway/src/events/Connection.ts')
-rw-r--r--gateway/src/events/Connection.ts84
1 files changed, 0 insertions, 84 deletions
diff --git a/gateway/src/events/Connection.ts b/gateway/src/events/Connection.ts
deleted file mode 100644

index bed3cf44..00000000 --- a/gateway/src/events/Connection.ts +++ /dev/null
@@ -1,84 +0,0 @@ -import WS from "ws"; -import { WebSocket } from "@fosscord/gateway"; -import { Send } from "../util/Send"; -import { CLOSECODES, OPCODES } from "../util/Constants"; -import { setHeartbeat } from "../util/Heartbeat"; -import { IncomingMessage } from "http"; -import { Close } from "./Close"; -import { Message } from "./Message"; -import { Deflate, Inflate } from "fast-zlib"; -import { URL } from "url"; -import { Config } from "@fosscord/util"; -var erlpack: any; -try { - erlpack = require("@yukikaze-bot/erlpack"); -} catch (error) {} - -// TODO: check rate limit -// TODO: specify rate limit in config -// TODO: check msg max size - -export async function Connection( - this: WS.Server, - socket: WebSocket, - request: IncomingMessage -) { - const forwardedFor = Config.get().security.forwadedFor; - const ipAddress = forwardedFor ? request.headers[forwardedFor] as string : request.socket.remoteAddress; - - socket.ipAddress = ipAddress; - - try { - // @ts-ignore - socket.on("close", Close); - // @ts-ignore - socket.on("message", Message); - console.log(`[Gateway] New connection from ${socket.ipAddress}, total ${this.clients.size}`); - - const { searchParams } = new URL(`http://localhost${request.url}`); - // @ts-ignore - socket.encoding = searchParams.get("encoding") || "json"; - if (!["json", "etf"].includes(socket.encoding)) { - if (socket.encoding === "etf" && erlpack) { - throw new Error( - "Erlpack is not installed: 'npm i @yukikaze-bot/erlpack'" - ); - } - return socket.close(CLOSECODES.Decode_error); - } - - socket.version = Number(searchParams.get("version")) || 8; - if (socket.version != 8) - return socket.close(CLOSECODES.Invalid_API_version); - - // @ts-ignore - socket.compress = searchParams.get("compress") || ""; - if (socket.compress) { - if (socket.compress !== "zlib-stream") - return socket.close(CLOSECODES.Decode_error); - socket.deflate = new Deflate(); - socket.inflate = new Inflate(); - } - - socket.events = {}; - socket.member_events = {}; - socket.permissions = {}; - socket.sequence = 0; - - setHeartbeat(socket); - - await Send(socket, { - op: OPCODES.Hello, - d: { - heartbeat_interval: 1000 * 30, - }, - }); - - socket.readyTimeout = setTimeout(() => { - return socket.close(CLOSECODES.Session_timed_out); - }, 1000 * 30); - } catch (error) { - console.error(error); - return socket.close(CLOSECODES.Unknown_error); - } -}