module LogicalModel::Associations::BelongsTo::ClassMethods

Public Instance Methods

belongs_to(key, options = {}) click to toggle source

@param key [String] association name @param options [Hash] @option options [String/Constant] class

# File lib/logical_model/associations/belongs_to.rb, line 13
def belongs_to(key, options = {})
  attr_accessor "#{key}_id"
  attr_class = get_attr_class(key, options)

  @belongs_to_keys ||= {}
  @belongs_to_keys.merge!({key => attr_class})

  define_method("#{key}=") do |param|
    if param.is_a?(Hash)
      param.stringify_keys!
      instance_variable_set("@#{key}_id", param['id']) if param['id']
      instance = attr_class.new(param)
    elsif param.is_a?(attr_class)
      instance_variable_set("@#{key}_id", param.id)
      instance = param
    else
      # ...
    end

    instance_variable_set("@#{key}",instance)
  end

  define_method(key) do
    instance = eval("@#{key}")
    if instance.nil?
      instance = attr_class.find(eval("#{key}_id"))
      instance_variable_set("@#{key}",instance)
    end
    instance
  end

  # TODO define_method("#{key}_attribute="){|param| ... }

  define_method "new_#{key}" do |param|
    attr_class

    return unless attr_class

    temp_object = attr_class.new(param.merge({"#{self.json_root}_id" => self.id}))
    eval(key.to_s) << temp_object
    temp_object
  end
end
belongs_to_keys() click to toggle source
# File lib/logical_model/associations/belongs_to.rb, line 57
def belongs_to_keys
  # This hack was needed to consider the case where the belongs_to was set on a parent class, for example:
  #
  # class ContactAttribute < LogicalModel
  #   ...
  #   belongs_to :contact
  # end
  #
  # class Telephone < ContactAttribute
  #   ...
  # end
  #
  # It returns the parent's class variable if present.
  result = nil
  if !@belongs_to_keys.nil?
    result = @belongs_to_keys
  elsif self.superclass.respond_to? :belongs_to_keys
    result = self.superclass.belongs_to_keys
  else
    result = nil
  end
  result
end

Private Instance Methods

get_attr_class(key, options) click to toggle source
# File lib/logical_model/associations/belongs_to.rb, line 83
def get_attr_class(key, options)
  if options[:class]
    attr_class = options[:class].is_a?(String) ? options[:class].constantize : options[:class]
  else
    attr_class = key.to_s.singularize.camelize.constantize
  end
  attr_class
end