# Copyright: (c) 2025, Luca Bilke <luca@bil.ke>
# MIT License (see LICENSE)

from __future__ import annotations

from typing import TYPE_CHECKING, Any

if TYPE_CHECKING:
    from ansible_collections.snailed.ez_compose.plugins.module_utils.models import State

EXTRA_ARGS = {
    "proxy_type": {"type": "str", "default": "http"},
    "name": {"type": "string"},
    "middleware": {"type": "str", "required": True},
    "settings": {"type": "list", "required": True},
}


def helper(state: State, service_name: str, params: dict[str, Any]) -> dict[str, Any]:
    project_name: str = state.module.params["name"]
    middleware: str = params["middleware"]
    settings: dict[str, str] = params["settings"]
    proxy_type: str = state.module.params["proxy_type"]
    name: str = (
        params.get("name") or f"{project_name}_{service_name}_{proxy_type}_{middleware.lower()}"
    )

    prefix = f"traefik.{proxy_type}.middlewares.{name}"

    labels = {f"{prefix}.{middleware}.{key}": var for key, var in settings.items()}

    return {
        "labels": labels,
    }