X-Git-Url: https://git.adam-barratt.org.uk/?a=blobdiff_plain;f=3rdparty%2Fmodules%2Fstdlib%2Flib%2Fpuppet%2Fparser%2Ffunctions%2Funion.rb;h=ac521e55c2993ff76dae431d88a7bf7797987017;hb=30caaa85aed7015ca0d77216bff175eebd917eb7;hp=c91bb805373e6695e34676611b155f77cfde3711;hpb=ad88f67c13ae0f1a08936dad643f1e3509ab5f40;p=mirror%2Fdsa-puppet.git

diff --git a/3rdparty/modules/stdlib/lib/puppet/parser/functions/union.rb b/3rdparty/modules/stdlib/lib/puppet/parser/functions/union.rb
index c91bb8053..ac521e55c 100644
--- a/3rdparty/modules/stdlib/lib/puppet/parser/functions/union.rb
+++ b/3rdparty/modules/stdlib/lib/puppet/parser/functions/union.rb
@@ -1,33 +1,28 @@
 #
 # union.rb
 #
-
 module Puppet::Parser::Functions
-  newfunction(:union, :type => :rvalue, :doc => <<-EOS
-This function returns a union of two arrays.
-
-*Examples:*
+  newfunction(:union, :type => :rvalue, :doc => <<-DOC
+    @summary
+      This function returns a union of two or more arrays.
 
-    union(["a","b","c"],["b","c","d"])
+    @return
+      a unionized array of two or more arrays
+    @example **Usage**
 
-Would return: ["a","b","c","d"]
-    EOS
-  ) do |arguments|
+      union(["a","b","c"],["b","c","d"])
+      Would return: ["a","b","c","d"]
+    DOC
+             ) do |arguments|
 
-    # Two arguments are required
-    raise(Puppet::ParseError, "union(): Wrong number of arguments " +
-      "given (#{arguments.size} for 2)") if arguments.size != 2
+    # Check that 2 or more arguments have been given ...
+    raise(Puppet::ParseError, "union(): Wrong number of arguments given (#{arguments.size} for < 2)") if arguments.size < 2
 
-    first = arguments[0]
-    second = arguments[1]
-
-    unless first.is_a?(Array) && second.is_a?(Array)
-      raise(Puppet::ParseError, 'union(): Requires 2 arrays')
+    arguments.each do |argument|
+      raise(Puppet::ParseError, 'union(): Every parameter must be an array') unless argument.is_a?(Array)
     end
 
-    result = first | second
-
-    return result
+    arguments.reduce(:|)
   end
 end