class Host < Puppet::Rails::Host

has_many :puppetclasses, :through => :host_classes

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",

["fv_#{fact}.value = ? and fn_#{fact}.name = ? and fv_#{fact}.fact_name_id = fn_#{fact}.id",value, fact ] }

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 >> #{BIT_NUM*METRIC.index("failed_restarts")} & #{MAX}) != 0)"

"((puppet_status >> #{BIT_NUM*METRIC.index("failed")} & #{MAX}) = 0) and

((puppet_status >> #{BIT_NUM*METRIC.index("failed_restarts")} & #{MAX}) = 0)"

}

"((puppet_status >> #{BIT_NUM*METRIC.index("applied")} & #{MAX}) = 0) or

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

}

named_scope :successful, lambda { without_changes.without_error.scope(:find)}

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

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 => Net::Validations::IP_REGEXP, :if => Proc.new {|host| host.managed}

validates_format_of :sp_ip, :with => Net::Validations::IP_REGEXP, :allow_nil => true, :allow_blank => true

end

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.

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"

# 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

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["foreman_env"] = environment.to_s unless environment.nil? or environment.name.nil?

if SETTINGS[:login]

param["owner_name"] = owner.name

param["owner_email"] = owner.is_a?(User) ? owner.mail : owner.users.map(&:mail)

end

param["mac"] = mac

end

info_hash = {}

info_hash['classes'] = self.puppetclasses_names

info_hash['parameters'] = param

return info_hash

end

@cached_host_params = nil

end

return cached_host_params unless cached_host_params.blank?

hp = {}

@cached_host_params = hp

end

return cached_lookup_keys_params unless cached_lookup_keys_params.blank?

p = {}

if Setting["Enable_Smart_Variables_in_ENC"]

klasses = puppetclasses.map(&:id)

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

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

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

# by default, puppet doesn't store an env name in the database

env = fv(:environment) || Setting[:default_puppet_environment]

if Setting[:update_environment_from_facts]

self.environment = Environment.find_or_create_by_name env

else

self.environment ||= Environment.find_or_create_by_name env

end

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

end

modelname = fv(:productname) || fv(:model) || (fv(:is_virtual) == "true" ? fv(:virtual) : nil)

self.model = Model.find_or_create_by_name(modelname.strip) unless modelname.empty?

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

operatingsystem.family == "Solaris" and architecture.name =~/Sparc/i rescue false

end

return unless hostgroup

assign_hostgroup_attributes(%w{operatingsystem medium architecture ptable root_pass subnet})

self.ip ||= subnet.unused_ip if subnet

end

def rundeck

unless self.params["rundeckfacts"].empty?

rdecktags += self.params["rundeckfacts"].split(",").map{|rdf| "#{rdf}=#{fact(rdf)[0].value}"}

end

{ name => { "description" => comment, "hostname" => name, "nodename" => name,