fix add api key ban and unban

index.js

@@ -133,56 +133,10 @@ const AllJsonReques = {
   },
 }
 
-
 app.get("/api/v1/json/all", async (req, res) => {
   res.json(AllJsonReques);
 });
 
-app.get('/api/v1/nsfw', async (req, res) => {
-  try {
-    const { url } = req.query;
-    if (!url) {
-      return res.status(400).json({ message: 'Invalid URL.' });
-    }
-
-    const response = await axios.get(url, { responseType: 'arraybuffer' });
-    const imageBuffer = Buffer.from(response.data);
-
-    let imageTensor;
-
-    if (url.endsWith('.gif')) {
-      const jpgBuffer = await sharp(imageBuffer)
-        .resize({ width: 299, height: 299 })
-        .toFormat('jpeg')
-        .toBuffer();
-
-      imageTensor = tf.node.decodeImage(jpgBuffer, 3);
-    } else {
-      imageTensor = tf.node.decodeImage(imageBuffer, 3);
-    }
-
-    const predictions = await model.classify(imageTensor);
-    imageTensor.dispose();
-
-    const formattedPredictions = predictions.reduce((acc, { className, probability }) => {
-      acc[className] = probability;
-      return acc;
-    }, {});
-
-    res.json(formattedPredictions);
-  } catch (error) {
-    console.error('Error processing image:', error);
-    
-    if (error.response) {
-      return res.status(error.response.status).json({ message: 'Error fetching image from URL.', details: error.message });
-    } else if (error.code === 'ERR_INVALID_URL') {
-      return res.status(400).json({ message: 'Invalid image URL.', details: error.message });
-    } else {
-      return res.status(500).json({ message: 'Internal server error.', details: error.message });
-    }
-  }
-});
-
 app.get('/broadcast-user', async (req, res) => {
     const dbClient = new Database("AkenoXJs");
     const collection = dbClient.collection("api_keys");
@@ -249,10 +203,6 @@ app.delete("/api/v1/delete-key", async (req, res) => {
     }
 });
 
-app.get("/api/v1/test", authenticateApiKey, apiLimiter, async (req, res) => {
-  res.json({ message: "access key" });
-})
-
 app.get("/runtime", async (req, res) => {
   res.send("Running lifetime");
 })
@@ -269,6 +219,12 @@ app.post('/api/v2/generate-key', async (req, res) => {
 
     const existingUser = await ApiKey.findOne({ owner: userIdNumber });
 
+    if (existingUser) {
+      return res.status(200).json({
+        is_banned: existingUser.is_banned || false
+      });
+    }
+
     if (existingUser) {
       return res.status(200).json({
         apiKey: existingUser.key,
@@ -288,6 +244,7 @@ app.post('/api/v2/generate-key', async (req, res) => {
       expiresAt: expirationDate,
       owner: userIdNumber,
       type: "premium",
+      is_banned: false,
       email: email || null
     });
     await userDocument.save();
@@ -312,6 +269,12 @@ app.post("/api/v1/revoked-key", async (req, res) => {
 
         const existingUser = await collection.findOne({ owner: userIdNumber });
 
+        if (existingUser) {
+          return res.status(200).json({
+            is_banned: existingUser.is_banned || false
+          });
+        }
+
         if (!existingUser) {
             return res.status(404).json({ error: "API key not found" });
         }
@@ -340,6 +303,99 @@ app.post("/api/v1/revoked-key", async (req, res) => {
     }
 });
 
+app.post("/api/v1/api-key-unban", async (req, res) => {
+    const dbClient = new Database("AkenoXJs");
+    const collection = dbClient.collection("api_keys");
+
+    try {
+        const userIdString = req.query.user_id;
+        const userIdNumber = Number(userIdString);
+
+        if (isNaN(userIdNumber)) {
+            return res.status(400).json({ error: "Invalid or missing user_id" });
+        }
+
+        if (userIdNumber === 0) {
+            return res.status(400).json({ error: "Invalid user_id" });
+        }
+
+        if (userIdNumber === 6477856957) {
+            return res.status(400).json({ error: "sorry only developer" });
+        }
+
+        const existingUser = await collection.findOne({ owner: userIdNumber });
+
+        if (!existingUser) {
+            return res.status(404).json({ error: "API key not found" });
+        }
+
+        const result = await collection.updateOne(
+            { owner: userIdNumber },
+            { $set: { is_banned: false } },
+            { upsert: false }
+        );
+
+        if (result.modifiedCount > 0) {
+            res.json({
+              message: "API key successfully unbanned",
+              owner: userIdNumber
+            });
+        } else {
+            res.status(500).json({ error: "Failed to unban API key" });
+        }
+
+    } catch (err) {
+        res.status(500).json({ error: `Key generation failed: ${err.message}` });
+    }
+});
+
+app.post("/api/v1/api-key-ban", async (req, res) => {
+    const dbClient = new Database("AkenoXJs");
+    const collection = dbClient.collection("api_keys");
+
+    try {
+        const userIdString = req.query.user_id;
+        const userIdNumber = Number(userIdString);
+
+        if (isNaN(userIdNumber)) {
+            return res.status(400).json({ error: "Invalid or missing user_id" });
+        }
+
+        if (userIdNumber === 0) {
+            return res.status(400).json({ error: "Invalid user_id" });
+        }
+
+        if (userIdNumber === 6477856957) {
+            return res.status(400).json({ error: "sorry only developer" });
+        }
+
+        const existingUser = await collection.findOne({ owner: userIdNumber });
+
+        if (!existingUser) {
+            return res.status(404).json({ error: "API key not found" });
+        }
+
+        const result = await collection.updateOne(
+            { owner: userIdNumber },
+            { $set: { is_banned: true } },
+            { upsert: false }
+        );
+
+        if (result.modifiedCount > 0) {
+            res.json({
+              message: "API key successfully banned",
+              owner: userIdNumber
+            });
+        } else {
+            res.status(500).json({ error: "Failed to ban API key" });
+        }
+
+    } catch (err) {
+        res.status(500).json({ error: `Key generation failed: ${err.message}` });
+    }
+});
+
+
 app.post('/api/v1/generate-key', async (req, res) => {
   const dbClient = new Database("AkenoXJs");
   const collection = dbClient.collection('api_keys');
@@ -353,8 +409,19 @@ app.post('/api/v1/generate-key', async (req, res) => {
     if (isNaN(userIdNumber)) {
       return res.status(400).json({ error: "Invalid or missing user_id" });
     }
+
+    if (userIdNumber === 0) {
+      return res.status(400).json({ error: "Invalid user_id" });
+    }
+  
     const existingUser = await collection.findOne({ owner: userIdNumber });
 
+    if (existingUser) {
+      return res.status(200).json({
+        is_banned: existingUser.is_banned || false
+      });
+    }
+
     if (existingUser) {
       return res.status(200).json({
         apiKey: existingUser.key,
@@ -367,6 +434,7 @@ app.post('/api/v1/generate-key', async (req, res) => {
       createdAt: new Date(),
       owner: userIdNumber,
       type: "free",
+      is_banned: false,
       username: username,
     };
     if (email) {

middleware/midware.js

@@ -18,6 +18,10 @@ const authenticateApiKeyPremium = async (req, res, next) => {
       return res.status(403).json({ error: 'Invalid or non-premium API Key' });
     }
 
+    if (keyData.is_banned === true) {
+      return res.status(403).json({ error: 'Banned API Key' });
+    }
+
     if (keyData.expiresAt && new Date() > keyData.expiresAt) {
       return res.status(403).json({ error: 'Premium API Key has expired' });
     }
@@ -41,6 +45,10 @@ const authenticateApiKey = async (req, res, next) => {
     if (!keyDoc) {
       return res.status(403).json({ error: 'Invalid API Key' });
     }
+    if (keyDoc.is_banned === true) {
+      return res.status(403).json({ error: 'Banned API Key' });
+    }
+  
     next();
   } catch (err) {
     res.status(500).json({ error: 'Server error' });

models.js

@@ -3,6 +3,7 @@ import mongoose from "mongoose";
 const apiKeySchema = new mongoose.Schema({
   key: { type: String, required: true, unique: true },
   owner: { type: Number, required: true, unique: true },
+  is_banned: { type: Boolean, default: false },
   email: { type: String },
   createdAt: { type: Date, default: Date.now },
   type: { type: String, enum: ["free", "premium"], default: "free" },