diff --git a/lib/puppet/rails/host.rb b/lib/puppet/rails/host.rb index ab435caa4..dd6ffce0e 100644 --- a/lib/puppet/rails/host.rb +++ b/lib/puppet/rails/host.rb @@ -1,107 +1,105 @@ require 'puppet/rails/resource' require 'pp' class Puppet::Rails::Host < ActiveRecord::Base has_many :fact_values, :through => :fact_names has_many :fact_names belongs_to :puppet_classes has_many :source_files has_many :resources, :include => [ :param_names, :param_values ] acts_as_taggable def facts(name) fv = self.fact_values.find(:first, :conditions => "fact_names.name = '#{name}'") return fv.value end # If the host already exists, get rid of its objects def self.clean(host) if obj = self.find_by_name(host) obj.rails_objects.clear return obj else return nil end end # Store our host in the database. def self.store(hash) unless hash[:name] raise ArgumentError, "You must specify the hostname for storage" end args = {} if hash[:facts].include?("ipaddress") args[:ip] = hash[:facts]["ipaddress"] end host = self.find_or_create_by_name(hash[:facts]["hostname"], args) hash[:facts].each do |name, value| fn = host.fact_names.find_or_create_by_name(name) fv = fn.fact_values.find_or_create_by_value(value) host.fact_names << fn end unless hash[:resources] raise ArgumentError, "You must pass resources" end typenames = [] Puppet::Type.loadall Puppet::Type.eachtype do |type| typenames << type.name.to_s end hash[:resources].each do |resource| resargs = resource.to_hash.stringify_keys if typenames.include?(resource.type) rtype = "Puppet#{resource.type.to_s.capitalize}" end res = host.resources.find_or_create_by_title(resource[:title]) res.type = rtype -f = File.new("/tmp/rtype.dump", "w+") -f.puts rtype res.save resargs.each do |param, value| pn = res.param_names.find_or_create_by_name(param) pv = pn.param_values.find_or_create_by_value(value) res.param_names << pn end end Puppet::Util.benchmark(:info, "Saved host to database") do host.save end return host end # Add all of our RailsObjects def addobjects(objects) objects.each do |tobj| params = {} tobj.each do |p,v| params[p] = v end args = {:ptype => tobj.type, :name => tobj.name} [:tags, :file, :line].each do |param| if val = tobj.send(param) args[param] = val end end robj = rails_objects.build(args) robj.addparams(params) if tobj.collectable robj.toggle(:collectable) end end end end # $Id$