import logging
import time
from contextlib import contextmanager
from typing import Optional


@contextmanager
def context_logger(operation_name: str, logger: Optional[logging.Logger] = None):
    """
    Context manager for logging operation timing.
    
    Args:
        operation_name: Name of the operation being performed
        logger: Logger instance to use. If None, uses root logger.
        
    Example:
        with context_logger("Loading search data", logger):
            data = load_data()
    """
    if logger is None:
        logger = logging.getLogger(__name__)
        
    start_time = time.time()
    logger.info(f"Starting {operation_name}...")
    
    try:
        yield
        elapsed = time.time() - start_time
        logger.info(f"Completed {operation_name} in {elapsed:.3f}s")
    except Exception as e:
        elapsed = time.time() - start_time
        logger.error(f"Failed {operation_name} after {elapsed:.3f}s: {e}")
        raise