59 lines
1.5 KiB
Ruby
59 lines
1.5 KiB
Ruby
#
|
|
# 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 :
|