Module: Tramway::Core::Associations::ClassHelper

Included in:
Tramway::Core::ApplicationDecorator
Defined in:
app/decorators/tramway/core/associations/class_helper.rb

Instance Method Summary collapse

Instance Method Details

#decorate_association(association_name, decorator: nil, as: nil, state_machines: []) ⇒ Object 



4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
 
# File 'app/decorators/tramway/core/associations/class_helper.rb', line 4

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

#decorate_associations(*association_names) ⇒ Object 



23
24
25
26
27
 
# File 'app/decorators/tramway/core/associations/class_helper.rb', line 23

def decorate_associations(*association_names)
  association_names.each do |association_name|
    decorate_association association_name
  end
end

#define_main_association_method(association_name, decorator) ⇒ Object 



29
30
31
32
33
34
35
36
37
38
39
 
# File 'app/decorators/tramway/core/associations/class_helper.rb', line 29

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