1 files changed, 55 insertions, 0 deletions
diff --git a/src/Server.ts b/src/Server.ts
new file mode 100644
index 00000000..d4b3271c
--- /dev/null
+++ b/src/Server.ts
@@ -0,0 +1,55 @@
+import "missing-native-js-functions";
+import dotenv from "dotenv";
+dotenv.config();
+import { Config, db, RabbitMQ } from "@fosscord/server-util";
+import { Server as WebSocketServer } from "ws";
+import { Connection } from "./events/Connection";
+import http from "http";
+
+export class Server {
+ public ws: WebSocketServer;
+ 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.ws = new WebSocketServer({
+ maxPayload: 4096,
+ server: this.server,
+ });
+ this.ws.on("connection", Connection);
+ this.ws.on("error", console.error);
+ }
+
+ async setupSchema() {
+ // TODO: adjust expireAfterSeconds -> lower
+ await Promise.all([db.collection("events").createIndex({ created_at: 1 }, { expireAfterSeconds: 60 })]);
+ }
+
+ async start(): Promise<void> {
+ // @ts-ignore
+ await (db as Promise<Connection>);
+ await this.setupSchema();
+ await Config.init();
+ await RabbitMQ.init();
+ console.log("[Database] connected");
+ if (!this.server.listening) {
+ this.server.listen(this.port);
+ console.log(`[Gateway] online on 0.0.0.0:${this.port}`);
+ }
+ }
+
+ async stop() {
+ await db.close();
+ this.server.close();
+ }
+}
|
