import logging.config
import sys
from typing import Optional

from audience_common.context import get_correlation_id, get_request_context
from audience_common.logger.constants import LogFormat
from audience_common.logger.formatters import DDJsonFormatter, DebugFormatter
from audience_common.utils.dictutil import flatten


class OwsStreamHandler(logging.StreamHandler):  # type: ignore[type-arg]
    def __init__(
        self,
        environment: str,
        service_name: str,
        service_version: str,
        logger_name: Optional[str] = None,
        log_format: LogFormat = LogFormat.JSON,
    ) -> None:
        super().__init__(stream=sys.stdout)
        if log_format in [LogFormat.DEBUG, LogFormat.DEBUG_EXTRA]:
            self.formatter = DebugFormatter(
                debug_extra=log_format == LogFormat.DEBUG_EXTRA
            )
        else:
            self.formatter = DDJsonFormatter(
                environment=environment,
                service_name=service_name,
                service_version=service_version,
                logger_name=logger_name,
                pretty=log_format == LogFormat.JSON_PRETTY,
            )

    def emit(self, record: logging.LogRecord) -> None:
        if correlation_id := get_correlation_id():
            record.correlation_id = correlation_id
        if request_context := get_request_context():
            record.request_context = flatten(
                request_context.dict(exclude_empty=True, exclude_keys=["authorization"])
            )
        super().emit(record)