from flask_pymongo import PyMongo
from flask import Flask
from configs import Config


def init_db(app: Flask) -> PyMongo:
    try:
        mongo = PyMongo(app)  # Initialize PyMongo with the app
        app.config["MONGO_URI"] = Config.MONGO_URI
        print(f"Connecting to MongoDB: {Config.MONGO_URI}")

        mongo.cx.server_info()  # Ping the MongoDB server
        print("✅ MongoDB connection successful!")

        return mongo

    except Exception as e:
        print(f"❌ MongoDB connection failed: {e}")
        return None  # Handle failure gracefully