import os
from motor.motor_asyncio import AsyncIOMotorClient
from typing import Optional

mongodb_client: Optional[AsyncIOMotorClient] = None
database = None

async def connect_to_mongo():
    global mongodb_client, database
    mongodb_url = os.getenv("MONGODB_URL", "mongodb://localhost:27017")
    mongodb_client = AsyncIOMotorClient(mongodb_url)
    database = mongodb_client.oauth_db

async def close_mongo_connection():
    global mongodb_client
    if mongodb_client:
        mongodb_client.close()

async def get_database():
    return database