diff options
author | Amos Shapira <ashapira@atlassian.com> | 2013-03-31 23:37:30 +1100 |
---|---|---|
committer | Jeff McCune <jeff@puppetlabs.com> | 2013-04-03 14:05:20 -0700 |
commit | 7a2fb80834d3ec4d21efe58e5e8af188e5bdd07c (patch) | |
tree | 2f72072e1c474cf21d826d102f7e7b4ea7298017 /lib/puppet | |
parent | f90c54e2df6e8cbd8490a59c7429938072dc1b8d (diff) | |
download | puppet-stdlib-7a2fb80834d3ec4d21efe58e5e8af188e5bdd07c.tar.gz puppet-stdlib-7a2fb80834d3ec4d21efe58e5e8af188e5bdd07c.tar.bz2 |
(#19998) Implement any2array
This change is to implement a new function "any2array", which will take any
argument or arguments and create an array which contains it. If the argument
is a single array then it will be returned as-is. If the argument is a single
hash then it will be converted into an array. Otherwise (if there are more than
one argument, or the only argument is not an array or a hash) the function will
return an array containing all the arguments.
Diffstat (limited to 'lib/puppet')
-rw-r--r-- | lib/puppet/parser/functions/any2array.rb | 35 |
1 files changed, 35 insertions, 0 deletions
diff --git a/lib/puppet/parser/functions/any2array.rb b/lib/puppet/parser/functions/any2array.rb new file mode 100644 index 0000000..d150743 --- /dev/null +++ b/lib/puppet/parser/functions/any2array.rb @@ -0,0 +1,35 @@ +# +# str2bool.rb +# + +module Puppet::Parser::Functions + newfunction(:any2array, :type => :rvalue, :doc => <<-EOS +This converts any object to an array containing that object. Empty argument +lists are converted to an empty array. Arrays are left untouched. Hashes are +converted to arrays of alternating keys and values. + EOS + ) do |arguments| + + if arguments.empty? + return [] + end + + if arguments.length == 1 + if arguments[0].kind_of?(Array) + return arguments[0] + elsif arguments[0].kind_of?(Hash) + result = [] + arguments[0].each do |key, value| + result << key << value + end + return result + else + return [arguments[0]] + end + end + + return arguments + end +end + +# vim: set ts=2 sw=2 et : |