| Current Path : /usr/share/ruby/vendor_ruby/ |
Linux gator3171.hostgator.com 4.19.286-203.ELK.el7.x86_64 #1 SMP Wed Jun 14 04:33:55 CDT 2023 x86_64 |
| Current File : //usr/share/ruby/vendor_ruby/hiera_puppet.rb |
require 'hiera'
require 'hiera/scope'
require 'puppet'
module HieraPuppet
module_function
def lookup(key, default, scope, override, resolution_type)
scope = Hiera::Scope.new(scope)
answer = hiera.lookup(key, default, scope, override, resolution_type)
if answer.nil?
raise(Puppet::ParseError, "Could not find data item #{key} in any Hiera data file and no default supplied")
end
answer
end
def parse_args(args)
# Functions called from Puppet manifests like this:
#
# hiera("foo", "bar")
#
# Are invoked internally after combining the positional arguments into a
# single array:
#
# func = function_hiera
# func(["foo", "bar"])
#
# Functions called from templates preserve the positional arguments:
#
# scope.function_hiera("foo", "bar")
#
# Deal with Puppet's special calling mechanism here.
if args[0].is_a?(Array)
args = args[0]
end
if args.empty?
raise(Puppet::ParseError, "Please supply a parameter to perform a Hiera lookup")
end
key = args[0]
default = args[1]
override = args[2]
return [key, default, override]
end
private
module_function
def hiera
@hiera ||= Hiera.new(:config => hiera_config)
end
def hiera_config
config = {}
if config_file = hiera_config_file
config = Hiera::Config.load(config_file)
end
config[:logger] = 'puppet'
config
end
def hiera_config_file
config_file = nil
if Puppet.settings[:hiera_config].is_a?(String)
expanded_config_file = File.expand_path(Puppet.settings[:hiera_config])
if Puppet::FileSystem.exist?(expanded_config_file)
config_file = expanded_config_file
end
elsif Puppet.settings[:confdir].is_a?(String)
expanded_config_file = File.expand_path(File.join(Puppet.settings[:confdir], '/hiera.yaml'))
if Puppet::FileSystem.exist?(expanded_config_file)
config_file = expanded_config_file
end
end
config_file
end
end