class Host < Puppet::Rails::Host

has_many :puppetclasses, :through => :host_classes

belongs_to :subnet

:params, :hostgroup, :domain, :ip, :mac

named_scope :out_of_sync, lambda { |*args| {:conditions => ["last_report < ? and enabled != ?", (args.first || (Setting[:puppet_interval] + 5).minutes.ago), false]} }

named_scope :with_fact, lambda { |fact,value|

unless fact.nil? or value.nil?

INNER JOIN fact_names fn_#{fact} ON fn_#{fact}.id = fv_#{fact}.fact_name_id",

:select => "DISTINCT hosts.name, hosts.id", :conditions =>

raise "invalid fact"

end

}

named_scope :with_class, lambda { |klass|

unless klass.nil?

{ :joins => :puppetclasses, :select => "hosts.name", :conditions => {:puppetclasses => {:name => klass }} }

else

}

"(puppet_status >> #{Report::BIT_NUM*Report::METRIC.index(arg[0])} & #{Report::MAX}) > #{arg[1] || 0}"}

}

named_scope :with_error, { :conditions => "(puppet_status > 0) and

named_scope :with_changes, { :conditions => "(puppet_status > 0) and

((puppet_status >> #{Report::BIT_NUM*Report::METRIC.index("restarted")} & #{Report::MAX}) !=0)"

named_scope :successful, {:conditions => "puppet_status = 0"}

user = User.current

owner_conditions = sanitize_sql_for_conditions(["((hosts.owner_id in (?) AND hosts.owner_type = 'Usergroup') OR (hosts.owner_id = ? AND hosts.owner_type = 'User'))", user.my_usergroups.map(&:id), user.id])

domain_conditions = sanitize_sql_for_conditions([" (hosts.domain_id in (?))",dms = (user.domains).map(&:id)])

hostgroup_conditions = sanitize_sql_for_conditions([" (hosts.hostgroup_id in (?))",(hgs = user.hostgroups).map(&:id)])

fact_conditions = ""

for user_fact in (ufs = user.user_facts)

fact_conditions += sanitize_sql_for_conditions ["(hosts.id = fact_values.host_id and fact_values.fact_name_id = ? and fact_values.value #{user_fact.operator} ?)", user_fact.fact_name_id, user_fact.criteria]

fact_conditions = user_fact.andor == "and" ? "(#{fact_conditions}) and " : "#{fact_conditions} or "

end

if match = fact_conditions.match(/^(.*).....$/)

fact_conditions = "(#{match[1]})"

end

conditions = ""

if user.filtering?

conditions = "#{owner_conditions}" if user.filter_on_owner

(conditions = (user.domains_andor == "and") ? "(#{conditions}) and #{domain_conditions} " : "#{conditions} or #{domain_conditions} ") unless dms.empty?

(conditions = (user.hostgroups_andor == "and") ? "(#{conditions}) and #{hostgroup_conditions} " : "#{conditions} or #{hostgroup_conditions} ") unless hgs.empty?

(conditions = (user.facts_andor == "and") ? "(#{conditions}) and #{fact_conditions} " : "#{conditions} or #{fact_conditions} ") unless ufs.empty?

conditions.sub!(/\s*\(\)\s*/, "")

conditions.sub!(/^(?:\(\))?\s?(?:and|or)\s*/, "")

conditions.sub!(/\(\s*(?:or|and)\s*\(/, "((")

end

{:conditions => conditions}

}

acts_as_audited :except => [:last_report, :puppet_status, :last_compile]

alias_attribute :os, :operatingsystem

alias_attribute :arch, :architecture

include Orchestration

validates_uniqueness_of :mac, :unless => Proc.new { |host| host.hypervisor? or !host.managed }

validates_uniqueness_of :sp_name, :sp_ip, :allow_blank => true, :allow_nil => true

validates_format_of :sp_name, :with => /.*-sp/, :allow_nil => true, :allow_blank => true

validates_format_of :ip, :with => (/(\d{1,3}\.){3}\d{1,3}/), :if => Proc.new {|host| host.managed}

validates_format_of :sp_ip, :with => /(\d{1,3}\.){3}\d{1,3}/, :allow_nil => true, :allow_blank => true

validates_format_of :serial, :with => /[01],\d{3,}n\d/, :message => "should follow this format: 0,9600n8", :allow_blank => true, :allow_nil => true

end

self.owner ||= User.current

end

name

end

self.name <=> other.name

end

def is_owned_by

owner.id_and_type if owner

end

# virtual attributes which sets the owner based on the user selection

# supports a simple user, or a usergroup

# selection parameter is expected to be an ActiveRecord id_and_type method (see Foreman's AR extentions).

def is_owned_by=(selection)

oid = Usergroup.find(selection.to_i) if selection =~ (/-Usergroups$/)

end

def clearFacts

# Called from the host build post install process to indicate that the base build has completed

# Build is cleared and the boot link and autosign entries are removed

# A site specific build script is called at this stage that can do site specific tasks

unless Rails.env == "test"

# Disallow any auto signing for our host.

GW::Puppetca.disable name unless puppetca?

GW::Tftp.remove mac unless respond_to?(:tftp?) and tftp?

end

logger.warn "Failed to set Build on #{self}: #{e}"

false

def pm_fqdn

# Cleans Certificate and enable Autosign

# Called after a host is given their provisioning template

# Returns : Boolean status of the operation

def handle_ca

return true if Rails.env == "test"

if puppetca?

# Legacy CA handling

GW::Puppetca.clean(name) && GW::Puppetca.sign(name)

end

def diskLayout

opts[:kind] ||= "provision"

opts[:operatingsystem_id] ||= operatingsystem_id

opts[:hostgroup_id] ||= hostgroup_id

opts[:environment_id] ||= environment_id

ConfigTemplate.find_template opts

def error_count

raise "invalid type #{type}" if type and not Report::METRIC.include?(type)

h = {}

(type || Report::METRIC).each do |m|

h[m] = (read_attribute(:puppet_status) || 0) >> (Report::BIT_NUM*Report::METRIC.index(m)) & Report::MAX

end

return type.nil? ? h : h[type]

# e.g. Report.last.applied

Report::METRIC.each do |method|

define_method method do

status method

end

def no_report

not enabled?

end

# TODO: remove hard coded default parameters into some selectable values in the database.

param["domainname"] = domain.fullname unless domain.nil? or domain.fullname.nil?

param["mac"] = mac

end

klasses = puppetclasses.map(&:id)

klasses += hostgroup.classes.map(&:id) if hostgroup

LookupKey.all(:conditions => {:puppetclass_id =>klasses.flatten } ).each do |k|

param[k.to_s] = k.value_for(self)

end unless klasses.empty?

end

info_hash['classes'] = self.puppetclasses_names

info_hash['parameters'] = param

info_hash['environment'] = environment.to_s unless environment.nil? or environment.name.nil?

return info_hash

parameters = {}

operatingsystem.os_parameters.each {|p| parameters.update Hash[p.name => p.value] } unless operatingsystem.nil?

host_parameters.each {|p| parameters.update Hash[p.name => p.value] }

end

h.save(false) if h.new_record?

h.importFacts(facts)

# import host facts, required when running without storeconfigs.

# expect a Puppet::Node::Facts

def importFacts facts

raise "invalid Fact" unless facts.is_a?(Puppet::Node::Facts)

# we are not importing facts for hosts in build state (e.g. waiting for a re-installation)

raise "Host is pending for Build" if build

time = time.to_time if time.is_a?(String)

# we are not doing anything we already processed this fact (or a newer one)

return true unless last_compile.nil? or (last_compile + 1.minute < time)

self.last_compile = time

# save all other facts - pre 0.25 it was called setfacts

save(false)

# we want to import other information only if this host was never installed via Foreman

populateFieldsFromFacts if installed_at.nil?

# we are saving here with no validations, as we want this process to be as fast

# as possible, assuming we already have all the right settings in Foreman.

# If we don't (e.g. we never install the server via Foreman, we populate the fields from facts

# TODO: if it was installed by Foreman and there is a mismatch,

# we should probably send out an alert.

return self.save(false)

rescue Exception => e

logger.warn "Failed to save #{facts.name}: #{e}"

end

major, minor = orel.split(".")

end

# Build is set

# The boot link and autosign entry are created

# Any existing puppet certificates are deleted

# Any facts are discarded

def setBuild

clearReports

# ensures that the legacy TFTP code is not called when using a smart proxy.

unless respond_to?(:tftp?) and tftp?

return false unless GW::Tftp.create([mac, os.to_s.gsub(" ","-"), arch.name, serial])

end

self.save

# it is assumed that you already have the node (e.g. imported by one of the rack tasks)

def importNode nodeinfo

nodeinfo["classes"].each do |klass|

if pc = Puppetclass.find_by_name(klass)

# parameters are a bit more tricky, as some classifiers provide the facts as parameters as well

# not sure what is puppet priority about it, but we ignore it if has a fact with the same name.

# additionally, we don't import any non strings values, as puppet don't know what to do with those as well.

myparams = self.info["parameters"]

nodeinfo["parameters"].each_pair do |param,value|

next if fact_names.exists? :name => param

next unless value.is_a?(String)

# we already have this parameter

next if myparams.has_key?(param) and myparams[param] == value

unless (hp = self.host_parameters.create(:name => param, :value => value))

logger.warn "Failed to import #{param}/#{value} for #{name}: #{hp.errors.full_messages.join(", ")}"

end

self.save

end

# e.g. how many hosts belongs to each os

# returns sorted hash

def self.count_distribution assocication

output = {}

begin

output[k.to_label] = v unless v == 0

rescue

logger.info "skipped #{k} as it has has no label"

end

end

end

# counts each association of a given host for HABTM relationships

# TODO: Merge these two into one method

# e.g. how many hosts belongs to each os

# returns sorted hash

def self.count_habtm assocication

output = {}

Host.count(:include => assocication.pluralize, :group => "#{assocication}_id").to_a.each do |a|

#Ugly Ugly Ugly - I guess I'm missing something basic here

label = eval(assocication.camelize).send("find",a[0].to_i).to_label if a[0]

output[label] = a[1]

end

output

end

data[:failed] << "[ #{r.reported_at.to_i}000, #{r.failed} ]"

data[:restarted] << "[ #{r.reported_at.to_i}000, #{r.restarted} ]"

data[:failed_restarts] << "[ #{r.reported_at.to_i}000, #{r.failed_restarts} ]"

data[:skipped] << "[ #{r.reported_at.to_i}000, #{r.skipped} ]"

end

return data

end

def runtime_chart(timerange = 1.day.ago)

data = {}

data[:config], data[:runtime] = [], []

reports.recent(timerange).each do |r|

data[:config] << "[ #{r.reported_at.to_i}000, #{r.config_retrieval} ]"

data[:runtime] << "[ #{r.reported_at.to_i}000, #{r.runtime} ]"

return klasses

hash = {}

fact_values.all(:include => :fact_name).collect do |fact|

hash[fact.fact_name.name] = fact.value

hash

end

return hash

end

if operation == "edit" and new_record?

return true # We get called again with the operation being set to create

end

current = User.current

if (operation == "edit") or operation == "destroy"

if current.allowed_to?("#{operation}_hosts".to_sym)

return true if Host.my_hosts(current).include? self

end

else # create

if current.allowed_to?(:create_hosts)

# We are unconstrained

return true if current.domains.empty? and current.hostgroups.empty?

(!current.hostgroups.empty? and current.hostgroups.include?(hostgroup))

end

end

errors.add_to_base "You do not have permission to #{operation} this host"

false

end

!sp_name.empty? and !sp_ip.empty? and !sp_mac.empty?

end

# align common mac and ip address input

helper = []

[self.mac,self.sp_mac].each do |m|

m.downcase!

if m=~/[a-f0-9]{12}/

m = m.gsub(/(..)/){|mh| mh + ":"}[/.{17}/]

elsif mac=~/([a-f0-9]{1,2}:){5}[a-f0-9]{1,2}/

m = m.split(":").map{|nibble| "%02x" % ("0x" + nibble)}.join(":")

end

end

[self.ip,self.sp_ip].each do |i|

unless i.empty?

i = i.split(".").map{|nibble| nibble.to_i}.join(".") if i=~/(\d{1,3}\.){3}\d{1,3}/

# it and let the validations to produce an error

if domain.nil? and name.match(/\./)

# try to assign the domain automaticilly based on our existing domains from the host FQDN

self.domain = Domain.all.select{|d| name.match(d.name)}.first rescue nil

else

# if our host is in short name, append the domain name