Update index.js

index.js

@@ -30,6 +30,7 @@ import { Database } from './database/database.js'
 import { Readable } from "stream";
 import { randomBytes } from "crypto";
 import { AkenoaiJs } from "akenoaijs";
+import mongoose from "mongoose";
 
 import { 
   CheckMilWare,
@@ -128,6 +129,17 @@ const AllJsonReques = {
     },
   },
 }
+
+const apiKeySchema = new mongoose.Schema({
+  key: { type: String, required: true, unique: true },
+  owner: { type: Number, required: true, unique: true },
+  email: { type: String },
+  createdAt: { type: Date, default: Date.now },
+  expiresAt: { type: Date, required: true, index: { expires: 0 } }
+});
+
+const ApiKey = mongoose.model("ApiKey", apiKeySchema);
+
 app.get("/api/v1/json/all", async (req, res) => {
   res.json(AllJsonReques);
 });
@@ -189,6 +201,44 @@ app.get("/runtime", async (req, res) => {
   res.send("Running lifetime");
 })
 
+app.post('/api/v2/generate-key', async (req, res) => {
+  try {
+    const newKey = generateAkenoKey();
+    const userIdNumber = Number(req.query.user_id);
+    const email = req.query.email;
+
+    if (isNaN(userIdNumber)) {
+      return res.status(400).json({ error: "Invalid or missing user_id" });
+    }
+
+    const existingUser = await ApiKey.findOne({ owner: userIdNumber });
+
+    if (existingUser) {
+      return res.status(200).json({
+        apiKey: existingUser.key,
+        createdAt: existingUser.createdAt,
+        expiresAt: existingUser.expiresAt,
+        owner: existingUser.owner
+      });
+    }
+
+    const expiresAt = new Date();
+    expiresAt.setDate(expiresAt.getDate() + 7);
+
+    const userDocument = new ApiKey({
+      key: newKey,
+      createdAt: new Date(),
+      expiresAt,
+      owner: userIdNumber,
+      email: email || null
+    });
+    await userDocument.save();
+    res.json({ apiKey: newKey, expiresAt });
+  } catch (err) {
+    res.status(500).json({ error: `Key generation failed: ${err.message}` });
+  }
+});
+
 app.post("/api/v1/revoked-key", async (req, res) => {
     const dbClient = new Database("AkenoXJs");
     const collection = dbClient.collection("api_keys");