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

   1 # Test whether a given class or definition is defined
   2 require 'puppet/parser/functions'
   3
   4 Puppet::Parser::Functions.newfunction(:defined_with_params,
   5                                       :type => :rvalue,
   6                                       :doc => <<-'ENDOFDOC'
   7 Takes a resource reference and an optional hash of attributes.
   8
   9 Returns true if a resource with the specified attributes has already been added
  10 to the catalog, and false otherwise.
  11
  12     user { 'dan':
  13       ensure => present,
  14     }
  15
  16     if ! defined_with_params(User[dan], {'ensure' => 'present' }) {
  17       user { 'dan': ensure => present, }
  18     }
  19 ENDOFDOC
  20 ) do |vals|
  21   reference, params = vals
  22   raise(ArgumentError, 'Must specify a reference') unless reference
  23   if (! params) || params == ''
  24     params = {}
  25   end
  26   ret = false
  27
  28   if Puppet::Util::Package.versioncmp(Puppet.version, '4.6.0') >= 0
  29     # Workaround for PE-20308
  30     if reference.is_a?(String)
  31       type_name, title = Puppet::Resource.type_and_title(reference, nil)
  32       type = Puppet::Pops::Evaluator::Runtime3ResourceSupport.find_resource_type_or_class(find_global_scope, type_name.downcase)
  33     elsif reference.is_a?(Puppet::Resource)
  34       type = reference.resource_type
  35       title = reference.title
  36     else
  37       raise(ArgumentError, "Reference is not understood: '#{reference.class}'")
  38     end
  39     #end workaround
  40   else
  41     type = reference.to_s
  42     title = nil
  43   end
  44
  45   if resource = findresource(type, title)
  46     matches = params.collect do |key, value|
  47       # eql? avoids bugs caused by monkeypatching in puppet
  48       resource_is_undef = resource[key].eql?(:undef) || resource[key].nil?
  49       value_is_undef = value.eql?(:undef) || value.nil?
  50       (resource_is_undef && value_is_undef) || (resource[key] == value)
  51     end
  52     ret = params.empty? || !matches.include?(false)
  53   end
  54   Puppet.debug("Resource #{reference} was not determined to be defined")
  55   ret
  56 end