Sha256: 64b77bdbc71515882eb56ed49deeb18a6b7da273637c810b4f3846d25cb4f7fd
Contents?: true
Size: 1.56 KB
Versions: 36
Compression:
Stored size: 1.56 KB
Contents
module ObjectAttorney
class AssociationReflection < Reflection
attr_reader :related_reflection, :macro
def initialize(association, related_reflection, options)
super(association, options)
@macro = options[:macro] || macro_default(association)
@related_reflection = related_reflection
end
def primary_key
@primary_key ||= options[:primary_key] || :id
end
def foreign_key
@foreign_key ||= options[:foreign_key] || foreign_key_default
end
def set_relational_keys(origin, destination)
if has_many? || has_one?
set_foreign_key(destination, primary_key_of(origin))
elsif belongs_to?
set_foreign_key(origin, primary_key_of(destination))
end
end
def set_foreign_key(object, id)
setter = "#{foreign_key}="
if object.respond_to?(setter)
object.send(setter, id)
elsif object.respond_to?("send_to_representative")
object.send_to_representative(setter, id)
end
end
def primary_key_of(object)
object.send(primary_key)
end
def has_many?
macro == :has_many
end
def has_one?
macro == :has_one
end
def belongs_to?
macro == :belongs_to
end
private ################################# private
def macro_default(association)
Helpers.plural?(association) ? :has_many : :belongs_to
end
def foreign_key_default
if has_many? || has_one?
"#{related_reflection.single_name}_id"
elsif belongs_to?
"#{single_name}_id"
end.to_sym
end
end
end
Version data entries
36 entries across 36 versions & 1 rubygems