Module: Konduto::Associations

Included in:
KondutoBase
Defined in:
lib/konduto-ruby/associations/associations.rb

Instance Method Summary collapse

Instance Method Details

#has_many(model, options = {}) ⇒ Object



24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
# File 'lib/konduto-ruby/associations/associations.rb', line 24

def has_many(model, options = {})
  name = options[:alias] || model

  self.send(:define_method, name) do
    instance_variable_get("@#{name}") if instance_variable_defined?("@#{name}")
  end

  self.send(:define_method, "#{name}=".to_sym) do |arr|
    klass = "Konduto#{model.to_s.gsub(/_/, ' ').split.map(&:capitalize).join('')}"
    temp_arr = []

    arr.each do |value|
      if value.class.to_s == klass
        temp_arr << value
      elsif value.is_a? Hash
        temp_arr << Object::const_get(klass).new(value)
      else
        raise ArgumentError, "Expected class #{klass}, got #{value.class} at #{self.class}"
      end
    end

    instance_variable_set("@#{name}", temp_arr)
  end
end

#has_one(model, options = {}) ⇒ Object



4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
# File 'lib/konduto-ruby/associations/associations.rb', line 4

def has_one(model, options = {})
  name = options[:alias] || model

  self.send(:define_method, name) do
    instance_variable_get("@#{name}") if instance_variable_defined?("@#{name}")
  end

  self.send(:define_method, "#{name}=".to_sym) do |value|
    klass = "Konduto#{model.to_s.gsub(/_/, ' ').split.map(&:capitalize).join('')}"

    if value.class.to_s == klass
      instance_variable_set("@#{name}", value)
    elsif value.is_a? Hash
      instance_variable_set("@#{name}", Object::const_get(klass).new(value))
    else
      raise ArgumentError, "Expected class #{klass}, got #{value.class} at #{self.class}"
    end
  end
end