29402f31e7
When prefix and suffix did error checking with positional arguments, they would not report the position of the argument that failed to validate. This commit changes the messages to indicate which argument failed.
45 lines
1 KiB
Ruby
45 lines
1 KiB
Ruby
#
|
|
# suffix.rb
|
|
#
|
|
|
|
module Puppet::Parser::Functions
|
|
newfunction(:suffix, :type => :rvalue, :doc => <<-EOS
|
|
This function applies a suffix to all elements in an array.
|
|
|
|
*Examples:*
|
|
|
|
suffix(['a','b','c'], 'p')
|
|
|
|
Will return: ['ap','bp','cp']
|
|
EOS
|
|
) do |arguments|
|
|
|
|
# Technically we support two arguments but only first is mandatory ...
|
|
raise(Puppet::ParseError, "suffix(): Wrong number of arguments " +
|
|
"given (#{arguments.size} for 1)") if arguments.size < 1
|
|
|
|
array = arguments[0]
|
|
|
|
unless array.is_a?(Array)
|
|
raise Puppet::ParseError, "suffix(): expected first argument to be an Array, got #{array.inspect}"
|
|
end
|
|
|
|
suffix = arguments[1] if arguments[1]
|
|
|
|
if suffix
|
|
unless suffix.is_a? String
|
|
raise Puppet::ParseError, "suffix(): expected second argument to be a String, got #{suffix.inspect}"
|
|
end
|
|
end
|
|
|
|
# Turn everything into string same as join would do ...
|
|
result = array.collect do |i|
|
|
i = i.to_s
|
|
suffix ? i + suffix : i
|
|
end
|
|
|
|
return result
|
|
end
|
|
end
|
|
|
|
# vim: set ts=2 sw=2 et :
|