from flask_pymongo import PyMongo
from flask import Flask, current_app


def init_db(app: Flask) -> PyMongo:
    try:
        mongo = PyMongo(app)

        mongo.cx.server_info()
        app.logger.info("MongoDB connection established")
        return mongo

    except Exception as e:
        app.logger.error(f"MongoDB connection failed: {e}")
        return None  # Handle failure gracefully