From 8870751d26e11afd592d85f85c3f236ceb8d903c Mon Sep 17 00:00:00 2001
From: Silvio Rhatto <rhatto@riseup.net>
Date: Sun, 27 May 2018 10:53:26 -0300
Subject: Deploy: support for class names

---
 share/hydra/deploy | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'share/hydra')

diff --git a/share/hydra/deploy b/share/hydra/deploy
index 9a43edf..015f81a 100755
--- a/share/hydra/deploy
+++ b/share/hydra/deploy
@@ -31,6 +31,16 @@ NODES="$*"
 # Build node list
 if [ -z "$NODES" ]; then
   NODES="`hydra $HYDRA nodes`"
+else
+  # Check if first param is a node name, otherwise assume its a class
+  if [ -z "$2" ]; then
+    NODEFILE="$(basename `find $HYDRA_FOLDER/puppet/config/node/ -name $1'.*' | head -n 1` .yaml)"
+
+    # Check for config/node name
+    if [ "$NODEFILE" == ".yaml" ]; then
+      NODES="`hydra $HYDRA nodes $1`"
+    fi
+  fi
 fi
 
 # Deploy requirements
-- 
cgit v1.2.3