connection.py

"""MongoDB connection manager."""
import ssl
from typing import Optional

import certifi
import structlog
from motor.motor_asyncio import AsyncIOMotorClient, AsyncIOMotorDatabase

logger = structlog.get_logger()


class MongoDBConnection:
    """MongoDB connection manager."""
    
    def __init__(self) -> None:
        self._client: Optional[AsyncIOMotorClient] = None
        self._database: Optional[AsyncIOMotorDatabase] = None
    
    async def connect(self, mongodb_url: str, database_name: str) -> None:
        """Connect to MongoDB."""
        try:
            # Configure SSL settings for MongoDB Atlas
            if "mongodb+srv://" in mongodb_url:
                # Create SSL context with proper certificate verification
                ssl_context = ssl.create_default_context(cafile=certifi.where())
                ssl_context.check_hostname = False  # Atlas handles this
                ssl_context.verify_mode = ssl.CERT_REQUIRED
                
                # For MongoDB Atlas, we need to handle SSL properly
                self._client = AsyncIOMotorClient(
                    mongodb_url,
                    tls=True,
                    tlsCAFile=certifi.where(),
                    retryWrites=True,
                    serverSelectionTimeoutMS=5000,  # 5 second timeout
                    connectTimeoutMS=10000,  # 10 second connection timeout
                    socketTimeoutMS=20000,   # 20 second socket timeout
                )
            else:
                # For local MongoDB
                self._client = AsyncIOMotorClient(mongodb_url)
            
            self._database = self._client[database_name]
            
            # Test connection with timeout
            await self._client.admin.command("ping")
            logger.info("Connected to MongoDB", database=database_name)
        
        except Exception as e:
            logger.error("Failed to connect to MongoDB", error=str(e))
            raise
    
    async def disconnect(self) -> None:
        """Disconnect from MongoDB."""
        if self._client:
            self._client.close()
            logger.info("Disconnected from MongoDB")
    
    @property
    def database(self) -> AsyncIOMotorDatabase:
        """Get database instance."""
        if self._database is None:
            raise RuntimeError("Database connection not established")
        return self._database
    
    @property
    def client(self) -> AsyncIOMotorClient:
        """Get client instance."""
        if self._client is None:
            raise RuntimeError("Database connection not established")
        return self._client


# Global connection instance
mongodb_connection = MongoDBConnection()