Project

General

Profile

Download (8.45 KB) Statistics
| Branch: | Tag: | Revision:
require 'fog_extensions'
class ComputeResource < ActiveRecord::Base
include Taxonomix
include Encryptable
include Authorizable
include Parameterizable::ByIdName
encrypts :password

class_attribute :supported_providers
self.supported_providers = {
'Libvirt' => 'Foreman::Model::Libvirt',
'Ovirt' => 'Foreman::Model::Ovirt',
'EC2' => 'Foreman::Model::EC2',
'Vmware' => 'Foreman::Model::Vmware',
'Openstack' => 'Foreman::Model::Openstack',
'Rackspace' => 'Foreman::Model::Rackspace',
'GCE' => 'Foreman::Model::GCE',
}

validates_lengths_from_database

audited :except => [:password, :attrs], :allow_mass_assignment => true
serialize :attrs, Hash
has_many :trends, :as => :trendable, :class_name => "ForemanTrend"

before_destroy EnsureNotUsedBy.new(:hosts)
validates :name, :presence => true, :uniqueness => true, :no_whitespace => true
validate :ensure_provider_not_changed, :on => :update
validates :provider, :presence => true, :inclusion => { :in => proc { self.providers } }
validates :url, :presence => true
scoped_search :on => :name, :complete_value => :true
scoped_search :on => :type, :complete_value => :true
scoped_search :on => :id, :complete_enabled => false, :only_explicit => true
before_save :sanitize_url
has_many_hosts
has_many :images, :dependent => :destroy
before_validation :set_attributes_hash
has_many :compute_attributes, :dependent => :destroy
has_many :compute_profiles, :through => :compute_attributes

# The DB may contain compute resource from disabled plugins - filter them out here
scope :live_descendants, lambda { where(:type => self.descendants.map(&:to_s)) unless Rails.env.development? }

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

def self.register_provider(provider)
name = provider.name.split('::').last
return if supported_providers.values.include?(provider) || supported_providers.keys.include?(name)
supported_providers[name] = provider.name
end

def self.providers
supported_providers.reject { |p,c| !SETTINGS[p.downcase.to_sym] }.keys
end

def self.provider_class(name)
supported_providers[name]
end

# allows to create a specific compute class based on the provider.
def self.new_provider(args)
raise ::Foreman::Exception.new(N_("must provide a provider")) unless provider = args.delete(:provider)
self.providers.each do |p|
return self.provider_class(p).constantize.new(args) if p.downcase == provider.downcase
end
raise ::Foreman::Exception.new N_("unknown provider")
end

def capabilities
[]
end

# attributes that this provider can provide back to the host object
def provided_attributes
{:uuid => :identity}
end

def test_connection(options = {})
valid?
end

def ping
test_connection
errors
end

def save_vm(uuid, attr)
vm = find_vm_by_uuid(uuid)
vm.attributes.merge!(attr.deep_symbolize_keys)
vm.save
end

def to_label
"#{name} (#{provider_friendly_name})"
end

# Override this method to specify provider name
def self.provider_friendly_name
self.name.split('::').last()
end

def provider_friendly_name
self.class.provider_friendly_name
end

def image_param_name
:image_id
end

def interfaces_attrs_name
:interfaces
end

# returns a new fog server instance
def new_vm(attr = {})
test_connection
client.servers.new vm_instance_defaults.merge(attr.to_hash.deep_symbolize_keys) if errors.empty?
end

# return fog new interface ( network adapter )
def new_interface(attr = {})
client.interfaces.new attr
end

# return a list of virtual machines
def vms(opts = {})
client.servers
end

def find_vm_by_uuid(uuid)
client.servers.get(uuid) || raise(ActiveRecord::RecordNotFound)
end

def start_vm(uuid)
find_vm_by_uuid(uuid).start
end

def stop_vm(uuid)
find_vm_by_uuid(uuid).stop
end

def create_vm(args = {})
options = vm_instance_defaults.merge(args.to_hash.deep_symbolize_keys)
logger.debug("creating VM with the following options: #{options.inspect}")
client.servers.create options
end

