"""공통 로깅 설정 — KST(Asia/Seoul) 타임존 포매터 + 민감 정보 마스킹 제공."""

import logging
import datetime
import re
import pytz

KST = pytz.timezone("Asia/Seoul")

# ── 민감 정보 마스킹 패턴 (API 키, 시크릿, 토큰, 계좌번호) ──────────────────────
_MASK_PATTERNS = [
    # Bearer 토큰
    (re.compile(r'(Bearer\s+)[A-Za-z0-9\-_.]{20,}', re.I), r'\1***'),
    # access_token 값
    (re.compile(r'(access_?token[":\s=]+)[A-Za-z0-9\-_.]{20,}', re.I), r'\1***'),
    # appsecret / app_secret 값
    (re.compile(r'(app_?secret[":\s=]+)[A-Za-z0-9+/=]{10,}', re.I), r'\1***'),
    # appkey / app_key 값 (8자 이상)
    (re.compile(r'(app_?key[":\s=]+)[A-Za-z0-9]{8,}', re.I), r'\1***'),
    # 계좌번호 (account_no / account_number)
    (re.compile(r'(account[_\s]?(?:no|number)[":\s=]+)\d{6,}', re.I), r'\1***'),
    # URL 쿼리의 token= (WebSocket 인증 토큰, M-6)
    (_QUERY_TOKEN_RE := re.compile(r'([?&](?:token|api_key|apikey)=)[^&\s"\']+', re.I), r'\1***'),
]


def _mask_sensitive(text: str) -> str:
    """로그 텍스트에서 API 키·토큰·계좌번호 등 민감 정보를 마스킹한다."""
    for pattern, replacement in _MASK_PATTERNS:
        text = pattern.sub(replacement, text)
    return text


class KSTFormatter(logging.Formatter):
    """%(asctime)s 를 KST(UTC+9) 기준으로 출력하고, 민감 정보를 마스킹하는 포매터."""

    def formatTime(self, record: logging.LogRecord, datefmt: str | None = None) -> str:
        ct = datetime.datetime.fromtimestamp(record.created, tz=KST)
        if datefmt:
            return ct.strftime(datefmt)
        t = ct.strftime(self.default_time_format)
        return self.default_msec_format % (t, record.msecs)

    def format(self, record: logging.LogRecord) -> str:
        """포맷팅 후 민감 정보 마스킹 적용."""
        formatted = super().format(record)
        return _mask_sensitive(formatted)


def _redact_query_token(value):
    if isinstance(value, str) and "=" in value:
        return _QUERY_TOKEN_RE.sub(r"\1***", value)
    return value


class QueryTokenRedactionFilter(logging.Filter):
    """레코드의 msg/args 안에 있는 `?token=...` 을 `***` 로 바꾼다 (M-6).

    uvicorn 은 자체 포매터를 쓰므로 KSTFormatter 마스킹이 적용되지 않는다. uvicorn.access 의
    AccessFormatter 는 args 튜플 구조에 의존하므로 args 를 지우지 않고 원소만 치환한다.
    """

    def filter(self, record: logging.LogRecord) -> bool:
        try:
            record.msg = _redact_query_token(record.msg)
            if isinstance(record.args, tuple):
                record.args = tuple(_redact_query_token(a) for a in record.args)
            elif isinstance(record.args, dict):
                record.args = {k: _redact_query_token(v) for k, v in record.args.items()}
        except Exception:  # 로깅 필터가 로그를 막으면 안 된다
            pass  # harness: allow-except-pass 로깅 필터 내부 — 여기서 로그를 남기면 재귀/로그 유실
        return True


_REDACT_LOGGERS = ("uvicorn", "uvicorn.error", "uvicorn.access", "websockets", "websockets.server")


def install_query_token_redaction() -> None:
    """uvicorn/websockets 로거에 토큰 마스킹 필터를 1회 설치(중복 설치 방지)."""
    for name in _REDACT_LOGGERS:
        lg = logging.getLogger(name)
        if not any(isinstance(f, QueryTokenRedactionFilter) for f in lg.filters):
            lg.addFilter(QueryTokenRedactionFilter())