#!/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()