1 files changed, 42 insertions, 0 deletions
diff --git a/server/socket-handlers/database-socket-handler.js b/server/socket-handlers/database-socket-handler.js
new file mode 100644
index 0000000..ee2394b
--- /dev/null
+++ b/server/socket-handlers/database-socket-handler.js
@@ -0,0 +1,42 @@
+const { checkLogin } = require("../util-server");
+const Database = require("../database");
+
+/**
+ * Handlers for database
+ * @param {Socket} socket Socket.io instance
+ * @returns {void}
+ */
+module.exports.databaseSocketHandler = (socket) => {
+
+    // Post or edit incident
+    socket.on("getDatabaseSize", async (callback) => {
+        try {
+            checkLogin(socket);
+            callback({
+                ok: true,
+                size: Database.getSize(),
+            });
+        } catch (error) {
+            callback({
+                ok: false,
+                msg: error.message,
+            });
+        }
+    });
+
+    socket.on("shrinkDatabase", async (callback) => {
+        try {
+            checkLogin(socket);
+            await Database.shrink();
+            callback({
+                ok: true,
+            });
+        } catch (error) {
+            callback({
+                ok: false,
+                msg: error.message,
+            });
+        }
+    });
+
+};