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 => <<-'DOC'
   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 DOC
  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.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   resource = findresource(type, title)
  46   if resource
  47     matches = params.map do |key, value|
  48       # eql? avoids bugs caused by monkeypatching in puppet
  49       resource_is_undef = resource[key].eql?(:undef) || resource[key].nil?
  50       value_is_undef = value.eql?(:undef) || value.nil?
  51       (resource_is_undef && value_is_undef) || (resource[key] == value)
  52     end
  53     ret = params.empty? || !matches.include?(false)
  54   end
  55   Puppet.debug("Resource #{reference} was not determined to be defined")
  56   ret
  57 end