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

   1 #
   2 # assert_private.rb
   3 #
   4 module Puppet::Parser::Functions
   5   newfunction(:assert_private, :doc => <<-DOC
   6     @summary
   7       Sets the current class or definition as private.
   8
   9     @return
  10       set the current class or definition as private.
  11
  12     Calling the class or definition from outside the current module will fail.
  13     DOC
  14              ) do |args|
  15
  16     raise(Puppet::ParseError, "assert_private(): Wrong number of arguments given (#{args.size}}) for 0 or 1)") if args.size > 1
  17
  18     scope = self
  19     if scope.lookupvar('module_name') != scope.lookupvar('caller_module_name')
  20       message = nil
  21       if args[0] && args[0].is_a?(String)
  22         message = args[0]
  23       else
  24         manifest_name = scope.source.name
  25         manifest_type = scope.source.type
  26         message = (manifest_type.to_s == 'hostclass') ? 'Class' : 'Definition'
  27         message += " #{manifest_name} is private"
  28       end
  29       raise(Puppet::ParseError, message)
  30     end
  31   end
  32 end