diff options
Diffstat (limited to 'code/environments/production/modules/stdlib/lib/puppet/parser/functions/suffix.rb')
-rw-r--r-- | code/environments/production/modules/stdlib/lib/puppet/parser/functions/suffix.rb | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/code/environments/production/modules/stdlib/lib/puppet/parser/functions/suffix.rb b/code/environments/production/modules/stdlib/lib/puppet/parser/functions/suffix.rb new file mode 100644 index 0000000..407cd53 --- /dev/null +++ b/code/environments/production/modules/stdlib/lib/puppet/parser/functions/suffix.rb @@ -0,0 +1,51 @@ +# +# suffix.rb +# +module Puppet::Parser::Functions + newfunction(:suffix, :type => :rvalue, :doc => <<-DOC + This function applies a suffix to all elements in an array, or to the keys + in a hash. + + *Examples:* + + suffix(['a','b','c'], 'p') + + Will return: ['ap','bp','cp'] + DOC + ) 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.empty? + + enumerable = arguments[0] + + unless enumerable.is_a?(Array) || enumerable.is_a?(Hash) + raise Puppet::ParseError, "suffix(): expected first argument to be an Array or a Hash, got #{enumerable.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 + + result = if enumerable.is_a?(Array) + # Turn everything into string same as join would do ... + enumerable.map do |i| + i = i.to_s + suffix ? i + suffix : i + end + else + Hash[enumerable.map do |k, v| + k = k.to_s + [suffix ? k + suffix : k, v] + end] + end + + return result + end +end + +# vim: set ts=2 sw=2 et : |