77 lines
1.9 KiB
Ruby
77 lines
1.9 KiB
Ruby
#
|
|
# load_variables.rb
|
|
#
|
|
|
|
module Puppet::Parser::Functions
|
|
newfunction(:load_variables, :type => :statement, :doc => <<-EOS
|
|
This function will allow for loading variables from an external YAML
|
|
file and expose them for further use inside the Puppet manifest file ...
|
|
|
|
For example:
|
|
|
|
Given following content of the data.yaml file:
|
|
|
|
---
|
|
host1.example.com:
|
|
foo: bar
|
|
baz: quux
|
|
question: 42
|
|
host2.example.com:
|
|
abc: def
|
|
this: that
|
|
darth: vader
|
|
|
|
Then calling load_variables in Puppet manifest file as follows:
|
|
|
|
load_variables("/etc/puppet/data.yaml", $fqdn)
|
|
|
|
Will result in addition of variables $foo, $baz and $question
|
|
for matching host name as per the variable $fqdn ...
|
|
|
|
Another example which uses per-host file:
|
|
|
|
Given following content of the file data-host1.example.com.yaml:
|
|
|
|
---
|
|
foo: bar
|
|
|
|
Then when we call load_variables like this:
|
|
|
|
load_variables("/etc/puppet/data-${fqdn}.yaml")
|
|
|
|
This will result in a variable $foo being added and ready for use.
|
|
EOS
|
|
) do |arguments|
|
|
|
|
raise(Puppet::ParseError, "load_variables(): Wrong number of " +
|
|
"arguments given (#{arguments.size} for 2)") if arguments.size < 2
|
|
|
|
data = {}
|
|
|
|
file = arguments[0]
|
|
key = arguments[1] if arguments[1]
|
|
|
|
if File.exists?(file)
|
|
|
|
begin
|
|
data = YAML.load_file(file)
|
|
rescue => error
|
|
raise(Puppet::ParseError, "load_variables(): Unable to load data " +
|
|
"from the file `%s': %s" % file, error.to_s)
|
|
end
|
|
|
|
raise(Puppet::ParseError, "load_variables(): Data in the file `%s' " +
|
|
"is not a hash" % file) unless data.is_a?(Hash)
|
|
|
|
data = ((data[key] and data[key].is_a?(Hash)) ? data[key] : {}) if key
|
|
end
|
|
|
|
data.each do |param, value|
|
|
value = strinterp(value) # Evaluate any interpolated variable names ...
|
|
|
|
setvar(param, value)
|
|
end
|
|
end
|
|
end
|
|
|
|
# vim: set ts=2 sw=2 et :
|