83 lines
2 KiB
Python
83 lines
2 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright: (c) 2024, Luca Bilke <luca@bil.ke>
|
|
# MIT License (see LICENSE)
|
|
|
|
# TODO: write ansible sections
|
|
|
|
DOCUMENTATION = r"""
|
|
"""
|
|
|
|
EXAMPLES = r"""
|
|
"""
|
|
|
|
RETURN = r"""
|
|
"""
|
|
|
|
from __future__ import annotations # pyright: ignore[reportGeneralTypeIssues]
|
|
|
|
from ansible_collections.snailed.ez_compose.plugins.module_utils.common import (
|
|
State,
|
|
run_label,
|
|
update_service,
|
|
)
|
|
|
|
|
|
def helper(state: State) -> State:
|
|
service_name = state.module.params["name"]
|
|
project_name = state.module.params["project_name"]
|
|
rule = state.module.params["rule"]
|
|
service = state.module.params.get("service")
|
|
entrypoints = state.module.params.get("entrypoints")
|
|
middlewares = state.module.params.get("middlewares")
|
|
certresolver = state.module.params.get("certresolver")
|
|
proxy_type = state.module.params.get("proxy_type", "http")
|
|
router_name = state.module.params.get(
|
|
"router_name",
|
|
f"{project_name}_{service_name}_{proxy_type}",
|
|
)
|
|
|
|
prefix = f"traefik.{proxy_type}.routers.{router_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 service:
|
|
labels[f"{prefix}.service"] = service
|
|
|
|
if middlewares:
|
|
labels[f"{prefix}.middlewares"] = ",".join(middlewares)
|
|
|
|
update = {
|
|
"labels": labels,
|
|
}
|
|
|
|
return update_service(state, update)
|
|
|
|
|
|
def main():
|
|
extra_args = {
|
|
"proxy_type": {"type": "str"},
|
|
"router_name": {
|
|
"type": "str",
|
|
},
|
|
"rule": {"type": "str", "required": True},
|
|
"service": {"type": "str"},
|
|
"certresolver": {"type": "str"},
|
|
"entrypoints": {"type": "list"},
|
|
"middlewares": {"type": "list"},
|
|
}
|
|
run_label(extra_args, helper)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|