diff options
author | Micah Anderson <micah@riseup.net> | 2015-01-27 15:14:18 -0500 |
---|---|---|
committer | Micah Anderson <micah@riseup.net> | 2015-01-27 15:14:18 -0500 |
commit | 71123634744b9fe2ec7d6a3e38e9789fd84801e3 (patch) | |
tree | 1794e812d83facd93b3007c42632c63ddf1eb2fc /lib/puppet/parser/functions/concat.rb | |
parent | 71cb0f4c2c3bf95f62c9f189f5cef155b09a9682 (diff) | |
parent | 5863ab3901368310186790980aea2b0bf7cecb06 (diff) |
Merge branch 'master' into leap
Diffstat (limited to 'lib/puppet/parser/functions/concat.rb')
-rw-r--r-- | lib/puppet/parser/functions/concat.rb | 22 |
1 files changed, 13 insertions, 9 deletions
diff --git a/lib/puppet/parser/functions/concat.rb b/lib/puppet/parser/functions/concat.rb index c86aa00..618e62d 100644 --- a/lib/puppet/parser/functions/concat.rb +++ b/lib/puppet/parser/functions/concat.rb @@ -4,31 +4,35 @@ module Puppet::Parser::Functions newfunction(:concat, :type => :rvalue, :doc => <<-EOS -Appends the contents of array 2 onto array 1. +Appends the contents of multiple arrays into array 1. *Example:* - concat(['1','2','3'],['4','5','6']) + concat(['1','2','3'],['4','5','6'],['7','8','9']) Would result in: - ['1','2','3','4','5','6'] + ['1','2','3','4','5','6','7','8','9'] EOS ) do |arguments| - # Check that 2 arguments have been given ... + # Check that more than 2 arguments have been given ... raise(Puppet::ParseError, "concat(): Wrong number of arguments " + - "given (#{arguments.size} for 2)") if arguments.size != 2 + "given (#{arguments.size} for < 2)") if arguments.size < 2 a = arguments[0] - b = arguments[1] - # Check that both args are arrays. - unless a.is_a?(Array) and b.is_a?(Array) + # Check that the first parameter is an array + unless a.is_a?(Array) raise(Puppet::ParseError, 'concat(): Requires array to work with') end - result = a.concat(b) + result = a + arguments.shift + + arguments.each do |x| + result = result + Array(x) + end return result end |