diff options
| author | Krzysztof Wilczynski <krzysztof.wilczynski@linux.com> | 2011-04-26 03:18:50 +0100 |
|---|---|---|
| committer | Krzysztof Wilczynski <krzysztof.wilczynski@linux.com> | 2011-04-26 03:18:50 +0100 |
| commit | 67ea75883f5b3346a20c85146cb177c2176f3cd5 (patch) | |
| tree | 00958f4ca84768566b535e331591a1171a02bde5 | |
| parent | 522557001849ab3cc0de9c8a095558c211b62d63 (diff) | |
Changing name of collect_indices.rb to values_at.rb in order to be
reasemble Ruby's Array#values_at more. Added support for ranges
with hyphen, two and three dots notation.
Signed-off-by: Krzysztof Wilczynski <krzysztof.wilczynski@linux.com>
| -rw-r--r-- | values_at.rb | 77 |
1 files changed, 77 insertions, 0 deletions
diff --git a/values_at.rb b/values_at.rb new file mode 100644 index 0000000..c2851b4 --- /dev/null +++ b/values_at.rb @@ -0,0 +1,77 @@ +# +# values_at.rb +# + +module Puppet::Parser::Functions + newfunction(:values_at, :type => :rvalue, :doc => <<-EOS + EOS + ) do |arguments| + + raise(Puppet::ParseError, "values_at(): Wrong number of " + + "arguments given (#{arguments.size} for 2)") if arguments.size < 2 + + array = arguments.shift + + if not array.is_a?(Array) + raise(Puppet::ParseError, 'values_at(): Requires an array ' + + 'to work with') + end + + indices = *arguments # Get them all ... Pokemon ... + + if not indices or indices.empty? + raise(Puppet::ParseError, 'values_at(): You must provide ' + + 'at least one positive index to collect') + end + + result = [] + indices_list = [] + + indices.each do |i| + if m = i.match(/^(\d+)(\.\.\.?|\-)(\d+)$/) + start = m[1].to_i + stop = m[3].to_i + + type = m[2] + + if start > stop + raise(Puppet::ParseError, 'values_at(): Stop index in ' + + 'given indices range is smaller than the start index') + elsif stop > array.size - 1 # First element is at index 0 is it not? + raise(Puppet::ParseError, 'values_at(): Stop index in ' + + 'given indices range exceeds array size') + end + + range = case type + when /^(\.\.|\-)$/ then (start .. stop) + when /^(\.\.\.)$/ then (start ... stop) # Exclusive of last element ... + end + + range.each { |i| indices_list << i.to_i } + else + # Only positive numbers allowed in this case ... + if not i.match(/^\d+$/) + raise(Puppet::ParseError, 'values_at(): Unknown format ' + + 'of given index') + end + + # In Puppet numbers are often string-encoded ... + i = i.to_i + + if i > array.size - 1 # Same story. First element is at index 0 ... + raise(Puppet::ParseError, 'values_at(): Given index ' + + 'exceeds array size') + end + + indices_list << i + end + end + + # We remove nil values as they make no sense in Puppet DSL ... + result = indices_list.collect { |i| array[i] }.compact + + return result + end +end + +# vim: set ts=2 sw=2 et : |