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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
require 'puppet'
require 'puppet/rails/param_name'
require 'puppet/rails/puppet_tag'
require 'puppet/util/rails/collection_merger'
class Puppet::Rails::Resource < ActiveRecord::Base
include Puppet::Util::CollectionMerger
has_many :param_values, :dependent => :destroy
has_many :param_names, :through => :param_values
has_many :resource_tags, :dependent => :destroy
has_many :puppet_tags, :through => :resource_tags
belongs_to :source_file
belongs_to :host
def add_resource_tag(tag)
pt = Puppet::Rails::PuppetTag.find_or_create_by_name(tag, :include => :puppet_tag)
resource_tags.create(:puppet_tag => pt)
end
def file
if f = self.source_file
return f.filename
else
return nil
end
end
def file=(file)
self.source_file = Puppet::Rails::SourceFile.find_or_create_by_filename(file)
end
# returns a hash of param_names.name => [param_values]
def get_params_hash(values = nil)
values ||= param_values.find(:all, :include => :param_name)
values.inject({}) do | hash, value |
hash[value.param_name.name] ||= []
hash[value.param_name.name] << value
hash
end
end
def get_tag_hash(tags = nil)
tags ||= resource_tags.find(:all, :include => :puppet_tag)
return tags.inject({}) do |hash, tag|
# We have to store the tag object, not just the tag name.
hash[tag.puppet_tag.name] = tag
hash
end
end
def [](param)
return super || parameter(param)
end
def name
ref()
end
def parameter(param)
if pn = param_names.find_by_name(param)
if pv = param_values.find(:first, :conditions => [ 'param_name_id = ?', pn] )
return pv.value
else
return nil
end
end
end
def parameters
result = get_params_hash
result.each do |param, value|
if value.is_a?(Array)
result[param] = value.collect { |v| v.value }
else
result[param] = value.value
end
end
result
end
def ref
"%s[%s]" % [self[:restype].capitalize, self[:title]]
end
# Convert our object to a resource. Do not retain whether the object
# is exported, though, since that would cause it to get stripped
# from the configuration.
def to_resource(scope)
hash = self.attributes
hash["type"] = hash["restype"]
hash.delete("restype")
# FIXME At some point, we're going to want to retain this information
# for logging and auditing.
hash.delete("host_id")
hash.delete("updated_at")
hash.delete("source_file_id")
hash.delete("id")
hash.each do |p, v|
hash.delete(p) if v.nil?
end
hash[:scope] = scope
hash[:source] = scope.source
obj = Puppet::Parser::Resource.new(hash)
names = []
self.param_names.each do |pname|
# We can get the same name multiple times because of how the
# db layout works.
next if names.include?(pname.name)
names << pname.name
obj.set(pname.to_resourceparam(self, scope.source))
end
# Store the ID, so we can check if we're re-collecting the same resource.
obj.rails_id = self.id
return obj
end
end
# $Id$
|
