summaryrefslogtreecommitdiff
path: root/lib/puppet
diff options
context:
space:
mode:
authorAmos Shapira <ashapira@atlassian.com>2013-03-31 23:37:30 +1100
committerJeff McCune <jeff@puppetlabs.com>2013-04-03 14:05:20 -0700
commit7a2fb80834d3ec4d21efe58e5e8af188e5bdd07c (patch)
tree2f72072e1c474cf21d826d102f7e7b4ea7298017 /lib/puppet
parentf90c54e2df6e8cbd8490a59c7429938072dc1b8d (diff)
downloadpuppet-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.rb35
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 :