diff options
Diffstat (limited to 'share')
| -rwxr-xr-x | share/hydra/nodes | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/share/hydra/nodes b/share/hydra/nodes index b070680..4ddf916 100755 --- a/share/hydra/nodes +++ b/share/hydra/nodes @@ -18,13 +18,22 @@ # Get a list of puppet node files function hydra_node_files { - if [ -d "$PUPPET/manifests/nodes/" ]; then + if [ -d "$PUPPET/hiera/node/" ]; then + FILES="$PUPPET/hiera/node/*" + elif [ -d "$PUPPET/manifests/nodes/" ]; then FILES="$PUPPET/manifests/nodes/*" else FILES="`find $PUPPET -name 'nodes.pp'`" fi } +# Get using the filename +function hydra_node_names { + for file in $FILES; do + basename $file .yaml + done +} + # Get all lines matching a node definition function hydra_node_lines { # See http://www.mail-archive.com/puppet-users@googlegroups.com/msg01615.html @@ -33,7 +42,7 @@ function hydra_node_lines { # Default node retrieval method function hydra_nodes { - hydra_node_lines | cut -d "'" -f2 + hydra_node_names } # Load |