Project

General

Profile

Download (10 KB) Statistics
| Branch: | Tag: | Revision:
class Hostgroup < ActiveRecord::Base
include Authorizable
extend FriendlyId
friendly_id :title
include Taxonomix
include HostCommon

include NestedAncestryCommon
validates :name, :presence => true, :uniqueness => {:scope => :ancestry, :case_sensitive => false}
validates :title, :presence => true, :uniqueness => true

include ScopedSearchExtensions
include PuppetclassTotalHosts::Indirect

validates_lengths_from_database :except => [:name]
before_destroy EnsureNotUsedBy.new(:hosts)
has_many :hostgroup_classes
has_many :puppetclasses, :through => :hostgroup_classes, :dependent => :destroy
validates :name, :presence => true
validates :root_pass, :allow_blank => true, :length => {:minimum => 8, :message => _('should be 8 characters or more')}
has_many :group_parameters, :dependent => :destroy, :foreign_key => :reference_id, :inverse_of => :hostgroup
accepts_nested_attributes_for :group_parameters, :allow_destroy => true
include ParameterValidators
has_many_hosts :after_add => :update_puppetclasses_total_hosts,
:after_remove => :update_puppetclasses_total_hosts
has_many :template_combinations, :dependent => :destroy
has_many :provisioning_templates, :through => :template_combinations

include CounterCacheFix
counter_cache = "#{model_name.split(":").first.pluralize.downcase}_count".to_sym # e.g. :hosts_count
belongs_to :domain, :counter_cache => counter_cache
belongs_to :subnet

before_save :remove_duplicated_nested_class
after_save :update_ancestry_puppetclasses, :if => :ancestry_changed?

alias_attribute :arch, :architecture
alias_attribute :os, :operatingsystem
has_many :trends, :as => :trendable, :class_name => "ForemanTrend"

nested_attribute_for :compute_profile_id, :environment_id, :domain_id, :puppet_proxy_id, :puppet_ca_proxy_id,
:operatingsystem_id, :architecture_id, :medium_id, :ptable_id, :subnet_id, :realm_id

# with proc support, default_scope can no longer be chained
# include all default scoping here
default_scope lambda {
with_taxonomy_scope do
order("hostgroups.title")
end
}

scoped_search :on => :name, :complete_value => :true
scoped_search :in => :group_parameters, :on => :value, :on_key=> :name, :complete_value => true, :only_explicit => true, :rename => :params
scoped_search :in => :hosts, :on => :name, :complete_value => :true, :rename => "host"
scoped_search :in => :puppetclasses, :on => :name, :complete_value => true, :rename => :class, :operators => ['= ', '~ ']
scoped_search :in => :environment, :on => :name, :complete_value => :true, :rename => :environment
scoped_search :on => :id, :complete_enabled => false, :only_explicit => true
# for legacy purposes, keep search on :label
scoped_search :on => :title, :complete_value => true, :rename => :label
scoped_search :in => :config_groups, :on => :name, :complete_value => true, :rename => :config_group, :only_explicit => true, :operators => ['= ', '~ '], :ext_method => :search_by_config_group

def self.search_by_config_group(key, operator, value)
conditions = sanitize_sql_for_conditions(["config_groups.name #{operator} ?", value_to_sql(operator, value)])
hostgroup_ids = Hostgroup.unscoped.with_taxonomy_scope.joins(:config_groups).where(conditions).map(&:subtree_ids).flatten.uniq

opts = 'hostgroups.id < 0'
opts = "hostgroups.id IN(#{hostgroup_ids.join(',')})" unless hostgroup_ids.blank?
{:conditions => opts}
end

if SETTINGS[:unattended]
scoped_search :in => :architecture, :on => :name, :complete_value => true, :rename => :architecture
scoped_search :in => :operatingsystem, :on => :name, :complete_value => true, :rename => :os
scoped_search :in => :operatingsystem, :on => :description, :complete_value => true, :rename => :os_description
scoped_search :in => :operatingsystem, :on => :title, :complete_value => true, :rename => :os_title
scoped_search :in => :operatingsystem, :on => :major, :complete_value => true, :rename => :os_major
scoped_search :in => :operatingsystem, :on => :minor, :complete_value => true, :rename => :os_minor
scoped_search :in => :operatingsystem, :on => :id, :complete_enabled => false, :rename => :os_id, :only_explicit => true
scoped_search :in => :medium, :on => :name, :complete_value => true, :rename => "medium"
scoped_search :in => :provisioning_templates, :on => :name, :complete_value => true, :rename => "template"
end

