Update index.js

index.js

@@ -2,12 +2,14 @@ const express = require('express')
 const app = express()
 const port = 7860
 const axios = require('axios')
-const { connectToMongoDB } = require('./database');
+const { Database } = require('./database');
 
+const dbClient = new Database("AkenoXJs", "FastJsAPI");
+  
 const startup = async () => {
     try {
         console.log("Starting application...");
-        await connectToMongoDB();
+        await dbClient.connect();
         console.log("MongoDB connected successfully.");
     } catch (error) {
         console.error("Error during startup:", error.message);
@@ -22,63 +24,6 @@ const startServer = async () => {
     });
 };
 
-const IPAddressAndUpdate = async (ip) => {
-    try {
-        const db = await connectToMongoDB();
-        const collection = db.collection("FastJsAPI");
-        const filter = { ip: ip };
-        const update = { $set: { ip: ip } };
-        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);
-    }
-};
-
-const AddIpisBlocked = async (ip) => {
-    try {
-        const db = await connectToMongoDB();
-        const collection = db.collection("FastJsAPI");
-        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);
-    }
-};
-
-
-const CheckIsBlocked = async (ip) => {
-    try {
-        const db = await connectToMongoDB();
-        const collection = db.collection("FastJsAPI");
-        const filter = { ip: ip };
-        const FindIp = await collection.findOne(filter);
-
-        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;
-    }
-};
-
 app.get('/', (req, res) => {
   res.redirect('https://t.me/RendyProjects');
 });
@@ -101,17 +46,17 @@ app.use(async (req, res, next) => {
   }
 
   req.realIP = realIP;
-  const isBlocked = await CheckIsBlocked(realIP)
+  const isBlocked = await dbClient.CheckIsBlocked(realIP)
   if (isBlocked && isBlocked.blocked === true) {
     return res.status(403).send("Access denied: IP is blocked");
   }
   if (req.path === '/env') {
     console.log("check path /env");
-    await AddIpisBlocked(realIP)
+    await dbClient.AddIpisBlocked(realIP)
     return res.status(403).send("Access denied: IP is blocked..");
   } 
   
-  await IPAddressAndUpdate(realIP);
+  await dbClient.IPAddressAndUpdate(realIP);
   console.log(`Real IP address is: ${realIP}, header: ${xForwardedFor ? "x-forwarded-for" : xRealIP ? "x-real-ip" : cfConnectingIP ? "cf-connecting-ip" : "req.ip"} `);
   await next();
 });