Module: BazaModels::Model::BelongsToRelations::ClassMethods

Defined in:
lib/baza_models/model/belongs_to_relations.rb

Instance Method Summary collapse

Instance Method Details

#belongs_to(relation_name, args = {}) ⇒ Object



7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
# File 'lib/baza_models/model/belongs_to_relations.rb', line 7

def belongs_to(relation_name, args = {})
  relation = {
    type: :belongs_to,
    relation_name: relation_name,
    table_name: args[:table_name] || StringCases.pluralize(relation_name),
    foreign_key: :"#{relation_name}_id"
  }

  if args[:class_name]
    relation[:class_name] = args.fetch(:class_name)
  else
    relation[:class_name] = StringCases.snake_to_camel(relation_name)
  end

  @belongs_to_relations ||= []
  @belongs_to_relations << relation

  @relationships ||= {}
  @relationships[relation_name] = relation

  define_method(relation_name) do
    if (model = @changes[relation_name])
      model
    elsif (model = autoloads[relation_name])
      model
    else
      if relation[:class_name]
        class_name = relation.fetch(:class_name)
      else
        class_name = StringCases.snake_to_camel(relation_name)
      end

      foreign_id = @data.fetch(relation.fetch(:foreign_key))
      StringCases.constantize(class_name).find(foreign_id) if foreign_id
    end
  end

  define_method("#{relation_name}=") do |new_model|
    @changes[relation.fetch(:foreign_key)] = new_model.id
    autoloads.delete(relation_name)
  end
end