diff options
author | Krzysztof Wilczynski <krzysztof.wilczynski@linux.com> | 2011-04-26 23:45:50 +0100 |
---|---|---|
committer | Krzysztof Wilczynski <krzysztof.wilczynski@linux.com> | 2011-04-26 23:45:50 +0100 |
commit | cd4904b9d92d2cec87ef9332a64559fa16298535 (patch) | |
tree | cc7e4ceaace35334846dfc3615539fedc0d7d987 /range.rb | |
parent | 0e3c0385ef1553e6e613724a8b7f2a157481b172 (diff) | |
download | puppet-stdlib-cd4904b9d92d2cec87ef9332a64559fa16298535.tar.gz puppet-stdlib-cd4904b9d92d2cec87ef9332a64559fa16298535.tar.bz2 |
First version. Simple range function to use within Puppet DSL.
Signed-off-by: Krzysztof Wilczynski <krzysztof.wilczynski@linux.com>
Diffstat (limited to 'range.rb')
-rw-r--r-- | range.rb | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/range.rb b/range.rb new file mode 100644 index 0000000..0c513ef --- /dev/null +++ b/range.rb @@ -0,0 +1,56 @@ +# +# range.rb +# + +module Puppet::Parser::Functions + newfunction(:range, :type => :rvalue, :doc => <<-EOS + EOS + ) do |arguments| + + raise(Puppet::ParseError, "range(): Wrong number of " + + "arguments given (#{arguments.size} for 1)") if arguments.size < 1 + + if arguments.size > 1 + start = arguments[0] + stop = arguments[1] + + type = '..' # We select simplest type for Range available in Ruby ... + + elsif arguments.size > 0 + value = arguments[0] + + if m = value.match(/^(\w+)(\.\.\.?|\-)(\w+)$/) + start = m[1] + stop = m[3] + + type = m[2] + + elsif value.match(/^.+$/) + raise(Puppet::ParseError, 'range(): Unable to compute range ' + + 'from the value given') + else + raise(Puppet::ParseError, 'range(): Unknown format of range given') + end + end + + # Check whether we have numeric value if so then make it so ... + if start.match(/^\d+$/) + start = start.to_i + stop = stop.to_i + else + start = start.to_s + stop = stop.to_s + end + + range = case type + when /^(\.\.|\-)$/ then (start .. stop) + when /^(\.\.\.)$/ then (start ... stop) # Exclusive of last element ... + end + + result = range.collect { |i| i } # Get them all ... Pokemon ... + + return result + end +end + +# vim: set ts=2 sw=2 et : |