blob: ddfc0ef77b04a8b3c7602a7786ca3ec417069591 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
#
# load_vars.rb
#
# This script 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_vars in Puppet manifest file as follows:
#
# load_vars("/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_vars like this:
#
# load_vars("/etc/puppet/data-$fqdn.yaml")
#
# This will result in a variable $foo being added and ready for use.
#
module Puppet::Parser::Functions
newfunction(:load_vars, :type => :statement) do |arguments|
raise(Puppet::ParseError, "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, "Unable to load data " +
"from the file `%s': %s" % file, error.to_s)
end
raise(Puppet::ParseError, "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 { |param, value| setvar(param, strinterp(value)) }
end
end
# vim: set ts=2 sw=2 et :
|
