Project

General

Profile

Download (10.6 KB) Statistics
| Branch: | Tag: | Revision:
class ProvisioningTemplate < Template
audited
has_many :audits, :as => :auditable, :class_name => Audited.audit_class.name

include Authorizable
extend FriendlyId
friendly_id :name
include Parameterizable::ByIdName
include DirtyAssociations
include TaxonomyCollisionFinder

class << self
# we have to override the base_class because polymorphic associations does not detect it correctly, more details at
# http://apidock.com/rails/ActiveRecord/Associations/ClassMethods/has_many#1010-Polymorphic-has-many-within-inherited-class-gotcha
def base_class
self
end
end
self.table_name = 'templates'

validates :name, :uniqueness => true
validates :template_kind_id, :presence => true, :unless => proc { |t| t.snippet }

before_destroy EnsureNotUsedBy.new(:hostgroups, :os_default_templates)
has_many :template_combinations, :dependent => :destroy
has_many :hostgroups, :through => :template_combinations
belongs_to :template_kind
accepts_nested_attributes_for :template_combinations, :allow_destroy => true,
:reject_if => :reject_template_combination_attributes?
has_and_belongs_to_many :operatingsystems, :join_table => :operatingsystems_provisioning_templates, :association_foreign_key => :operatingsystem_id, :foreign_key => :provisioning_template_id
has_many :os_default_templates
before_save :check_for_snippet_assoications

validate :no_os_for_registration

# these can't be shared in parent class, scoped search can't handle STI properly
# tested with scoped_search 3.2.0
include Taxonomix
include TemplateTax
scoped_search :on => :id, :complete_enabled => false, :only_explicit => true, :validator => ScopedSearch::Validators::INTEGER
scoped_search :on => :name, :complete_value => true, :default_order => true
scoped_search :on => :locked, :complete_value => {:true => true, :false => false}
scoped_search :on => :snippet, :complete_value => {:true => true, :false => false}
scoped_search :on => :template
scoped_search :on => :vendor, :only_explicit => true, :complete_value => true
scoped_search :on => :default, :only_explicit => true, :complete_value => {:true => true, :false => false}, :rename => 'default_template'

scoped_search :relation => :operatingsystems, :on => :name, :rename => :operatingsystem, :complete_value => true
scoped_search :relation => :hostgroups, :on => :name, :rename => :hostgroup, :complete_value => true
scoped_search :relation => :template_kind, :on => :name, :rename => :kind, :complete_value => true

attr_exportable({
:kind => proc { |template| template.template_kind.try(:name) },
:oses => proc { |template| template.operatingsystems.map(&:name).uniq },
}.merge(taxonomy_exportable))

dirty_has_many_associations :template_combinations, :os_default_templates, :operatingsystems

# Override method in Taxonomix as Template is not used attached to a Host,
# and matching a Host does not prevent removing a template from its taxonomy.
def used_taxonomy_ids(type)
[]
end

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

scope :of_kind, ->(kind) { joins(:template_kind).where("template_kinds.name" => kind) }

def self.template_ids_for(hosts)
hosts = hosts.with_os.distinct
oses = hosts.pluck(:operatingsystem_id)
templates = ProvisioningTemplate.reorder(nil).joins(:operatingsystems, :template_kind).where('operatingsystems.id' => oses, 'template_kinds.name' => 'provision')
ids = templates_by_template_combinations(templates, hosts).pluck(:id)
ids += templates.joins(:os_default_templates).where("os_default_templates.operatingsystem_id" => oses).distinct.pluck(:id)
ids.uniq
end

def self.templates_by_template_combinations(templates, hosts_or_conditions)
conditions = hosts_or_conditions
unless hosts_or_conditions.is_a?(Hash)
conditions = { hostgroup_id: hosts_or_conditions.pluck(:hostgroup_id) | [nil] }
end
templates.joins(:template_combinations).where("template_combinations.hostgroup_id" => conditions[:hostgroup_id]).distinct
end

def self.template_includes
super + [:template_kind, :template_combinations => [:hostgroup]]
end

def self.default_render_scope_class
Foreman::Renderer::Scope::Provisioning
end

def clone
deep_clone(:include => [:operatingsystems, :organizations, :locations],
:except => [:name, :locked, :default, :vendor])
end

def self.find_template(opts = {})
raise ::Foreman::Exception.new(N_("Must provide template kind")) unless opts[:kind]
raise ::Foreman::Exception.new(N_("Must provide an operating system")) unless opts[:operatingsystem_id]

# first filter valid templates to our OS and requested template kind.
templates = ProvisioningTemplate.joins(:operatingsystems, :template_kind).where('operatingsystems.id' => opts[:operatingsystem_id], 'template_kinds.name' => opts[:kind])
# once a template has been matched, we no longer look for others.
template = templates_by_template_combinations(templates, opts).first

