blob: 4dd29159e7ff6a8f8cfa1f220ce19c664251fd4d (
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
|
require 'puppet/indirector/terminus'
# The base class for YAML indirection termini.
class Puppet::Indirector::Yaml < Puppet::Indirector::Terminus
# Read a given name's file in and convert it from YAML.
def find(name)
raise ArgumentError.new("You must specify the name of the object to retrieve") unless name
file = path(name)
return nil unless FileTest.exist?(file)
begin
return from_yaml(File.read(file))
rescue => detail
raise Puppet::Error, "Could not read YAML data for %s %s: %s" % [indirection.name, name, detail]
end
end
# Convert our object to YAML and store it to the disk.
def save(object)
raise ArgumentError.new("You can only save objects that respond to :name") unless object.respond_to?(:name)
file = path(object.name)
basedir = File.dirname(file)
# This is quite likely a bad idea, since we're not managing ownership or modes.
unless FileTest.exist?(basedir)
Dir.mkdir(basedir)
end
begin
File.open(file, "w", 0660) { |f| f.print to_yaml(object) }
rescue TypeError => detail
Puppet.err "Could not save %s %s: %s" % [self.name, object.name, detail]
end
end
def version(name)
return nil unless FileTest.exist?(path(name))
return File.stat(path(name)).mtime
end
private
def from_yaml(text)
YAML.load(text)
end
def to_yaml(object)
YAML.dump(object)
end
# Return the path to a given node's file.
def path(name)
File.join(Puppet[:yamldir], self.class.indirection_name.to_s, name.to_s + ".yaml")
end
end
|