def destroy_vm(uuid)
find_vm_by_uuid(uuid).destroy
rescue ActiveRecord::RecordNotFound
# if the VM does not exists, we don't really care.
true
end

def provider
read_attribute(:type).to_s.split('::').last
end

def provider=(value)
if self.class.providers.include? value
self.type = self.class.provider_class(value)
else
self.type = value #this will trigger validation error since value is one of supported_providers
logger.debug("unknown provider for compute resource")
end
end

def vm_instance_defaults
ActiveSupport::HashWithIndifferentAccess.new(:name => "foreman_#{Time.now.to_i}")
end

def templates(opts = {})
end

def template(id,opts = {})
end

def update_required?(old_attrs, new_attrs)
old_attrs.merge(new_attrs) do |k,old_v,new_v|
update_required?(old_v, new_v) if old_v.is_a?(Hash)
return true unless old_v == new_v
new_v
end
false
end

def console(uuid = nil)
raise ::Foreman::Exception.new(N_("%s console is not supported at this time"), provider_friendly_name)
end

# by default, our compute providers do not support updating an existing instance
def supports_update?
false
end

def available_images
[]
end

def available_networks
raise ::Foreman::Exception.new(N_("Not implemented for %s"), provider_friendly_name)
end

def available_clusters
raise ::Foreman::Exception.new(N_("Not implemented for %s"), provider_friendly_name)
end

def available_folders
raise ::Foreman::Exception.new(N_("Not implemented for %s"), provider_friendly_name)
end

def available_resource_pools
raise ::Foreman::Exception.new(N_("Not implemented for %s"), provider_friendly_name)
end

def available_storage_domains(storage_domain = nil)
raise ::Foreman::Exception.new(N_("Not implemented for %s"), provider_friendly_name)
end

# this method is overwritten for Libvirt
def editable_network_interfaces?
networks.any?
end

# this method is overwritten for Libvirt and VMware
def set_console_password?
false
end
alias_method :set_console_password, :set_console_password?

# this method is overwritten for Libvirt and VMware
def set_console_password=(setpw)
self.attrs[:setpw] = nil
end

def compute_profile_for(id)
compute_attributes.find_by_compute_profile_id(id)
end

def compute_profile_attributes_for(id)
compute_profile_for(id).try(:vm_attrs) || {}
end

def vm_compute_attributes_for(uuid)
vm = find_vm_by_uuid(uuid)
vm_attrs = vm.attributes rescue {}
vm_attrs = vm_attrs.reject{|k,v| k == :id }

if vm.respond_to?(:volumes)
volumes = vm.volumes || []
vm_attrs[:volumes_attributes] = Hash[volumes.each_with_index.map { |volume, idx| [idx.to_s, volume.attributes] }]
end
vm_attrs
rescue ActiveRecord::RecordNotFound
logger.warn("VM with UUID '#{uuid}' not found on #{self}")
{}
end

def user_data_supported?
false
end

def image_exists?(image)
true
end

protected

def client
raise ::Foreman::Exception.new N_("Not implemented")
end

def sanitize_url
self.url.chomp!("/") unless url.empty?
end

def random_password
return nil unless set_console_password?
SecureRandom.hex(8)
end

def nested_attributes_for(type, opts)
return [] unless opts
opts = opts.dup #duplicate to prevent changing the origin opts.
opts.delete("new_#{type}") || opts.delete("new_#{type}".to_sym) # delete template
# convert our options hash into a sorted array (e.g. to preserve nic / disks order)
opts = opts.sort { |l, r| l[0].to_s.sub('new_','').to_i <=> r[0].to_s.sub('new_','').to_i }.map { |e| Hash[e[1]] }
opts.map do |v|
if v[:"_delete"] == '1' && v[:id].blank?
nil
else
v.deep_symbolize_keys # convert to symbols deeper hashes
end
end.compact
end

def associate_by(name, attributes)
Host.authorized(:view_hosts, Host).joins(:primary_interface).
where(:nics => {:primary => true}).
where("nics.#{name}" => attributes).
readonly(false).
first
end

private

def set_attributes_hash
self.attrs ||= {}
end

def ensure_provider_not_changed
errors.add(:provider, _("cannot be changed")) if self.type_changed?
end
end
(8-8/62)