diff options
author | Paweł Tomulik <ptomulik@meil.pw.edu.pl> | 2013-07-29 11:30:47 +0200 |
---|---|---|
committer | Paweł Tomulik <ptomulik@meil.pw.edu.pl> | 2013-08-05 22:12:26 +0200 |
commit | 206941520467a5cbf1ba4131c68c4814b5ab181a (patch) | |
tree | 038b83732eea904567b9188d055277475efdfd45 /lib/puppet/parser/functions/delete_undef_values.rb | |
parent | e8dcc691934f11172e2a3b5064a375f6b025865b (diff) |
added delete_values() and delete_undef_values() functions
Diffstat (limited to 'lib/puppet/parser/functions/delete_undef_values.rb')
-rw-r--r-- | lib/puppet/parser/functions/delete_undef_values.rb | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/lib/puppet/parser/functions/delete_undef_values.rb b/lib/puppet/parser/functions/delete_undef_values.rb new file mode 100644 index 0000000..4eecab2 --- /dev/null +++ b/lib/puppet/parser/functions/delete_undef_values.rb @@ -0,0 +1,34 @@ +module Puppet::Parser::Functions + newfunction(:delete_undef_values, :type => :rvalue, :doc => <<-EOS +Returns a copy of input hash or array with all undefs deleted. + +*Examples:* + + $hash = delete_undef_values({a=>'A', b=>'', c=>undef, d => false}) + +Would return: {a => 'A', b => '', d => false} + + $array = delete_undef_values(['A','',undef,false]) + +Would return: ['A','',false] + + EOS + ) do |args| + + raise(Puppet::ParseError, + "delete_undef_values(): Wrong number of arguments given " + + "(#{args.size})") if args.size < 1 + + result = args[0] + if result.is_a?(Hash) + result.delete_if {|key, val| val.equal? :undef} + elsif result.is_a?(Array) + result.delete :undef + else + raise(Puppet::ParseError, + "delete_undef_values(): Wrong argument type #{args[0].class} " + + "for first argument") + end + result + end +end |