|
require 'facets'
|
|
|
|
module Facets
|
|
module ManagedHostExtensions
|
|
extend ActiveSupport::Concern
|
|
|
|
included do
|
|
Facets::ManagedHostExtensions.refresh_facet_relations(self)
|
|
|
|
def attributes
|
|
hash = super
|
|
|
|
# include all facet attributes by default
|
|
facets_with_definitions.each do |facet, facet_definition|
|
|
hash["#{facet_definition.name}_attributes"] = facet.attributes.reject { |key| %w(created_at updated_at).include? key }
|
|
end
|
|
hash
|
|
end
|
|
end
|
|
|
|
class << self
|
|
def refresh_facet_relations(klass)
|
|
Facets.registered_facets.values.each do |facet_config|
|
|
self.register_facet_relation(klass, facet_config)
|
|
end
|
|
end
|
|
|
|
# This method is used to add all relation objects necessary for accessing facet from the host object.
|
|
# It:
|
|
# 1. Adds active record one to one association
|
|
# 2. Adds the ability to set facet's attributes via Host#attributes= method
|
|
# 3. Extends Host::Managed model with extension module defined by facet's configuration
|
|
# 4. Includes facet in host's cloning mechanism
|
|
# 5. Adds compatibility properties forwarders so old property calls will still work after moving them to a facet:
|
|
# host.foo # => will call Host.my_facet.foo
|
|
def register_facet_relation(klass, facet_config)
|
|
klass.class_eval do
|
|
has_one facet_config.name, :class_name => facet_config.model.name, :foreign_key => :host_id, :inverse_of => :host
|
|
accepts_nested_attributes_for facet_config.name, :update_only => true
|
|
alias_method "#{facet_config.name}_attributes", facet_config.name
|
|
attr_accessible "#{facet_config.name}_attributes"
|
|
|
|
include facet_config.extension if facet_config.extension
|
|
|
|
include_in_clone facet_config.name
|
|
|
|
facet_config.compatibility_properties.each do |prop|
|
|
define_method(prop) { |*args| forward_property_call(prop, args, facet_config.name) }
|
|
end if facet_config.compatibility_properties
|
|
end
|
|
end
|
|
end
|
|
|
|
def facets
|
|
facets_with_definitions.keys
|
|
end
|
|
|
|
# This method will return a hash of facets for a specific host including the coresponding definitions.
|
|
# The output should look like this:
|
|
# { host.puppet_aspect => Facets.registered_facets[:puppet_aspect] }
|
|
def facets_with_definitions
|
|
Hash[(Facets.registered_facets.values.map do |facet_config|
|
|
facet = send(facet_config.name)
|
|
[facet, facet_config] if facet
|
|
end).compact]
|
|
end
|
|
|
|
private
|
|
|
|
def forward_property_call(property, args, facet)
|
|
facet_instance = send(facet)
|
|
return nil unless facet_instance
|
|
|
|
facet_instance.send(property, *args)
|
|
end
|
|
end
|
|
end
|
