38 lines
1.1 KiB
Python
38 lines
1.1 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_docker.plugins.module_utils.models import State
|
|
|
|
EXTRA_ARGS = {
|
|
"proxy_type": {"type": "str", "default": "http"},
|
|
"name": {"type": "str"},
|
|
"port": {"type": "int"},
|
|
"protocol_version": {"type": "int"},
|
|
}
|
|
|
|
|
|
def helper(state: State, service_name: str, params: dict[str, Any]) -> dict[str, Any]:
|
|
project_name: str = state.params["name"]
|
|
port: int | None = params.get("port")
|
|
protocol_version: int | None = params.get("protocol_version")
|
|
proxy_type: str = params["proxy_type"]
|
|
name: str = params.get("name", f"{project_name}_{service_name}_{proxy_type}")
|
|
|
|
prefix = f"traefik.{proxy_type}.services.{name}"
|
|
|
|
labels: dict[str, Any] = {}
|
|
|
|
if port:
|
|
labels[f"{prefix}.loadbalancer.server.port"] = str(port)
|
|
|
|
if protocol_version:
|
|
labels[f"{prefix}.loadbalancer.proxyProtocol.version"] = str(protocol_version)
|
|
|
|
return {
|
|
"labels": labels,
|
|
}
|