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