Sha256: a64f90226bacfb65c3c5612974bf6b42dddd99a4060a8f72b4297757a23a7706
Contents?: true
Size: 1.38 KB
Versions: 23
Compression:
Stored size: 1.38 KB
Contents
# frozen_string_literal: true
module Tramway::Core::Associations::ClassHelper
def decorate_association(association_name, decorator: nil, as: nil, state_machines: [])
@@decorated_associations ||= []
@@decorated_associations << association_name
define_main_association_method association_name, decorator
define_method "#{association_name}_as" do
as
end
define_method "#{association_name}_state_machines" do
state_machines
end
define_method "add_#{association_name}_form" do
add_association_form_class_name(object, association_name).new object
end
end
def decorate_associations(*association_names)
association_names.each do |association_name|
decorate_association association_name
end
end
def define_main_association_method(association_name, decorator)
define_method association_name do
association = object.class.reflect_on_association(association_name)
check_association object, association_name, association
decorator_class_name = decorator || decorator_class_name(class_name(association))
if association_type(association).in? %i[has_many has_and_belongs_to_many]
return associations_collection(object, association_name, decorator_class_name)
end
return decorator_class_name.decorate object.send association_name if association_type(association) == :belongs_to
end
end
end
Version data entries
23 entries across 23 versions & 1 rubygems