puppetlabs-stdlib/collect_indices.rb
Krzysztof Wilczynski 908459c1ad First version. Function that allows to collect selected indices
from an array within Puppet manifest.  More or less how array slicing
works in Perl ...

Signed-off-by: Krzysztof Wilczynski <krzysztof.wilczynski@linux.com>
2011-04-23 17:45:49 +01:00

27 lines
647 B
Ruby

#
# collect_indices.rb
#
module Puppet::Parser::Functions
newfunction(:collect_indices, :type => :rvalue, :doc => <<-EOS
EOS
) do |arguments|
raise(Puppet::ParseError, "Wrong number of arguments " +
"given (#{arguments.size} for 2)") if arguments.size < 2
array = arguments.shift
indices = *arguments # Get them all ... Pokemon ...
if not indices or indices.empty?
raise(Puppet::ParseError, 'You must provide indices to collect')
end
# In Puppet numbers are often string-encoded ...
array = indices.collect { |i| array[i.to_i] }.compact
return array
end
end
# vim: set ts=2 sw=2 et :