Update plugins/antiban.js

plugins/antiban.js

@@ -21,30 +21,48 @@ AntibanRoutes.post("/api/v1/user/ban-user", authenticateApiKey, apiLimiter, asyn
             return res.status(400).json({ error: "Invalid or missing user_id" });
         }
 
-        const existingUser = await collection.findOne({ user_ban: userIdNumber });
         const existingUserKey = await collectionKey.findOne({ key: apiKey });
-
         if (!existingUserKey) {
             return res.status(401).json({ message: "API key not found" });
         }
 
-        if (existingUser) {
-            return res.status(200).json({ message: `User is already banned: ${existingUser.user_ban}`, is_ban: true });
+        const existingBan = await collection.findOne({ user_ban: userIdNumber });
+        if (existingBan) {
+            return res.status(200).json({ message: `User ${userIdNumber} is already banned`, is_ban: true });
         }
 
-        const userDocument = {
-            owner: existingUserKey.owner,
-            key: existingUserKey.key,
-            user_ban: userIdNumber,
-            createdAt: new Date(),
-        };
+        await collection.updateOne(
+            { key: existingUserKey.key },
+            { $addToSet: { user_ban: userIdNumber }, $set: { updatedAt: new Date(), owner: existingUserKey.owner } },
+            { upsert: true }
+        );
+
+        res.json({ message: `User ${userIdNumber} successfully banned`, is_ban: true });
+
+    } catch (error) {
+        res.status(500).json({ error: `Internal server error: ${error.message}` });
+    }
+});
+
+AntibanRoutes.get("/api/v1/user/check-ban", authenticateApiKey, async (req, res) => {
+    const dbClient = new Database("AkenoXJs");
+    const collection = dbClient.collection("ban_users");
+
+    try {
+        const userIdString = req.query.user_id;
+        const userIdNumber = Number(userIdString);
 
-        if (reasonString) {
-            userDocument.reason = reasonString;
+        if (isNaN(userIdNumber)) {
+            return res.status(400).json({ error: "Invalid or missing user_id" });
         }
 
-        await collection.insertOne(userDocument);
-        res.json({ message: `User successfully banned: ${userIdNumber}`, is_ban: true });
+        const bannedUser = await collection.findOne({ user_ban: userIdNumber });
+
+        if (bannedUser) {
+            return res.status(200).json({ message: `User ${userIdNumber} is banned`, is_ban: true });
+        } else {
+            return res.status(200).json({ message: `User ${userIdNumber} is not banned`, is_ban: false });
+        }
 
     } catch (error) {
         res.status(500).json({ error: `Internal server error: ${error.message}` });