puppetlabs-stdlib/lib/puppet/parser/functions/assert_private.rb

#
# assert_private.rb
#

module Puppet::Parser::Functions
  newfunction(:assert_private, :doc => <<-'EOS'
    Sets the current class or definition as private.
    Calling the class or definition from outside the current module will fail.
    EOS
  ) do |args|

    raise(Puppet::ParseError, "assert_private(): Wrong number of arguments "+
      "given (#{args.size}}) for 0 or 1)") if args.size > 1

    scope = self
    if scope.lookupvar('module_name') != scope.lookupvar('caller_module_name')
      message = nil
      if args[0] and args[0].is_a? String
        message = args[0]
      else
        manifest_name = scope.source.name
        manifest_type = scope.source.type
        message = (manifest_type.to_s == 'hostclass') ? 'Class' : 'Definition'
        message += " #{manifest_name} is private"
      end
      raise(Puppet::ParseError, message)
    end
  end
end