kotovalexarian-likes-github
/
paper-trail-gem--paper_trail
mirror of https://github.com/paper-trail-gem/paper_trail.git
1
0
Fork 0
Code Releases Activity

Extract private method save_bt_association

This commit is contained in:
Jared Beck 2016-12-13 22:32:11 -05:00
parent b4149864f4
commit 14eee417a7
1 changed files with 30 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
lib/paper_trail/record_trail.rb
View File

@ -346,29 +346,15 @@ module PaperTrail
# Saves associations if the join table for `VersionAssociation` exists. # Saves associations if the join table for `VersionAssociation` exists.
def save_associations(version) def save_associations(version)
return unless PaperTrail.config.track_associations? return unless PaperTrail.config.track_associations?
save_associations_belongs_to(version) save_bt_associations(version)
save_associations_habtm(version) save_habtm_associations(version)
end end
def save_associations_belongs_to(version) # Save all `belongs_to` associations.
# @api private
def save_bt_associations(version)
@record.class.reflect_on_all_associations(:belongs_to).each do |assoc| @record.class.reflect_on_all_associations(:belongs_to).each do |assoc|
assoc_version_args = { save_bt_association(assoc, version)
version_id: version.id,
foreign_key_name: assoc.foreign_key
}
if assoc.options[:polymorphic]
associated_record = @record.send(assoc.name) if @record.send(assoc.foreign_type)
if associated_record && associated_record.class.paper_trail.enabled?
assoc_version_args[:foreign_key_id] = associated_record.id
end
elsif assoc.klass.paper_trail.enabled?
assoc_version_args[:foreign_key_id] = @record.send(assoc.foreign_key)
end
if assoc_version_args.key?(:foreign_key_id)
PaperTrail::VersionAssociation.create(assoc_version_args)
end
end end
end end
@ -376,7 +362,8 @@ module PaperTrail
# HABTM associations looked like before any changes were made, by using # HABTM associations looked like before any changes were made, by using
# the `paper_trail_habtm` data structure. Then, we create # the `paper_trail_habtm` data structure. Then, we create
# `VersionAssociation` records for each of the associated records. # `VersionAssociation` records for each of the associated records.
def save_associations_habtm(version) # @api private
def save_habtm_associations(version)
@record.class.reflect_on_all_associations(:has_and_belongs_to_many).each do |a| @record.class.reflect_on_all_associations(:has_and_belongs_to_many).each do |a|
next unless save_habtm_association?(a) next unless save_habtm_association?(a)
habtm_assoc_ids(a).each do |id| habtm_assoc_ids(a).each do |id|
@ -525,6 +512,28 @@ module PaperTrail
) )
end end
# Save a single `belongs_to` association.
# @api private
def save_bt_association(assoc, version)
assoc_version_args = {
version_id: version.id,
foreign_key_name: assoc.foreign_key
}
if assoc.options[:polymorphic]
associated_record = @record.send(assoc.name) if @record.send(assoc.foreign_type)
if associated_record && associated_record.class.paper_trail.enabled?
assoc_version_args[:foreign_key_id] = associated_record.id
end
elsif assoc.klass.paper_trail.enabled?
assoc_version_args[:foreign_key_id] = @record.send(assoc.foreign_key)
end
if assoc_version_args.key?(:foreign_key_id)
PaperTrail::VersionAssociation.create(assoc_version_args)
end
end
# Returns true if the given HABTM association should be saved. # Returns true if the given HABTM association should be saved.
# @api private # @api private
def save_habtm_association?(assoc) def save_habtm_association?(assoc)