diff options
author | Ken Barber <ken@bob.sh> | 2011-04-29 09:27:10 +0200 |
---|---|---|
committer | Ken Barber <ken@bob.sh> | 2011-04-30 15:59:31 +0200 |
commit | 99a93d366f2e1efb977fcc8fe300d3d8357c8214 (patch) | |
tree | fab7510323f123c83da6d7c6cae4fe9e3f2f1e27 /lib/puppet/parser/functions/range.rb | |
parent | 781a8720577d0cafe2147f3a1b938db7bec31789 (diff) |
Convert to module format.
Diffstat (limited to 'lib/puppet/parser/functions/range.rb')
-rw-r--r-- | lib/puppet/parser/functions/range.rb | 59 |
1 files changed, 59 insertions, 0 deletions
diff --git a/lib/puppet/parser/functions/range.rb b/lib/puppet/parser/functions/range.rb new file mode 100644 index 0000000..6afb50c --- /dev/null +++ b/lib/puppet/parser/functions/range.rb @@ -0,0 +1,59 @@ +# +# range.rb +# + +# TODO(Krzysztof Wilczynski): We probably need to approach numeric values differently ... + +module Puppet::Parser::Functions + newfunction(:range, :type => :rvalue, :doc => <<-EOS + EOS + ) do |arguments| + + # We support more than one argument but at least one is mandatory ... + 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 integer 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 : |