From 37d3e5c6463e2b62e34ed06de3b371f5614e7376 Mon Sep 17 00:00:00 2001
From: Silvio Rhatto <rhatto@riseup.net>
Date: Thu, 24 May 2018 11:23:51 -0300
Subject: Adds hydra_hiera_query

---
 lib/hydra/misc | 33 ++++++++++++++++++++++++++++++++-
 1 file changed, 32 insertions(+), 1 deletion(-)

(limited to 'lib')

diff --git a/lib/hydra/misc b/lib/hydra/misc
index 583e43d..77c2988 100644
--- a/lib/hydra/misc
+++ b/lib/hydra/misc
@@ -2,7 +2,7 @@
 
 # Set needed environment variables and do basic checks
 function hydra_set_env {
-  export BASEREPOS="config doc bootless dns git/public git/private puppet"
+  export BASEREPOS="config doc bootless dns git/public git/private puppet ansible"
   export OSVERSION="`cut -d . -f 1 /etc/debian_version`"
   export COMMIT="`( cd $APP_BASE && git log -n 1 --pretty=oneline 2> /dev/null | cut -d " " -f 1 )`"
   export CONFIG="$HOME/.hydra/config"
@@ -170,3 +170,34 @@ function hydra_get_fqdn_from_nodename {
     echo $node.$domain
   fi
 }
+
+function hydra_hiera_query {
+  local node="$1"
+  local param="$2"
+
+  # Build basic query command
+  hiera="hiera --config $HYDRA_FOLDER/puppet/hiera.yaml"
+  hiera_params="settings::confdir=$HYDRA_FOLDER/puppet ::clientcert=$node"
+  role="`$hiera nodo::role $hiera_params 2> /dev/null`"
+  location="`$hiera nodo::location $hiera_params 2> /dev/null`"
+  virtual="$(grep "^virtual: " $FACTS/${node}.yaml | cut -d ':' -f 2 | sed -e 's/ //g' -e 's/"//g')"
+  domain="$(grep "^domain: " $FACTS/${node}.yaml   | cut -d ':' -f 2 | sed -e 's/ //g' -e 's/"//g')"
+
+  if [ ! -z "$virtual" ]; then
+    hiera_params="$hiera_params ::virtual=$virtual"
+  fi
+
+  if [ ! -z "$role" ]; then
+    hiera_params="$hiera_params ::nodo::role=$role"
+  fi
+
+  if [ ! -z "$location" ]; then
+    hiera_params="$hiera_params ::nodo::location=$location"
+  fi
+
+  if [ ! -z "$domain" ]; then
+    hiera_params="$hiera_params ::domain=$domain"
+  fi
+
+  $hiera $param $hiera_params 2> /dev/null
+}
-- 
cgit v1.2.3