"""세션 영속성에서 공유하는 정규화·마스킹 도우미."""

from __future__ import annotations

from datetime import datetime, timezone
from typing import Any

from .services import guardrail


def _ts(value: datetime | None) -> float | None:
    if value is None:
        return None
    if value.tzinfo is None:
        value = value.replace(tzinfo=timezone.utc)
    return value.timestamp()


def _clean_text(value: Any) -> str | None:
    if value is None:
        return None
    text = str(value).strip()
    return text or None


def _clean_masked_text(value: Any) -> str | None:
    text = _clean_text(value)
    if text is None:
        return None
    masked = guardrail.mask_pii(text).text_masked.strip()
    return masked or None


def _clean_role_masked_text(
    value: Any,
    *,
    counselor_identity: str | None = None,
    client_identity: str | None = None,
    synthetic_generated: bool = False,
) -> str | None:
    text = _clean_text(value)
    if text is None:
        return None
    masked = guardrail.mask_role_identities(
        text,
        counselor_identity=counselor_identity,
        client_identity=client_identity,
        synthetic_generated=synthetic_generated,
    ).text_masked.strip()
    return masked or None


def _mask_json_text_values(
    value: Any,
    *,
    counselor_identity: str | None = None,
    client_identity: str | None = None,
    synthetic_generated: bool = False,
) -> Any:
    if isinstance(value, str):
        return (
            _clean_role_masked_text(
                value,
                counselor_identity=counselor_identity,
                client_identity=client_identity,
                synthetic_generated=synthetic_generated,
            )
            or ""
        )
    if isinstance(value, dict):
        return {
            key: _mask_json_text_values(
                child,
                counselor_identity=counselor_identity,
                client_identity=client_identity,
                synthetic_generated=synthetic_generated,
            )
            for key, child in value.items()
        }
    if isinstance(value, list):
        return [
            _mask_json_text_values(
                child,
                counselor_identity=counselor_identity,
                client_identity=client_identity,
                synthetic_generated=synthetic_generated,
            )
            for child in value
        ]
    if isinstance(value, tuple):
        return [
            _mask_json_text_values(
                child,
                counselor_identity=counselor_identity,
                client_identity=client_identity,
                synthetic_generated=synthetic_generated,
            )
            for child in value
        ]
    return value