5 # TODO(Krzysztof Wilczynski): We probably need to approach numeric values differently ...
7 module Puppet::Parser::Functions
8 newfunction(:range, :type => :rvalue, :doc => <<-EOS
9 When given range in the form of (start, stop) it will extrapolate a range as
16 Will return: [0,1,2,3,4,5,6,7,8,9]
20 Will return: [0,1,2,3,4,5,6,7,8,9] (Zero padded strings are converted to
21 integers automatically)
25 Will return: ["a","b","c"]
27 range("host01", "host10")
29 Will return: ["host01", "host02", ..., "host09", "host10"]
31 Passing a third argument will cause the generated range to step by that
36 Will return: [0,2,4,6,8]
40 # We support more than one argument but at least one is mandatory ...
41 raise(Puppet::ParseError, "range(): Wrong number of " +
42 "arguments given (#{arguments.size} for 1)") if arguments.size < 1
47 step = arguments[2].nil? ? 1 : arguments[2].to_i.abs
49 type = '..' # We select simplest type for Range available in Ruby ...
51 elsif arguments.size > 0
54 if m = value.match(/^(\w+)(\.\.\.?|\-)(\w+)$/)
60 elsif value.match(/^.+$/)
61 raise(Puppet::ParseError, 'range(): Unable to compute range ' +
62 'from the value given')
64 raise(Puppet::ParseError, 'range(): Unknown format of range given')
68 # Check whether we have integer value if so then make it so ...
69 if start.to_s.match(/^\d+$/)
78 when /^(\.\.|\-)$/ then (start .. stop)
79 when /^(\.\.\.)$/ then (start ... stop) # Exclusive of last element ...
82 result = range.step(step).collect { |i| i } # Get them all ... Pokemon ...
88 # vim: set ts=2 sw=2 et :