From 38dc5981fd9b3987ee8f3e80c39b9addaee8981a Mon Sep 17 00:00:00 2001
From: Moe <moe@chocola.dev>
Date: Thu, 7 Apr 2022 11:39:23 +0700
Subject: [PATCH] Add Enable/Disable OTP Feature

---
 routes/api.js | 53 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)

diff --git a/routes/api.js b/routes/api.js
index 80c2243..2154829 100644
--- a/routes/api.js
+++ b/routes/api.js
@@ -353,6 +353,59 @@ router.post("/otp-submit", async (req, res) => {
     }
 });
 
+router.post("/otp-toggle", async (req, res) => {
+    if (!req.query.key) {
+        res.status(401);
+        res.json({
+            message: "Unauthorized",
+        });
+    } else {
+        auth.auth_checker(req.query.key).then((status) => {
+            if (status) {
+                const conn = db.connect();
+                conn.query(
+                    "SELECT * FROM users WHERE token = ?",
+                    [req.query.key],
+                    async function (error, response, fields) {
+                        if (error) {
+                            res.status(401);
+                            res.json({
+                                message: "Unauthorized",
+                            });
+                        } else {
+                            let otp = response[0].otp == 1 ? true : false;
+                            if (otp) {
+                                otp = false;
+                            } else {
+                                otp = true;
+                            }
+                            conn.query(
+                                'UPDATE users SET ? WHERE token ="' + req.query.key + '"', {
+                                    otp: otp == true ? 1 : 0,
+                                },
+                                function (err, result) {
+                                    if (err) {
+                                        res.status(400);
+                                        res.json({
+                                            message: "Bad Request",
+                                        });
+                                    } else {
+                                        res.status(200);
+                                        res.json({
+                                            message: `Success set OTP to ${otp}`,
+                                        });
+                                    }
+                                    db.disconnect(conn);
+                                }
+                            );
+                        }
+                    }
+                );
+            }
+        });
+    }
+});
+
 router.post("/verify-mail", async (req, res) => {
     const conn = db.connect();
     conn.query(