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