1 files changed, 47 insertions, 0 deletions
diff --git a/prediction/index.js b/prediction/index.js
new file mode 100644
index 0000000..c5bb81e
--- /dev/null
+++ b/prediction/index.js
@@ -0,0 +1,47 @@
+const express = require("express");
+const db = require("../db");
+const app = express.Router();
+const { checkUser } = require("../auth");
+
+app.use("/", checkUser, express.static(__dirname + "/public"));
+
+app.get("/api/get", checkUser, async (req, res) => {
+    try {
+        const answer = (await db.query("SELECT * FROM teacher_prediction WHERE user_id = ?", [req.session.uid]))[0] || {};
+        // console.log(answer);
+        res.json(answer);
+    } catch (e) {
+        console.error(e);
+        res.json({ success: false });
+    }
+});
+
+app.post("/api/set", checkUser, async (req, res) => {
+    const { teacher } = req.body;
+    if (!teacher) return res.json({ success: false });
+    try {
+        const isTeacher = (await db.query("SELECT id FROM users WHERE type_id = 2 AND id = ?", [teacher])).length > 0;
+        if (!isTeacher) return res.json({ success: false });
+        await db.query("INSERT INTO teacher_prediction (user_id, teacher_id) VALUE (?,?)", [req.session.uid, teacher]);
+        res.json({ success: true });
+    } catch (e) {
+        console.error(e);
+        res.json({ success: false });
+    }
+});
+
+app.put("/api/set", checkUser, async (req, res) => {
+    const { teacher } = req.body;
+    if (!teacher) return res.json({ success: false });
+    try {
+        const isTeacher = (await db.query("SELECT id FROM users WHERE type_id = 2 AND id = ?", [teacher])).length > 0;
+        if (!isTeacher) return res.json({ success: false });
+        await db.query("UPDATE teacher_prediction SET teacher_id = ? WHERE user_id = ?", [+teacher, req.session.uid]);
+        res.json({ success: true });
+    } catch (e) {
+        console.error(e);
+        res.json({ success: false });
+    }
+});
+
+module.exports = app;