database.js

const { MongoClient } = require('mongodb');
require('dotenv').config();

const dbUri = process.env.DB_URI;

if (!dbUri) {
  throw new Error("Missing DB_URI environment variable");
}

const client = new MongoClient(dbUri);

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);

            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 = Database;