4f1a5e26c481c6737f3b009b9c4e77f91f81d496
[mirror/dsa-puppet.git] / 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