From f44f5d7ac2d24ff836c2e1d4b2fa58da04b13052 Mon Sep 17 00:00:00 2001 From: Madeline <46743919+MaddyUnderStars@users.noreply.github.com> Date: Sun, 25 Sep 2022 18:24:21 +1000 Subject: Refactor to mono-repo + upgrade packages --- src/gateway/Server.ts | 63 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 63 insertions(+) create mode 100644 src/gateway/Server.ts (limited to 'src/gateway/Server.ts') diff --git a/src/gateway/Server.ts b/src/gateway/Server.ts new file mode 100644 index 00000000..7e1489be --- /dev/null +++ b/src/gateway/Server.ts @@ -0,0 +1,63 @@ +import "missing-native-js-functions"; +import dotenv from "dotenv"; +dotenv.config(); +import { closeDatabase, Config, initDatabase, initEvent } from "@fosscord/util"; +import ws from "ws"; +import { Connection } from "./events/Connection"; +import http from "http"; + +export class Server { + public ws: ws.Server; + public port: number; + public server: http.Server; + public production: boolean; + + constructor({ + port, + server, + production, + }: { + port: number; + server?: http.Server; + production?: boolean; + }) { + this.port = port; + this.production = production || false; + + if (server) this.server = server; + else { + this.server = http.createServer(function (req, res) { + res.writeHead(200).end("Online"); + }); + } + + this.server.on("upgrade", (request, socket, head) => { + // @ts-ignore + this.ws.handleUpgrade(request, socket, head, (socket) => { + this.ws.emit("connection", socket, request); + }); + }); + + this.ws = new ws.Server({ + maxPayload: 4096, + noServer: true, + }); + this.ws.on("connection", Connection); + this.ws.on("error", console.error); + } + + async start(): Promise { + await initDatabase(); + await Config.init(); + await initEvent(); + if (!this.server.listening) { + this.server.listen(this.port); + console.log(`[Gateway] online on 0.0.0.0:${this.port}`); + } + } + + async stop() { + closeDatabase(); + this.server.close(); + } +} -- cgit 1.5.1