fix: now deleting a line deletes also intents and chatbots

Author: joszamama

routes/LineRoutes.js

@@ -6,6 +6,7 @@ const fs = require('fs');
 const Intent = require('../models/Intent');
 const FormData = require('form-data');
 const fetch = require('node-fetch');
+const Chatbot = require('../models/Chatbot');
 
 const router = express.Router()
 require('dotenv').config();
@@ -282,7 +283,23 @@ router.delete('/mine/:id', async (req, res) => {
         try {
             const pl = await Line.findById(req.params.id);
             if (pl.owner.toString() === decoded.id) {
-                const folderPath = "./lines/" + pl._id;
+
+                // delete all intents from attribute intents in pl
+                for (let i = 0; i < pl.intents.length; i++) {
+                    await Intent.findByIdAndDelete(pl.intents[i]);
+                }
+
+                const chatbots = await Chatbot.find({ pl: pl.id });
+                if (chatbots.length !== 0) {
+                    for (let i = 0; i < chatbots.length; i++) {
+                        const folderPath = "./bots/" + chatbots.id;
+                        fs.rmdirSync(folderPath, { recursive: true });
+                        await Chatbot.findByIdAndDelete(chatbots[i].id);
+                    }
+                }
+
+                // delete the pl file in the server
+                const folderPath = "./lines/" + pl.id;
                 fs.rmdirSync(folderPath, { recursive: true });
                 await pl.remove();
                 res.json({ message: "Line deleted" });
@@ -304,8 +321,22 @@ router.delete('/:id', async (req, res) => {
         const decoded = jwt.verify(token, process.env.JWT_SECRET);
         if (decoded.id === ADMIN_ID) {
             try {
-                const pl = await Line.findById(req.params.id);
-                const folderPath = "./lines/" + pl._id;
+                // delete all intents from attribute intents in pl
+                for (let i = 0; i < pl.intents.length; i++) {
+                    await Intent.findByIdAndDelete(pl.intents[i]);
+                }
+
+                const chatbots = await Chatbot.find({ pl: pl.id });
+                if (chatbots.length !== 0) {
+                    for (let i = 0; i < chatbots.length; i++) {
+                        const folderPath = "./bots/" + chatbots.id;
+                        fs.rmdirSync(folderPath, { recursive: true });
+                        await Chatbot.findByIdAndDelete(chatbots[i].id);
+                    }
+                }
+
+                // delete the pl file in the server
+                const folderPath = "./lines/" + pl.id;
                 fs.rmdirSync(folderPath, { recursive: true });
                 await pl.remove();
                 res.json({ message: "Line deleted" });

routes/UserRoutes.js

@@ -68,17 +68,24 @@ router.get('/:id', async (req, res) => {
 
 // Create a new user
 router.post('/', async (req, res) => {
-    const data = new User({
-        username: req.body.username,
-        email: req.body.email,
-        password: req.body.password
-    })
     try {
-        const dataToSave = await data.save();
-        res.status(200).json(dataToSave.cleanup())
-    }
-    catch (error) {
-        res.status(400).json({ message: error.message })
+        const verify = jwt.verify(req.headers.authorization.split(' ')[1], JWT_SECRET);
+        if (verify) {
+            res.status(404).json({ message: "You cannot create a user while logged in" });
+        }
+    } catch {
+        const data = new User({
+            username: req.body.username,
+            email: req.body.email,
+            password: req.body.password
+        })
+        try {
+            const dataToSave = await data.save();
+            res.status(200).json(dataToSave.cleanup())
+        }
+        catch (error) {
+            res.status(400).json({ message: error.message })
+        }
     }
 })
 
@@ -210,28 +217,35 @@ router.delete('/:id', async (req, res) => {
 
 // Log in user and return user data
 router.post('/login', async (req, res) => {
-    if (!req.body.username || !req.body.password) {
-        return res.status(400).json({ error: 'Username and password are required' });
-    } else {
-        try {
-            const user = await User.findOne({ username: req.body.username });
-            if (user) {
-                const validPassword = await bcrypt.compare(req.body.password, user.password);
-                if (validPassword) {
-                    const payload = user.cleanup();
-                    const accessToken = generateAccessToken(payload, JWT_SECRET);
-                    res.status(200).send({
-                        message: "User authenticated",
-                        accessToken,
-                    });
+    try {
+        const verify = jwt.verify(req.headers.authorization.split(' ')[1], JWT_SECRET);
+        if (verify) {
+            res.status(404).json({ message: "You cannot login a different user while logged in" });
+        }
+    } catch {
+        if (!req.body.username || !req.body.password) {
+            return res.status(400).json({ error: 'Username and password are required' });
+        } else {
+            try {
+                const user = await User.findOne({ username: req.body.username });
+                if (user) {
+                    const validPassword = await bcrypt.compare(req.body.password, user.password);
+                    if (validPassword) {
+                        const payload = user.cleanup();
+                        const accessToken = generateAccessToken(payload, JWT_SECRET);
+                        res.status(200).send({
+                            message: "User authenticated",
+                            accessToken,
+                        });
+                    } else {
+                        res.status(400).json({ message: 'Invalid password' });
+                    }
                 } else {
-                    res.status(400).json({ message: 'Invalid password' });
+                    res.status(400).json({ message: 'User not found' });
                 }
-            } else {
-                res.status(400).json({ message: 'User not found' });
+            } catch (err) {
+                res.status(400).json({ message: 'Cannot login user right now, try again later' });
             }
-        } catch (err) {
-            res.status(400).json({ message: 'Cannot login user right now, try again later' });
         }
     }
 });