diff options
Diffstat (limited to 'code/environments/production/modules/stdlib/lib/puppet/parser/functions/flatten.rb')
-rw-r--r-- | code/environments/production/modules/stdlib/lib/puppet/parser/functions/flatten.rb | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/code/environments/production/modules/stdlib/lib/puppet/parser/functions/flatten.rb b/code/environments/production/modules/stdlib/lib/puppet/parser/functions/flatten.rb new file mode 100644 index 0000000..15970df --- /dev/null +++ b/code/environments/production/modules/stdlib/lib/puppet/parser/functions/flatten.rb @@ -0,0 +1,31 @@ +# +# flatten.rb +# +module Puppet::Parser::Functions + newfunction(:flatten, :type => :rvalue, :doc => <<-DOC + This function flattens any deeply nested arrays and returns a single flat array + as a result. + + *Examples:* + + flatten(['a', ['b', ['c']]]) + + Would return: ['a','b','c'] + DOC + ) do |arguments| + + raise(Puppet::ParseError, "flatten(): Wrong number of arguments given (#{arguments.size} for 1)") if arguments.size != 1 + + array = arguments[0] + + unless array.is_a?(Array) + raise(Puppet::ParseError, 'flatten(): Requires array to work with') + end + + result = array.flatten + + return result + end +end + +# vim: set ts=2 sw=2 et : |