"""Audit logger connector."""

from functools import wraps

from auditlogger import logger as auditlogger

from collaborator import config


def logger(function):
    """Wrap a function, passing the logger into it if necessary."""

    @wraps(function)
    def wrapper(*args, **kwargs):
        audit_logger = kwargs.pop("logger", None)
        if not audit_logger:
            audit_logger = auditlogger.AuditLogger(config.AUDIT_LOGGER_HOSTS)

        return function(*args, logger=audit_logger, **kwargs)

    return wrapper