from __future__ import annotations

"""Helpers for merging incoming user-configuration updates with what is already
stored, while honouring masked API keys.
"""

from typing import Dict

from api.schemas.user_configuration import UserConfiguration
from api.services.configuration.masking import (
    SERVICE_SECRET_FIELDS,
    resolve_masked_api_keys,
)

SERVICE_FIELDS = ("llm", "tts", "stt", "embeddings", "realtime")


def merge_user_configurations(
    existing: UserConfiguration, incoming_partial: Dict[str, dict]
) -> UserConfiguration:
    """Merge *incoming_partial* onto *existing* and return a new UserConfiguration.

    *incoming_partial* is the body of the PUT request (already `model_dump()`ed or
    extracted via Pydantic `model_dump`).

    Rules:
    1. If a service block is absent in the request, keep the existing one.
    2. If provider unchanged and the api_key field is either missing or equal to
       the masked placeholder, preserve the existing real key.
    3. If provider changes, the incoming api_key is used verbatim (validation
       will fail later if it is missing).
    4. Non-service top-level fields (e.g. `test_phone_number`) are overwritten
       when supplied.
    """

    merged = existing.model_dump(exclude_none=True)

    def _merge_service_block(service_name: str):
        incoming_cfg = incoming_partial.get(service_name)
        if incoming_cfg is None:
            return  # nothing to do

        old_cfg = merged.get(service_name, {})

        provider_changed = (
            old_cfg.get("provider") is not None
            and incoming_cfg.get("provider") is not None
            and incoming_cfg.get("provider") != old_cfg.get("provider")
        )

        if not provider_changed:
            for secret_field in SERVICE_SECRET_FIELDS:
                incoming_secret = incoming_cfg.get(secret_field)
                if incoming_secret is not None:
                    if old_cfg and secret_field in old_cfg:
                        incoming_cfg[secret_field] = resolve_masked_api_keys(
                            incoming_secret, old_cfg[secret_field]
                        )
                elif secret_field in old_cfg:
                    incoming_cfg[secret_field] = old_cfg[secret_field]

        merged[service_name] = incoming_cfg

    for service in SERVICE_FIELDS:
        _merge_service_block(service)

    # other simple fields
    if "is_realtime" in incoming_partial:
        merged["is_realtime"] = incoming_partial["is_realtime"]

    if "test_phone_number" in incoming_partial:
        merged["test_phone_number"] = incoming_partial["test_phone_number"]

    if "timezone" in incoming_partial:
        merged["timezone"] = incoming_partial["timezone"]

    return UserConfiguration.model_validate(merged)