51 lines
1.4 KiB
Ruby
51 lines
1.4 KiB
Ruby
module ActiveRecord
|
|
module Associations
|
|
class BelongsToPolymorphicAssociation < AssociationProxy #:nodoc:
|
|
def replace(record)
|
|
if record.nil?
|
|
@target = @owner[@reflection.primary_key_name] = @owner[@reflection.options[:foreign_type]] = nil
|
|
else
|
|
@target = (AssociationProxy === record ? record.target : record)
|
|
|
|
unless record.new_record?
|
|
@owner[@reflection.primary_key_name] = record.id
|
|
@owner[@reflection.options[:foreign_type]] = record.class.base_class.name.to_s
|
|
end
|
|
|
|
@updated = true
|
|
end
|
|
|
|
loaded
|
|
record
|
|
end
|
|
|
|
def updated?
|
|
@updated
|
|
end
|
|
|
|
private
|
|
def find_target
|
|
return nil if association_class.nil?
|
|
|
|
if @reflection.options[:conditions]
|
|
association_class.find(
|
|
@owner[@reflection.primary_key_name],
|
|
:conditions => conditions,
|
|
:include => @reflection.options[:include]
|
|
)
|
|
else
|
|
association_class.find(@owner[@reflection.primary_key_name], :include => @reflection.options[:include])
|
|
end
|
|
end
|
|
|
|
def foreign_key_present
|
|
!@owner[@reflection.primary_key_name].nil?
|
|
end
|
|
|
|
def association_class
|
|
@owner[@reflection.options[:foreign_type]] ? @owner[@reflection.options[:foreign_type]].constantize : nil
|
|
end
|
|
end
|
|
end
|
|
end
|