30 lines
877 B
Ruby
30 lines
877 B
Ruby
#
|
|
# 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
|