from sys import exit
from os.path import join

files = {
    "/etc/deployments/prometheus/kube.yaml": {
        "content_type": "text",
        "source": "kube.yaml",
        "triggers": {"svc_s6rc:prometheus:restart"},
    },
    "/etc/prometheus/prometheus.yml": {
        "content_type": "text",
        "source": "prometheus.yml",
        "triggers": {"svc_s6rc:prometheus:restart"},
    },
}

repo.libs.gen.add_files_recursive(
    files,
    join(repo.path, "bundles", "prometheus", "files", "consoles"),
    target_root="/etc/prometheus",
    files_root=join(repo.path, "bundles", "prometheus", "files"),
)
repo.libs.gen.add_files_recursive(
    files,
    join(repo.path, "bundles", "prometheus", "files", "console_libraries"),
    target_root="/etc/prometheus",
    files_root=join(repo.path, "bundles", "prometheus", "files"),
)

directories = {
        "/etc/prometheus/services": {},
        "/etc/prometheus/consoles": {},
        "/etc/prometheus/console_libraries": {},
}

# NOTE: services that want to have their metrics scraped need to expose them
#       under traefik:4000/metrics/<svc> and set metadata.metrics.<svc>
for node in repo.nodes:
    for job_name, _ in node.metadata.get("metrics", {}).items():
        files[f"/etc/prometheus/services/{node.hostname}.{job_name}.json"] = {
            "content_type": "mako",
            "source": "service.json",
            "context": {
                "job_name": job_name,
                "hostname": node.hostname,
            },
        }