1 files changed, 7 insertions, 3 deletions
diff --git a/share/hydra/nodes b/share/hydra/nodes
index 455b00b..b070680 100755
--- a/share/hydra/nodes
+++ b/share/hydra/nodes
@@ -54,9 +54,13 @@ hydra_node_files
 
 # Get nodes
 if [ ! -z "$CLASS" ]; then
-  FILES="`hydra_node_lines | cut -d ':' -f 1`"
-  FILES="`grep nodo::$CLASS $FILES | cut -d ':' -f 1`"
-  hydra_nodes
+  hydra_nodes | while read node; do
+    find $PUPPET/hiera -name "$node.yaml"
+  done | while read candidate; do
+    if grep -e "$CLASS" $candidate | grep -qe "^nodo::role"; then
+      basename $candidate .yaml
+    fi
+  done
 else
   hydra_nodes
 fi