From 0e9bcca72da1d1100c6f7df9d3588593f78b86b5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?R=C3=A2u=20Cao?= <raucao@kosmos.org>
Date: Sun, 5 Jul 2026 16:19:36 +0200
Subject: [PATCH] Add vm_host label to prometheus targets

---
 site-cookbooks/kosmos_prometheus/recipes/server.rb             | 1 +
 .../kosmos_prometheus/templates/default/prometheus.yml.erb     | 3 +++
 2 files changed, 4 insertions(+)

diff --git a/site-cookbooks/kosmos_prometheus/recipes/server.rb b/site-cookbooks/kosmos_prometheus/recipes/server.rb
index a7b2bb1..345bd94 100644
--- a/site-cookbooks/kosmos_prometheus/recipes/server.rb
+++ b/site-cookbooks/kosmos_prometheus/recipes/server.rb
@@ -79,6 +79,7 @@ jobs = node["kosmos_prometheus"]["jobs"].merge(
       "targets" => search(:node, config["query"]).map do |n|
         target = { "target" => "#{n['knife_zero']['host']}:#{config['port']}", "instance" => n.name }
         target["env"] = n.chef_environment if n.chef_environment
+        target["vm_host"] = n["vm_host"] if n["vm_host"]
         target
       end.compact.sort_by { |t| t["instance"] },
     }
diff --git a/site-cookbooks/kosmos_prometheus/templates/default/prometheus.yml.erb b/site-cookbooks/kosmos_prometheus/templates/default/prometheus.yml.erb
index 9cc3ea4..a7bd9f3 100644
--- a/site-cookbooks/kosmos_prometheus/templates/default/prometheus.yml.erb
+++ b/site-cookbooks/kosmos_prometheus/templates/default/prometheus.yml.erb
@@ -36,6 +36,9 @@ scrape_configs:
 <% if t['env'] %>
         env: <%= t['env'] %>
 <% end %>
+<% if t['vm_host'] %>
+        vm_host: <%= t['vm_host'] %>
+<% end %>
 <% end %>
 <% end %>