# fall back to the os default template
template ||= templates.joins(:os_default_templates).find_by("os_default_templates.operatingsystem_id" => opts[:operatingsystem_id])
template.is_a?(ProvisioningTemplate) ? template : nil
end

def self.find_global_default_template(name, kind)
ProvisioningTemplate.unscoped.joins(:template_kind).find_by(:name => name, "template_kinds.name" => kind)
end

def self.local_boot_name(kind)
Setting["local_boot_#{kind}"]
end

def self.global_template_name_for(kind)
Setting["global_#{kind}"]
end

def self.build_pxe_default
return [:unprocessable_entity, _("No TFTP proxies defined, can't continue")] if (proxies = SmartProxy.with_features("TFTP")).empty?
error_msgs = []
used_templates = []
TemplateKind::PXE.each do |kind|
next if kind == 'iPXE'
global_template_name = global_template_name_for(kind)
if (default_template = find_global_default_template global_template_name, kind).nil?
error_msgs << _("Could not find a Configuration Template with the name \"%s\", please create one.") % global_template_name
else
begin
@profiles = pxe_default_combos
menu = default_template.render(variables: { profiles: @profiles })
rescue => exception
Foreman::Logging.exception("Cannot render '#{global_template_name}'", exception)
error_msgs << "#{exception.message} (#{kind})"
end
return [:unprocessable_entity, error_msgs.join(', ')] unless error_msgs.empty?
proxies.each do |proxy|
tftp = ProxyAPI::TFTP.new(:url => proxy.url)
tftp.create_default(kind, {:menu => menu})
fetch_boot_files_combo(tftp)
rescue => exception
Foreman::Logging.exception("Cannot deploy rendered template '#{global_template_name}' to '#{proxy}'", exception)
error_msgs << "#{proxy}: #{exception.message} (#{kind})"
end
end
used_templates << global_template_name
end

if error_msgs.empty?
[:ok, _("PXE files for templates %s have been deployed to all Smart Proxies") % used_templates.to_sentence]
else
[:internal_server_error, _("There was an error creating the PXE file: %s") % error_msgs.join(", ")]
end
end

def self.fetch_boot_files_combo(tftp)
@profiles.each do |combo|
medium_provider = Foreman::Plugin.medium_providers_registry.find_provider(combo[:hostgroup])
combo[:hostgroup].operatingsystem.pxe_files(medium_provider).each do |bootfile_info|
bootfile_info.each do |prefix, path|
tftp.fetch_boot_file(:prefix => prefix.to_s, :path => path)
end
end
end
end

def self.preview_host_collection
super.where(:managed => true)
end

# get a list of all hostgroup, template combinations that a pxemenu will be
# generated for
def self.pxe_default_combos
combos = []
ProvisioningTemplate.joins(:template_kind).where("template_kinds.name" => "provision").includes(:template_combinations => [{:hostgroup => [:operatingsystem, :architecture]}]).find_each do |template|
template.template_combinations.each do |combination|
hostgroup = combination.hostgroup
if hostgroup&.operatingsystem && hostgroup&.architecture
if (medium_provider = Foreman::Plugin.medium_providers_registry.find_provider(hostgroup))
combos << {
:hostgroup => hostgroup,
:template => template,
:kernel => hostgroup.operatingsystem.kernel(medium_provider),
:initrd => hostgroup.operatingsystem.initrd(medium_provider),
:pxe_type => hostgroup.operatingsystem.pxe_type,
}
else
Rails.logger.warn "Could not find medium_provider for hostgroup #{hostgroup}, skipping"
end
end
end
end
combos.sort_by! { |profile| [profile[:hostgroup].title, profile[:template]] }
end

def global_default?
return unless template_kind
setting = Setting.find_by :name => "global_#{template_kind.name}"
return unless setting
setting.value == name
end

def self.acceptable_template_input_types
[:fact, :variable, :puppet_parameter]
end

def no_os_for_registration
return unless template_kind&.name == 'registration'
errors.add(:operatingsystems, N_("can't assign operating system to Registration template")) if operatingsystems.any?
end

def support_single_host_render?
!registration_template?
end

def registration_template?
try(:template_kind)&.name == 'registration'
end

def host_init_config_template?
try(:template_kind)&.name == 'host_init_config'
end

private

def import_custom_data(options)
super
self.template_kind = nil if snippet

if @importing_metadata.key?('kind') && !snippet
kind = TemplateKind.find_by_name @importing_metadata['kind']
if kind.nil?
errors.add :template_kind_id, _('specified template "%s" kind was not found') % @importing_metadata['kind']
return
end
self.template_kind = kind
end
import_oses(options)
end

def allowed_changes
super + %w(template_combinations template_associations)
end

# check if our template is a snippet, and remove its associations just in case they were selected.
def check_for_snippet_assoications
return unless snippet
hostgroups.clear
template_combinations.clear
operatingsystems.clear
self.template_kind = nil
end

def reject_template_combination_attributes?(params)
params[:hostgroup_id].blank?
end
end
(40-40/69)