Project

General

Profile

Download (6.79 KB) Statistics
| Branch: | Tag: | Revision:
require 'tempfile'

module Foreman
module Renderer
ALLOWED_HELPERS = [ :foreman_url, :grub_pass, :snippet, :snippets,
:snippet_if_exists, :ks_console, :root_pass,
:media_path, :param_true?, :param_false?, :match, :indent ]

ALLOWED_VARIABLES = [ :arch, :host, :osver, :mediapath, :mediaserver, :static,
:repos, :dynamic, :kernel, :initrd,
:preseed_server, :preseed_path, :provisioning_type ]

def render_safe(template, allowed_methods = [], allowed_vars = {})
if Setting[:safemode_render]
box = Safemode::Box.new self, allowed_methods
box.eval(ERB.new(template, nil, '-').src, allowed_vars)
else
allowed_vars.each { |k,v| instance_variable_set "@#{k}", v }
ERB.new(template, nil, '-').result(binding)
end
end

#returns the URL for Foreman Built status (when a host has finished the OS installation)
def foreman_url(action = "built")
# Get basic stuff
config = URI.parse(Setting[:unattended_url])
protocol = config.scheme || 'http'
host = config.host || request.host
port = config.port || request.port
path = config.path

proxy = @host.try(:subnet).try(:tftp)

# use template_url from the request if set, but otherwise look for a Template
# feature proxy, as PXE templates are written without an incoming request.
url = if @template_url && @host.try(:token).present?
@template_url
elsif proxy.present? && proxy.has_feature?('Templates') && @host.try(:token).present?
temp_url = ProxyAPI::Template.new(:url => proxy.url).template_url
if temp_url.nil?
logger.warn("unable to obtain template url set by proxy #{proxy.url}. falling back on proxy url.")
temp_url = proxy.url
end
temp_url
end

if url.present?
uri = URI.parse(url)
host = uri.host
port = uri.port
protocol = uri.scheme
path = config.path
end

url_for :only_path => false, :controller => "/unattended", :action => action,
:protocol => protocol, :host => host, :port => port, :script_name => path,
:token => (@host.token.value unless @host.try(:token).nil?)
end

# provide embedded snippets support as simple erb templates
def snippets(file)
if Template.where(:name => file, :snippet => true).empty?
render :partial => "unattended/snippets/#{file}"
else
return snippet(file.gsub(/^_/, ""))
end
end

def snippet(name, options = {})
if (template = Template.where(:name => name, :snippet => true).first)
logger.debug "rendering snippet #{template.name}"
begin
return unattended_render(template)
rescue => exc
raise "The snippet '#{name}' threw an error: #{exc}"
end
else
if options[:silent]
nil
else
raise "The specified snippet '#{name}' does not exist, or is not a snippet."
end
end
end

def snippet_if_exists(name)
snippet name, :silent => true
end

def indent(count)
return unless block_given? && text=yield.to_s
prefix = " " * count
prefix + text.gsub(/\n/, "\n#{prefix}")
end

def unattended_render(template, template_name = nil)
content = template.respond_to?(:template) ? template.template : template
template_name ||= template.respond_to?(:name) ? template.name : 'Unnamed'
allowed_variables = ALLOWED_VARIABLES.reduce({}) do |mapping, var|
mapping.update(var => instance_variable_get("@#{var}"))
end
allowed_variables[:template_name] = template_name
render_safe content, ALLOWED_HELPERS, allowed_variables
end
alias_method :pxe_render, :unattended_render

def unattended_render_to_temp_file(content, prefix = id.to_s, options = {})
file = ""
Tempfile.open(prefix, Rails.root.join('tmp') ) do |f|
f.print(unattended_render(content))
f.flush
f.chmod options[:mode] if options[:mode]
file = f
end
file
end

# can be used to load additional variable relevant for give pxe type, requires @host to be present
def load_template_vars
# load the os family default variables
send "#{@host.operatingsystem.pxe_type}_attributes"

@provisioning_type = @host.is_a?(Hostgroup) ? 'hostgroup' : 'host'

# force static network configuration if static http parameter is defined, in the future this needs to go into the GUI
@static = !params[:static].empty?

# this is sent by the proxy when the templates feature is enabled
# and is needed to direct the host to the correct url. without it, we increase
# latency by requesting the correct url directly from the proxy.
@template_url = params['url']
end

private

def alterator_attributes
@mediapath = @host.operatingsystem.mediumpath @host
@mediaserver = URI(@mediapath).host
@metadata = params[:metadata].to_s
end

def jumpstart_attributes
if @host.operatingsystem.supports_image and @host.use_image
@install_type = "flash_install"
# We have an individual override for the host's image file
@archive_location = @host.image_file ? @host.image_file : @host.default_image_file
else
@install_type = "initial_install"
@system_type = "standalone"
@cluster = "SUNWCreq"
@packages = "SUNWgzip"
@locale = "C"
end
@disk = @host.diskLayout if @host.disk.present? || @host.ptable.present?
end

def kickstart_attributes
@dynamic = @host.diskLayout =~ /^#Dynamic/ if (@host.respond_to?(:disk) && @host.disk.present?) || @host.ptable.present?
@arch = @host.architecture.name
@osver = @host.operatingsystem.major.to_i
@mediapath = @host.operatingsystem.mediumpath @host if @host.medium
@repos = @host.operatingsystem.repos @host
end

def preseed_attributes
@preseed_path = @host.operatingsystem.preseed_path @host
@preseed_server = @host.operatingsystem.preseed_server @host
end

def yast_attributes
end

def coreos_attributes
@mediapath = @host.operatingsystem.mediumpath @host
end

def aif_attributes
@mediapath = @host.operatingsystem.mediumpath @host
end

def memdisk_attributes
@mediapath = @host.operatingsystem.mediumpath @host
end

def ZTP_attributes
@mediapath = @host.operatingsystem.mediumpath @host
end

def waik_attributes
end

def pxe_config
@kernel = @host.operatingsystem.kernel @host.arch
@initrd = @host.operatingsystem.initrd @host.arch
end
end
end
(7-7/8)