From 5bf47c73a7c9df1cf9d933b451b7be22fdc65105 Mon Sep 17 00:00:00 2001
From: uumas <gitcommit@uumas.fi>
Date: Thu, 21 Dec 2023 01:30:00 +0200
Subject: [PATCH] container: add switch for hcloud

---
 roles/prometheus/defaults/main.yml           |  1 +
 roles/prometheus/meta/argument_specs.yml     | 11 ++++++++---
 roles/prometheus/templates/prometheus.yml.j2 |  4 ++--
 3 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/roles/prometheus/defaults/main.yml b/roles/prometheus/defaults/main.yml
index 47b1272..c39adbb 100644
--- a/roles/prometheus/defaults/main.yml
+++ b/roles/prometheus/defaults/main.yml
@@ -4,6 +4,7 @@ prometheus_scrape_interval: 5s
 prometheus_evaluation_interval: 15s
 prometheus_storage_retention: 3650d
 prometheus_install_grafana: false
+prometheus_hcloud_enabled: false
 prometheus_hcloud_relabel_configs: []
 
 prometheus_install_alertmanager: true
diff --git a/roles/prometheus/meta/argument_specs.yml b/roles/prometheus/meta/argument_specs.yml
index cdd612a..58ea97d 100644
--- a/roles/prometheus/meta/argument_specs.yml
+++ b/roles/prometheus/meta/argument_specs.yml
@@ -19,10 +19,15 @@ argument_specs:
         type: str
         required: false
         default: 3650d
-      prometheus_hcloud_token:
-        description: Access token for hetzner cloud service discovery. It will be enabled if this variable is defined
-        type: str
+      prometheus_hcloud_enabled:
+        description: Whether to use hcloud discovery
+        type: bool
         required: false
+        default: false
+      prometheus_hcloud_token:
+        description: Access token for hetzner cloud service discovery.
+        type: str
+        required: "{{ prometheus_hcloud_enabled }}"
       prometheus_hcloud_relabel_configs:
         description: Relabel configs for hcloud
         type: list
diff --git a/roles/prometheus/templates/prometheus.yml.j2 b/roles/prometheus/templates/prometheus.yml.j2
index 21a2ab6..fafa33d 100644
--- a/roles/prometheus/templates/prometheus.yml.j2
+++ b/roles/prometheus/templates/prometheus.yml.j2
@@ -26,11 +26,11 @@ scrape_configs:
     static_configs:
       - targets: ["localhost:9090"]
 
-{% if prometheus_hcloud_token is defined %}
+{% if prometheus_hcloud_enabled %}
   - job_name: hcloud
     hetzner_sd_configs:
       - role: hcloud
         authorization:
           credentials: {{ prometheus_hcloud_token }}
     relabel_configs: {{ prometheus_hcloud_relabel_configs }}
-{% endif %}
+{%- endif %}