more to update puppet facts

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

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