From ec04cc6deebd6ade638c03a4a49d0f42d6efc50b Mon Sep 17 00:00:00 2001
From: Maria Haider <mariah@sunet.se>
Date: Mon, 13 May 2024 18:56:03 +0200
Subject: [PATCH] more to update puppet facts

---
 Makefile                                   | 16 +++++++++++++
 global/overlay/etc/puppet/facter/cosmos.rb | 27 ++++++++++++++++++++++
 2 files changed, 43 insertions(+)

diff --git a/Makefile b/Makefile
index a284f955..51765f99 100644
--- a/Makefile
+++ b/Makefile
@@ -1,5 +1,7 @@
 DIST	:= "ubuntu:latest"
 
+FACTS_YAML   := global/overlay/etc/puppet/static-cosmos-facts.yaml
+
 cosmos:
 	fab all cosmos
 
@@ -14,3 +16,17 @@ test_in_docker:
 		-v ${CURDIR}:/multiverse:ro \
 		\
 		$(DIST) /multiverse/scripts/test-in-docker.sh
+
+static-facts:
+	@echo
+	@echo Facts:
+	@echo "-------------------------------------------------------------------------------"
+	mkdir -p $(dir ${FACTS_YAML})
+	./scripts/cosmos-facts \
+		--dirs .komreg.net \
+		--classesfile cosmos-rules.yaml \
+		--ignore_nonexistent \
+		--roles metadata/roles-in.yaml \
+		--outfile ${FACTS_YAML}
+	git add           ${FACTS_YAML} metadata/roles-in.yaml
+	git diff --cached ${FACTS_YAML} metadata/roles-in.yaml
diff --git a/global/overlay/etc/puppet/facter/cosmos.rb b/global/overlay/etc/puppet/facter/cosmos.rb
index d810082f..5f9c98ac 100644
--- a/global/overlay/etc/puppet/facter/cosmos.rb
+++ b/global/overlay/etc/puppet/facter/cosmos.rb
@@ -20,3 +20,30 @@ Facter.add(:cosmos_repo_origin_url) do
   end
 end
 
+# Construct the structured facts under $facts['cosmos']
+Facter.add('cosmos') do
+  require 'yaml'
+
+  # Load YAML file with all the roles and their hosts/addresses
+  res_h = YAML.load_file('/etc/puppet/static-cosmos-facts.yaml')
+
+  # Figure out which roles match the current host
+  fqdn = Facter.value(:fqdn)
+
+  res = []
+  res_h['cosmos'].each do | key, values |
+    if key.end_with? '_hosts'
+      if values.include? fqdn
+        role = key.chomp('_hosts')
+        res = res + [role]
+      end
+    end
+  end
+
+  warn("Cosmos roles for host #{fqdn}: #{res}")
+  res_h['cosmos']['host_roles'] = res
+
+  setcode do
+    res_h['cosmos']
+  end
+end