57 lines
1.1 KiB
Ruby
57 lines
1.1 KiB
Ruby
|
# vim: set ts=2 sw=2 et :
|
||
|
|
#
|
||
|
|
# load_data loads varibles from external yaml file.
|
||
|
|
#
|
||
|
|
# EXAMPLE 1:
|
||
|
|
# data.yaml:
|
||
|
|
# --
|
||
|
|
# host1.client.com:
|
||
|
|
# abc: def
|
||
|
|
# foo: bar
|
||
|
|
# test: other
|
||
|
|
# host2.client.com:
|
||
|
|
# abc: abc
|
||
|
|
# foo: baz
|
||
|
|
# test: other2
|
||
|
|
#
|
||
|
|
# load_vars("/etc/puppet/data.yaml", $fqdn)
|
||
|
|
# will try to find matching $fqdn key in data.yaml
|
||
|
|
# and, if found, will add variables $abc $foo and $test
|
||
|
|
#
|
||
|
|
#
|
||
|
|
# EXAMPLE 2:
|
||
|
|
# data-host1.clent.com.yaml
|
||
|
|
# abc: def
|
||
|
|
#
|
||
|
|
# load_vars("/etc/puppet/data-$fqdn.yaml")
|
||
|
|
# will add variable $abc
|
||
|
|
|
||
|
|
Puppet::Parser::Functions.newfunction :load_vars, :type => :statement do |args|
|
||
|
|
file = args[0]
|
||
|
|
data = {}
|
||
|
|
if args[1]
|
||
|
|
key = args[1]
|
||
|
|
end
|
||
|
|
|
||
|
|
if FileTest.exist?(file) # file exists
|
||
|
|
|
||
|
|
data = YAML.load_file(file)
|
||
|
|
raise ArgumentError, "Data in %s is not a hash" % file unless data.is_a?(Hash)
|
||
|
|
# data is a hash for sure
|
||
|
|
|
||
|
|
if key
|
||
|
|
# if we have key then use it:
|
||
|
|
if data[key].is_a?(Hash)
|
||
|
|
data = data[key]
|
||
|
|
else
|
||
|
|
data = {}
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|
||
|
|
# add values from hash:
|
||
|
|
data.each do |param, value|
|
||
|
|
setvar(param, strinterp(value))
|
||
|
|
end
|
||
|
|
end
|