3rdparty/modules/stdlib/lib/puppet/parser/functions/delete_undef_values.rb

   1 module Puppet::Parser::Functions
   2   newfunction(:delete_undef_values, :type => :rvalue, :doc => <<-EOS
   3 Returns a copy of input hash or array with all undefs deleted.
   4
   5 *Examples:*
   6
   7     $hash = delete_undef_values({a=>'A', b=>'', c=>undef, d => false})
   8
   9 Would return: {a => 'A', b => '', d => false}
  10
  11     $array = delete_undef_values(['A','',undef,false])
  12
  13 Would return: ['A','',false]
  14
  15       EOS
  16     ) do |args|
  17
  18     raise(Puppet::ParseError, "delete_undef_values(): Wrong number of arguments given (#{args.size})") if args.size < 1
  19
  20     unless args[0].is_a? Array or args[0].is_a? Hash
  21       raise(Puppet::ParseError, "delete_undef_values(): expected an array or hash, got #{args[0]} type  #{args[0].class} ")
  22     end
  23     result = args[0].dup
  24     if result.is_a?(Hash)
  25       result.delete_if {|key, val| val.equal? :undef}
  26     elsif result.is_a?(Array)
  27       result.delete :undef
  28     end
  29     result
  30   end
  31 end