From 3c8688ee34f2afa2e89b01494e96ad60458cf21f Mon Sep 17 00:00:00 2001
From: "Navy.gif" <navydotgif@gmail.com>
Date: Fri, 14 Jul 2023 19:39:05 +0300
Subject: [PATCH] optional loading of mariadb

---
 src/server/Server.ts | 36 +++++++++++++++++++++---------------
 1 file changed, 21 insertions(+), 15 deletions(-)

diff --git a/src/server/Server.ts b/src/server/Server.ts
index d4c031c..62cd58d 100644
--- a/src/server/Server.ts
+++ b/src/server/Server.ts
@@ -9,6 +9,7 @@ import express, { Express, NextFunction } from 'express';
 import helmet from 'helmet';
 import cors from 'cors';
 import MongoStore from 'connect-mongo';
+import { Collection } from '@discordjs/collection';
 
 // Own
 import { LogFunction, LoggerClient } from '@navy.gif/logger';
@@ -25,7 +26,6 @@ import { MongoMemory } from './database/index.js';
 import { IPCMessage, SignupCode } from '../../@types/Other.js';
 import { DoneCallback } from 'passport';
 import FlagManager from './components/FlagManager.js';
-import { Collection } from '@discordjs/collection';
 
 // const pkg = JSON.parse(readFileSync('../../package.json', { encoding: 'utf8' }));
 
@@ -57,7 +57,7 @@ class Server extends EventEmitter
     #server: http.Server | null;
     #app: Express;
 
-    #mariadb: MariaDB;
+    #mariadb?: MariaDB;
     #mongodb: MongoDB;
     #memoryStoreProvider: MongoMemory;
     #userDatabase: UserDatabaseInterface;
@@ -137,16 +137,19 @@ class Server extends EventEmitter
         
         // TODO: Database definitions should probably be elsewhere through injection
         // Mariadb isn't strictly necessary here for anything, it's just here pre-emptively
-        this.#mariadb = new MariaDB(this, {
-            ...databases.mariadb,
-            credentials: {
-                host: MARIA_HOST,
-                user: MARIA_USER,
-                port: parseInt(MARIA_PORT),
-                password: MARIA_PASS,
-                database: MARIA_DB
-            }
-        });
+        if (databases.mariadb?.load)
+        {
+            this.#mariadb = new MariaDB(this, {
+                ...databases.mariadb,
+                credentials: {
+                    host: MARIA_HOST,
+                    user: MARIA_USER,
+                    port: parseInt(MARIA_PORT),
+                    password: MARIA_PASS,
+                    database: MARIA_DB
+                }
+            }); 
+        }
         // Mongo is used for session and user storage
         this.#mongodb = new MongoDB(this, {
             ...databases.mongodb,
@@ -243,8 +246,11 @@ class Server extends EventEmitter
         const start = Date.now();
         this.#logger.status('Starting server');
 
-        this.#logger.info('Initialising MariaDB');
-        this.#mariadb.init();
+        if (this.#mariadb)
+        {
+            this.#logger.info('Initialising MariaDB');
+            this.#mariadb.init(); 
+        }
 
         this.#logger.info('Initialising MongoDB');
         await this.#mongodb.init();
@@ -360,7 +366,7 @@ class Server extends EventEmitter
             this.#server.close(async () => 
             {
                 await this.#mongodb.close();
-                await this.#mariadb.close();
+                await this.#mariadb?.close();
                 await this.#memoryStoreProvider.close();
                 await this.#messageBroker?.close();
                 this.#logger.status('DB shutdowns complete.');