1 files changed, 10 insertions, 0 deletions
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