# returns reports for hosts in the User's filter set
scope :my_groups, lambda {
user = User.current
unless user.admin?
conditions = sanitize_sql_for_conditions([" (hostgroups.id in (?))", user.hostgroup_ids])
conditions.sub!(/\s*\(\)\s*/, "")
conditions.sub!(/^(?:\(\))?\s?(?:and|or)\s*/, "")
conditions.sub!(/\(\s*(?:or|and)\s*\(/, "((")
end
where(conditions)
}

class Jail < Safemode::Jail
allow :name, :diskLayout, :puppetmaster, :operatingsystem, :architecture,
:environment, :ptable, :url_for_boot, :params, :puppetproxy, :param_true?,
:param_false?, :puppet_ca_server, :indent, :os, :arch, :domain, :subnet,
:realm, :root_pass
end

#TODO: add a method that returns the valid os for a hostgroup

def hostgroup
self
end

def diskLayout
ptable.layout.tr("\r", '')
end

def all_config_groups
(config_groups + parent_config_groups).uniq
end

def parent_config_groups
return [] unless parent
groups = []
ancestors.each do |hostgroup|
groups += hostgroup.config_groups
end
groups.uniq
end

# the environment used by #clases nees to be self.environment and not self.parent.environment
def parent_classes
return [] unless parent
parent.classes(self.environment)
end

def inherited_lookup_value(key)
ancestors.reverse.each do |hg|
if (v = LookupValue.where(:lookup_key_id => key.id, :id => hg.lookup_values).first)
return v.value, hg.to_label
end
end if key.path_elements.flatten.include?("hostgroup") && Setting["host_group_matchers_inheritance"]
[key.default_value, _("Default value")]
end

# returns self and parent parameters as a hash
def parameters(include_source = false)
hash = {}
ids = ancestor_ids
ids << id unless new_record? or self.frozen?
# need to pull out the hostgroups to ensure they are sorted first,
# otherwise we might be overwriting the hash in the wrong order.
groups = ids.size == 1 ? [self] : Hostgroup.sort_by_ancestry(Hostgroup.includes(:group_parameters).find(ids))
groups.each do |hg|
hg.group_parameters.each {|p| hash[p.name] = include_source ? {:value => p.value, :source => N_('hostgroup').to_sym, :safe_value => p.safe_value, :source_name => hg.title} : p.value }
end

hash
end

def global_parameters
Hostgroup.sort_by_ancestry(Hostgroup.includes(:group_parameters).find(ancestor_ids + id)).map(&:group_parameters).uniq
end

def inherited_params(include_source = false)
# same method in Host, similar method below
hp = {}
# read common parameters
CommonParameter.all.each {|p| hp.update Hash[p.name => include_source ? {:value => p.value, :source => N_('common').to_sym} : p.value] }
# read organization and location parameters
hp.update organization.parameters(include_source) if SETTINGS[:organizations_enabled] && organization
hp.update location.parameters(include_source) if SETTINGS[:locations_enabled] && location
# read domain parameters
domain.domain_parameters.each {|p| hp.update Hash[p.name => include_source ? {:value => p.value, :source => N_('domain').to_sym} : p.value] } unless domain.nil?
# read OS parameters
operatingsystem.os_parameters.each {|p| hp.update Hash[p.name => include_source ? {:value => p.value, :source => N_('os').to_sym} : p.value] } unless operatingsystem.nil?
# read group parameters only if a host belongs to a group
hp.update hostgroup.parameters(include_source) if hostgroup
hp
end

def params
parameters = {}
# read common parameters
CommonParameter.scoped.each {|p| parameters.update Hash[p.name => p.value] }
# read OS parameters
operatingsystem.os_parameters.each {|p| parameters.update Hash[p.name => p.value] } if operatingsystem
# read group parameters only if a host belongs to a group
parameters.update self.parameters if hostgroup
parameters
end

# no need to store anything in the db if the password is our default
def root_pass
return read_attribute(:root_pass) if read_attribute(:root_pass).present?
npw = nested_root_pw
return npw if npw.present?
Setting[:root_pass]
end

# Clone the hostgroup
def clone(name = "")
new = self.dup
new.name = name
new.title = name
new.puppetclasses = puppetclasses
new.locations = locations
new.organizations = organizations
new.config_groups = config_groups
new.set_lookup_value_matcher

# Clone any parameters as well
self.group_parameters.each{|param| new.group_parameters << param.clone}
self.lookup_values.each do |lookup_value|
new_lookup_value = lookup_value.dup
new_lookup_value.match = new.lookup_value_matcher
new.lookup_values << new_lookup_value
end
new
end

def update_ancestry_puppetclasses
unscoped_find(ancestry_was.to_s.split('/').last.to_i).update_puppetclasses_total_hosts if ancestry_was.present?
unscoped_find(ancestry.to_s.split('/').last.to_i).update_puppetclasses_total_hosts if ancestry.present?
end

private

def lookup_value_match
"hostgroup=#{to_label}"
end

def nested_root_pw
Hostgroup.sort_by_ancestry(ancestors).reverse.each do |a|
return a.root_pass unless a.root_pass.blank?
end if ancestry.present?
nil
end

def remove_duplicated_nested_class
self.puppetclasses -= ancestors.map(&:puppetclasses).flatten
end

# overwrite method in taxonomix, since hostgroup has ancestry
def used_taxonomy_ids(type)
return [] if new_record? && parent_id.blank?
Host::Base.where(:hostgroup_id => self.path_ids).uniq.pluck(type).compact
end
end
(24-24/62)