# 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 = {
    "name": {"type": "str"},
    "rule": {"type": "str", "required": True},
    "proxy_type": {"type": "str", "default": "http"},
    "service": {"type": "str"},
    "certresolver": {"type": "str"},
    "entrypoints": {"type": "list"},
    "middlewares": {"type": "list"},
}


def helper(state: State, service_name: str, params: dict[str, Any]) -> dict[str, Any]:
    project_name: str = state.module.params["name"]
    rule: str = params["rule"]
    traefik_service: str | None = params.get("service")
    entrypoints: list[str] | None = params.get("entrypoints")
    middlewares: list[str] | None = params.get("middlewares")
    certresolver: str | None = params.get("certresolver")
    proxy_type: str = params["proxy_type"]
    name: str = params.get("name") or f"{project_name}_{service_name}_{proxy_type}"

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

    labels = {
        "traefik.enable": True,
        f"traefik.{prefix}.rule": rule,
    }

    if certresolver:
        labels[f"traefik.{prefix}.tls.certresolver"] = certresolver

    if entrypoints:
        labels[f"{prefix}.entrypoints"] = ",".join(entrypoints)

    if traefik_service:
        labels[f"{prefix}.service"] = traefik_service

    if middlewares:
        labels[f"{prefix}.middlewares"] = ",".join(middlewares)

    return {
        "labels": labels,
    }