From 9de9cf7b37979b19831b530355a113bb0ad6f6dc Mon Sep 17 00:00:00 2001
From: flifloo <flifloo@gmail.com>
Date: Tue, 24 Nov 2020 08:47:13 +0100
Subject: [PATCH] Socket getter for database

---
 sockets/get/agendaGet.js   | 39 ++++++++++++++++++++++++++++++++++++++
 sockets/get/groupGet.js    | 21 ++++++++++++++++++++
 sockets/get/semesterGet.js | 15 +++++++++++++++
 sockets/get/userGet.js     | 17 +++++++++++++++++
 sockets/index.js           |  3 ++-
 5 files changed, 94 insertions(+), 1 deletion(-)
 create mode 100644 sockets/get/agendaGet.js
 create mode 100644 sockets/get/groupGet.js
 create mode 100644 sockets/get/semesterGet.js
 create mode 100644 sockets/get/userGet.js

diff --git a/sockets/get/agendaGet.js b/sockets/get/agendaGet.js
new file mode 100644
index 0000000..c27a05c
--- /dev/null
+++ b/sockets/get/agendaGet.js
@@ -0,0 +1,39 @@
+const models = require("../../models");
+
+module.exports = socket => {
+    return async (data) => {
+        let startDate = data.startDate ? data.startDate : new Date(), endDate = data.endDate ? data.endDate : new Date();
+        let options = {where: {startDate: startDate, endDate: endDate}, include: []};
+        if (data.semesters) {
+            for (let semester of data.semesters) {
+                let s = await models.Semester.findByPk(semester);
+                if (!s) {
+                    socket.emit("agendaGet", {error: {message: "semester_not_found"}});
+                    return
+                } else
+                    options.include.push({model: models.Semester, where: {id: s.id}, require: true});
+            }
+        }
+        if (data.groups) {
+            for (let group of data.groups) {
+                let g = await models.Group.findByPk(group);
+                if (!g) {
+                    socket.emit("agendaGet", {error: {message: "group_not_found"}});
+                    return
+                } else
+                    options.include.push({model: models.Group, where: {id: g.id}, require: true});
+            }
+        }
+        if (data.teachers) {
+            for (let teacher of data.teachers) {
+                let t = await models.User.findByPk(teacher);
+                if (!t) {
+                    socket.emit("agendaGet", {error: {message: "teacher_not_found"}});
+                    return
+                } else
+                    options.include.push({model: models.User, where: {id: t.email}, require: true});
+            }
+        }
+        socket.emit("agendaGet", await models.Event.findAll(options));
+    }
+};
diff --git a/sockets/get/groupGet.js b/sockets/get/groupGet.js
new file mode 100644
index 0000000..12f16ba
--- /dev/null
+++ b/sockets/get/groupGet.js
@@ -0,0 +1,21 @@
+const models = require("../../models");
+
+module.exports = socket => {
+    return async (data) => {
+        let options = {where: {}, include: [{model: models.Semester, require: true}]};
+        if (data.number)
+            options.where.number = data.number;
+        if (data.semester) {
+            let s = await models.Semester.findByPk(data.semester);
+            if (!s) {
+                socket.emit("groupGet", {error: {message: "semester_not_found"}});
+                return
+            }
+            options.include[0].where = {id: s.id};
+        }
+        if (data.users)
+            options.include.push({model: models.User, require: true});
+
+        socket.emit("groupGet", await models.Group.findAll(options));
+    }
+};
diff --git a/sockets/get/semesterGet.js b/sockets/get/semesterGet.js
new file mode 100644
index 0000000..a7ef457
--- /dev/null
+++ b/sockets/get/semesterGet.js
@@ -0,0 +1,15 @@
+const models = require("../../models");
+
+module.exports = socket => {
+    return async (data) => {
+        let options = {where: {}, include: []};
+        if (data.name)
+            options.where.name = data.name;
+        if (data.year)
+            options.where.year = data.year;
+        if (data.groups)
+            options.include.push({model: models.Group, require: true});
+
+        socket.emit("semesterGet", await models.Semester.findAll(options));
+    }
+};
diff --git a/sockets/get/userGet.js b/sockets/get/userGet.js
new file mode 100644
index 0000000..0b905c5
--- /dev/null
+++ b/sockets/get/userGet.js
@@ -0,0 +1,17 @@
+const models = require("../../models");
+
+module.exports = socket => {
+    return async (data) => {
+        let options = {where: {}};
+        if (data.email)
+            options.where.email = data.email;
+        if (data.firstName && data.lastName) {
+            options.where.firstName = data.firstName;
+            options.where.lastName = data.lastName;
+        }
+        if (data.permissions)
+            options.where.permissions = data.permissions;
+
+        socket.emit("userGet", await models.User.findAll(options));
+    }
+};
diff --git a/sockets/index.js b/sockets/index.js
index 604338a..b420000 100644
--- a/sockets/index.js
+++ b/sockets/index.js
@@ -10,6 +10,7 @@ module.exports = socket => {
     } else {
         socket.on("profileEdit", require("./profile/edit")(socket));
         socket.on("logout", require("./logout")(socket));
+        socket.on("agendaGet", require("./get/agendaGet")(socket));
     }
     socket.emit("connected");
-}
+};