Update index.js

index.js

@@ -145,37 +145,40 @@ app.get("/v1/show-key", async (req, res) => {
 });
 
 app.post("/v1/revoked-key", 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" });
-    }
+    const dbClient = new Database("AkenoXJs");
+    const collection = dbClient.collection("api_keys");
 
-    const existingUser = await collection.findOne({ owner: userIdNumber });
-  
-    if (existingUser) {
-      const newKey = generateAkenoKey();
-      const result = await collection.updateOne(
-        { owner: existingUser.owner },
-        { $setOnInsert: { key: newKey, createdAt: new Date(), email: existingUser.email || null } },
-        { upsert: true });
-
-      if (result.upsertedCount > 0) {
-        res.json({ message: "New API key generated", "apiKey": newKey });
-      } else {
-        res.json({ message: "API key already exists for this user" });
-      }
+    try {
+        const userIdString = req.query.user_id;
+        const userIdNumber = Number(userIdString);
 
-    } else {
-      res.status(500).json({ error: "API key not found" });
+        if (isNaN(userIdNumber)) {
+            return res.status(400).json({ error: "Invalid or missing user_id" });
+        }
+
+        const existingUser = await collection.findOne({ owner: userIdNumber });
+
+        if (!existingUser) {
+            return res.status(404).json({ error: "API key not found" });
+        }
+
+        const newKey = generateAkenoKey();
+
+        const result = await collection.updateOne(
+            { owner: userIdNumber },
+            { $set: { key: newKey, createdAt: new Date() } },
+            { upsert: false }
+        );
+
+        if (result.modifiedCount > 0) {
+            res.json({ message: "API key successfully revoked and regenerated", apiKey: newKey });
+        } else {
+            res.status(500).json({ error: "Failed to update API key" });
+        }
+
+    } catch (err) {
+        res.status(500).json({ error: `Key generation failed: ${err.message}` });
     }
-  } catch (err) {
-    res.status(500).json({ error: `Key generation failed: ${err.message}` });
-  }
 });
 
 app.post('/v1/generate-key', async (req, res) => {