diff options
Diffstat (limited to 'gateway/src/events')
-rw-r--r-- | gateway/src/events/Close.ts | 46 | ||||
-rw-r--r-- | gateway/src/events/Connection.ts | 79 | ||||
-rw-r--r-- | gateway/src/events/Message.ts | 47 |
3 files changed, 0 insertions, 172 deletions
diff --git a/gateway/src/events/Close.ts b/gateway/src/events/Close.ts deleted file mode 100644 index 5b7c512c..00000000 --- a/gateway/src/events/Close.ts +++ /dev/null @@ -1,46 +0,0 @@ -import { WebSocket } from "@fosscord/gateway"; -import { - emitEvent, - PresenceUpdateEvent, - PrivateSessionProjection, - Session, - SessionsReplace, - User, -} from "@fosscord/util"; - -export async function Close(this: WebSocket, code: number, reason: string) { - console.log("[WebSocket] closed", code, reason); - if (this.heartbeatTimeout) clearTimeout(this.heartbeatTimeout); - if (this.readyTimeout) clearTimeout(this.readyTimeout); - this.deflate?.close(); - this.removeAllListeners(); - - if (this.session_id) { - await Session.delete({ session_id: this.session_id }); - const sessions = await Session.find({ - where: { user_id: this.user_id }, - select: PrivateSessionProjection, - }); - await emitEvent({ - event: "SESSIONS_REPLACE", - user_id: this.user_id, - data: sessions, - } as SessionsReplace); - const session = sessions.first() || { - activities: [], - client_info: {}, - status: "offline", - }; - - await emitEvent({ - event: "PRESENCE_UPDATE", - user_id: this.user_id, - data: { - user: await User.getPublicUser(this.user_id), - activities: session.activities, - client_status: session?.client_info, - status: session.status, - }, - } as PresenceUpdateEvent); - } -} diff --git a/gateway/src/events/Connection.ts b/gateway/src/events/Connection.ts deleted file mode 100644 index 4954cd08..00000000 --- a/gateway/src/events/Connection.ts +++ /dev/null @@ -1,79 +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 { createDeflate } from "zlib"; -import { URL } from "url"; -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 -) { - try { - // @ts-ignore - socket.on("close", Close); - // @ts-ignore - socket.on("message", Message); - console.log(`[Gateway] Connections: ${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); - } - - // @ts-ignore - 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 = createDeflate({ chunkSize: 65535 }); - socket.deflate.on("data", (chunk) => socket.send(chunk)); - } - - 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); - } -} diff --git a/gateway/src/events/Message.ts b/gateway/src/events/Message.ts deleted file mode 100644 index acc39bb9..00000000 --- a/gateway/src/events/Message.ts +++ /dev/null @@ -1,47 +0,0 @@ -import { CLOSECODES, OPCODES } from "../util/Constants"; -import { WebSocket, Payload } from "@fosscord/gateway"; -var erlpack: any; -try { - erlpack = require("@yukikaze-bot/erlpack"); -} catch (error) {} -import OPCodeHandlers from "../opcodes"; -import { Tuple } from "lambert-server"; -import { check } from "../opcodes/instanceOf"; -import WS from "ws"; - -const PayloadSchema = { - op: Number, - $d: new Tuple(Object, Number), // or number for heartbeat sequence - $s: Number, - $t: String, -}; - -export async function Message(this: WebSocket, buffer: WS.Data) { - // TODO: compression - var data: Payload; - - if (this.encoding === "etf" && buffer instanceof Buffer) - data = erlpack.unpack(buffer); - else if (this.encoding === "json" && typeof buffer === "string") - data = JSON.parse(buffer); - else return; - - check.call(this, PayloadSchema, data); - - // @ts-ignore - const OPCodeHandler = OPCodeHandlers[data.op]; - if (!OPCodeHandler) { - console.error("[Gateway] Unkown opcode " + data.op); - // TODO: if all opcodes are implemented comment this out: - // this.close(CLOSECODES.Unknown_opcode); - return; - } - - try { - return await OPCodeHandler.call(this, data); - } catch (error) { - console.error(error); - if (!this.CLOSED && this.CLOSING) - return this.close(CLOSECODES.Unknown_error); - } -} |