Update database.js

database.js

@@ -7,19 +7,99 @@ if (!dbUri) {
   throw new Error("Missing DB_URI environment variable");
 }
 
-const dbName = "AkenoXJs";
 const client = new MongoClient(dbUri);
 
-const connectToMongoDB = async () => {
-    try {
-        await client.connect();
-        console.log("Connected to MongoDB");
-        const db = client.db(dbName);
+class Database {
+    constructor(dbname, collectionName) {
+        this.dbname = dbname;
+        this.collectionName = collectionName;
+    }
+
+    async connect() {
+        try {
+            if (!client.isConnected) {
+                await client.connect();
+                console.log("Connected to MongoDB");
+            }
+        } catch (error) {
+            console.error("Error connecting to MongoDB:", error.message);
+        }
+    }
+
+    collection() {
+        try {
+            const db = client.db(this.dbname);
+            return db.collection(this.collectionName);
+        } catch (error) {
+            console.error("Error accessing collection:", error.message);
+        }
+    }
+
+    async close() {
+        try {
+            await client.close();
+            console.log("MongoDB connection closed");
+        } catch (error) {
+            console.error("Error closing MongoDB connection:", error.message);
+        }
+    }
+
+    async IPAddressAndUpdate(ip) {
+        try {
+            const collection = this.collection(); // Access collection
+            const filter = { ip: ip }; // Filter to find the IP
+            const update = { $set: { ip: ip } }; // Update IP field
+
+            const result = await collection.updateOne(filter, update, { upsert: true });
+
+            if (result.upsertedCount > 0) {
+                console.log("Inserted a new IP address:", ip);
+            } else {
+                console.log("Updated an existing IP address:", ip);
+            }
+        } catch (error) {
+            console.error("Error updating IP address:", error.message);
+        }
+    }
+
+    async AddIpisBlocked(ip) {
+        try {
+            const collection = this.collection();
+            const filter = { ip: ip };
+            const update = { $set: { blocked: true } };
+
+            const result = await collection.updateOne(filter, update, { upsert: true });
+
+            if (result.upsertedCount > 0) {
+                console.log("Inserted a new IP address:", ip);
+            } else {
+                console.log("Updated an existing IP address:", ip);
+            }
+        } catch (error) {
+            console.error("Error updating IP address:", error.message);
+        }
+    }
+
+    async CheckIsBlocked(ip) {
+        try {
+            const collection = this.collection();
+            const filter = { ip: ip };
+            const update = { $set: { blocked: true } };
+
+            const FindIp = await collection.findOne(filter);
 
-        return db;
-    } catch (error) {
-        console.error("Connection failed:", error.message);
+            if (FindIp) {
+                console.log("IP found in the database:", FindIp);
+                return FindIp;
+            } else {
+                console.log("IP not found in the database");
+                return null
+            }
+        } catch (error) {
+            console.error("Error checking IP:", error.message);
+            return null
+        }
     }
 };
 
-module.exports = { client, connectToMongoDB };
+module.exports = Database;