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


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

        mongo.cx.server_info()
        print("✅ MongoDB connection successful!")

        return mongo

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