ez_docker/plugins/module_utils/label/traefik_router.py
2025-01-17 18:31:17 +01:00

53 lines
1.6 KiB
Python

# 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,
}