draft_punk 0.3.0 → 0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 904244863752dbaa1f78496c19b136af59d1dba5
-  data.tar.gz: 8c4e60640cd7728e6988253af6cdbf0bfc5d86d4
+  metadata.gz: d72097293299128db20d9aefa7a69a4722eb1a10
+  data.tar.gz: abd611675c762f6ac56e51b908489fb11260cc26
 SHA512:
-  metadata.gz: 02d77960c5c1771d34b2d2b17bb32d1061e9d53ae8223c64b8df0286814bb3056d24e50a93ef37c51b439be10fed8b79090b6c491d0a822deae18783e618ae8f
-  data.tar.gz: d7844a4ccdc4e2cfe552499e913d25a805d9bbadfcbdc14974a703fa0d72d13ee34aff01ef93538ea2308f50610eeb0a55b6f424b005ba2d13363a57886c3cd5
+  metadata.gz: 5b42ee587d685ad6a454e8aea9b42b014fad40b9aeecfa0b9fa92500c1639bf1bde49a73ebe03bf27bf506fa60bd09d0116faa3ee00f29e986794a5a5ff191bb
+  data.tar.gz: 303f073323eb193f5e5a87aed1546abe3251ca09e7456123ac6c98efdc847c30466a5abb180c8228c463743197ee57de1b5fbbd35ffb197b4f9ad24e2f0dd33a

data/lib/activerecord/activerecord_class_methods.rb

@@ -0,0 +1,52 @@
+module DraftPunk
+  module Model
+  	module ActiveRecordClassMethods
+      # List of association names this model is configured to have drafts for
+      # @return [Array]
+      def draft_target_associations
+        targets = if const_defined?(:CREATES_NESTED_DRAFTS_FOR) && const_get(:CREATES_NESTED_DRAFTS_FOR).is_a?(Array)
+          const_get(:CREATES_NESTED_DRAFTS_FOR).compact
+        else
+          default_draft_target_associations
+        end.map(&:to_sym)
+      end
+
+      # Whether this model is configured to track the approved version of a draft object.
+      # This will be true if the model has an approved_version_id column
+      #
+      # @return (Boolean)
+      def tracks_approved_version?
+        column_names.include? 'approved_version_id'
+      end
+
+      # Whether this model is configured to store previously-approved versions of the model.
+      # This will be true if the model has an current_approved_version_id column
+      #
+      # @return (Boolean)
+      def tracks_approved_version_history?
+        column_names.include?('current_approved_version_id')
+      end
+
+      # Callback runs after save to set the approved version id on the draft object.
+      #
+      # @return (true)
+      def set_approved_version_id_callback
+        lambda do |live_obj, draft_obj|
+          draft_obj.approved_version_id = live_obj.id if draft_obj.respond_to?(:approved_version_id)
+          draft_obj.temporary_approved_object = live_obj
+          true
+        end
+      end
+
+    protected #################################################################
+
+      def default_draft_target_associations
+        reflect_on_all_associations.select do |reflection|
+          DraftPunk.is_relevant_association_type?(reflection) &&
+          !reflection.name.in?(%i(draft approved_version)) &&
+          reflection.class_name != name # Self referential associations!!! Don't do them!
+        end.map{|r| r.name.downcase.to_sym }
+      end
+  	end
+  end
+end

data/lib/{activerecord_instance_methods.rb → activerecord/activerecord_instance_methods.rb}

@@ -59,13 +59,18 @@ module DraftPunk
         @live_version  = get_approved_version
         @draft_version = editable_version
         return unless changes_require_approval? && @draft_version.is_draft? # No-op. ie. the live version is in a state that doesn't require approval.
+        @live_version.instance_variable_set :@publishing_draft, true
         transaction do
-          create_historic_version_of_approved_object if tracks_approved_version_history?
-          save_attribute_changes_and_belongs_to_assocations_from_draft
-          update_has_many_and_has_one_associations_from_draft
-          # We have to destroy the draft this since we moved all the draft's has_many associations to @live_version. If you call "editable_version" later, it'll build the draft.
-          # We destroy_all in case extra drafts are in the database. Extra drafts can potentially be created due to race conditions in the application.
-          self.class.unscoped.where(approved_version_id: @live_version.id).destroy_all
+          begin
+            create_historic_version_of_approved_object if tracks_approved_version_history?
+            save_attribute_changes_and_belongs_to_assocations_from_draft
+            update_has_many_and_has_one_associations_from_draft
+            # We have to destroy the draft this since we moved all the draft's has_many associations to @live_version. If you call "editable_version" later, it'll build the draft.
+            # We destroy_all in case extra drafts are in the database. Extra drafts can potentially be created due to race conditions in the application.
+            self.class.unscoped.where(approved_version_id: @live_version.id).destroy_all
+          ensure
+            @live_version.remove_instance_variable :@publishing_draft
+          end
         end
         @live_version = self.class.find(@live_version.id)
       end
@@ -90,6 +95,10 @@ module DraftPunk
         self.class.tracks_approved_version_history?
       end
 
+      def publishing_draft?
+        !!@publishing_draft
+      end
+
     protected #################################################################
 
       def get_draft
@@ -190,7 +199,7 @@ module DraftPunk
       # @return [Boolean] whether the current ActiveRecord object has a draft version
       def has_draft?
         raise DraftPunk::ApprovedVersionIdError unless respond_to?(:approved_version_id)
-        draft.present?
+        draft.present? && !publishing_draft?
       end
 
       # @return [Boolean] whether the current ActiveRecord object is a previously-approved

data/lib/activerecord/macros.rb

@@ -0,0 +1,70 @@
+require_relative 'setup_model'
+
+module DraftPunk
+  module Model
+    module Macros
+      # Call this method in your model to setup approval. It will recursively apply to its associations,
+      # thus does not need to be explicity called on its associated models (and will error if you try).
+      #
+      # This model must have an approved_version_id column (Integer), which will be used to track its draft
+      # 
+      # For instance, your Business model:
+      #  class Business << ActiveRecord::Base
+      #    has_many :employees
+      #    has_many :images
+      #    has_one  :address
+      #    has_many :vending_machines
+      #
+      #    requires_approval # When creating a business's draft, :employees, :vending_machines, :images, and :address will all have drafts created
+      #  end
+      #
+      # Optionally, specify which associations which the user will edit - associations which should have a draft created
+      # by defining a CREATES_NESTED_DRAFTS_FOR constant for this model.
+      #  
+      #  CREATES_NESTED_DRAFTS_FOR = [:address] # When creating a business's draft, only :address will have drafts created
+      #
+      # To disable drafts for all assocations for this model, simply pass an empty array:
+      # by defining a CREATES_NESTED_DRAFTS_FOR constant for this model.
+      #  CREATES_NESTED_DRAFTS_FOR = [] # When creating a business's draft, no associations will have drafts created
+      #
+      # WARNING: If you are setting associations via accepts_nested_attributes all changes to the draft, including associations, get set on the
+      # draft object (as expected). If your form includes associated objects which weren't defined in requires_approval, your save will fail since
+      # the draft object doesn't HAVE those associations to update! In this case, you should probably add that association to the
+      # +associations+ param here.
+      #
+      # If you want your draft associations to track their live version, add an :approved_version_id column
+      # to each association's table. You'll be able to access that associated object's live version, just
+      # like you can with the original model which called requires_approval. 
+      #
+      # @param nullify [Array] A list of attributes on this model to set to null on the draft when it is created. For
+      #    instance, the _id_ and _created_at_ columns are nullified by default, since you don't want Rails to try to
+      #    persist those on the draft.
+      # @param set_default_scope [Boolean] If true, set a default scope on this model for the approved scope; only approved objects
+      #    will be returned in ActiveRecord queries, unless you call Model.unscoped
+      # @param allow_previous_versions_to_be_changed [Boolean] If the model tracks approved version history, and this
+      #    param is false, previously-approved versions of the object cannot be saved (via a before_save callback )
+      # @param associations [Array] Use internally; set associations to create drafts for in the CREATES_NESTED_DRAFTS_FOR constant
+      # @return true
+      def requires_approval(associations: [], nullify: [], set_default_scope: false, allow_previous_versions_to_be_changed: true)
+        DraftPunk::SetupModel.new(self,
+                                   associations: associations,
+                                        nullify: nullify,
+                              set_default_scope: set_default_scope,
+          allow_previous_versions_to_be_changed: allow_previous_versions_to_be_changed
+        ).make_approvable!
+      end
+
+      # This will generally be only used in testing scenarios, in cases when requires_approval need to be
+      # called multiple times. Only the usage for that use case is supported. Use at your own risk for other
+      # use cases.
+      def disable_approval!
+        send(:remove_const, :DRAFT_PUNK_IS_SETUP)                   if const_defined? :DRAFT_PUNK_IS_SETUP
+        send(:remove_const, :DRAFT_NULLIFY_ATTRIBUTES)              if const_defined? :DRAFT_NULLIFY_ATTRIBUTES
+        send(:remove_const, :ALLOW_PREVIOUS_VERSIONS_TO_BE_CHANGED) if const_defined? :ALLOW_PREVIOUS_VERSIONS_TO_BE_CHANGED
+        fresh_amoeba do
+          disable
+        end
+      end
+    end
+  end
+end

data/lib/activerecord/setup_model.rb

@@ -0,0 +1,116 @@
+module DraftPunk
+
+  def self.is_relevant_association_type?(activerecord_reflection)
+    activerecord_reflection.macro.in? Amoeba::Config::DEFAULTS[:known_macros]
+  end
+
+	class SetupModel
+		require_relative 'activerecord_class_methods'
+		require_relative 'activerecord_instance_methods'
+		require_relative 'draft_diff_instance_methods'
+		require_relative 'previous_version_instance_methods'
+
+		def initialize(top_level_model, associations: [], nullify: [], set_default_scope: false, allow_previous_versions_to_be_changed: true)
+			@top_level_model, @associations, @nullify = top_level_model, associations, nullify
+			@set_default_scope, @allow_previous_versions_to_be_changed = set_default_scope, allow_previous_versions_to_be_changed
+		end
+
+		def make_approvable!
+      return unless model_table_exists? # Short circuits if you're migrating
+      @top_level_model.extend Model::ActiveRecordClassMethods
+
+      @associations = @top_level_model.draft_target_associations if @associations.empty?
+      set_valid_associations @top_level_model, @associations
+
+      raise DraftPunk::ConfigurationError, "Cannot call requires_approval multiple times for #{@top_level_model.name}" if @top_level_model.const_defined?(:DRAFT_PUNK_IS_SETUP)
+      @top_level_model.const_set :DRAFT_NULLIFY_ATTRIBUTES, Array(@nullify).flatten.freeze
+      @top_level_model.amoeba do
+        nullify @nullify
+        # Note that the amoeba associations and customize options are being set in setup_associations_and_scopes_for
+      end
+      setup_amoeba_for @top_level_model, set_default_scope: @set_default_scope, allow_previous_versions_to_be_changed: @allow_previous_versions_to_be_changed
+      true
+		end
+
+	private #####################################################################
+
+    def model_table_exists?(table_name=@top_level_model.table_name)
+      if ActiveRecord::Base.connection.respond_to? :data_source_exists?
+        ActiveRecord::Base.connection.data_source_exists?(table_name)
+      else
+        ActiveRecord::Base.connection.table_exists?(table_name)
+      end
+    end
+
+    def setup_amoeba_for(target_model, options={})
+      return if target_model.const_defined?(:DRAFT_PUNK_IS_SETUP)
+			target_model.extend Model::ActiveRecordClassMethods
+      target_associations = target_model.draft_target_associations
+      target_associations = set_valid_associations(target_model, target_associations)
+      target_model.amoeba do
+        enable 
+        include_associations target_model.const_get(:DRAFT_VALID_ASSOCIATIONS) unless target_model.const_get(:DRAFT_VALID_ASSOCIATIONS).empty?
+        customize target_model.set_approved_version_id_callback
+      end
+      target_model.const_set :DRAFT_PUNK_IS_SETUP, true.freeze
+
+      setup_associations_and_scopes_for target_model, **options
+      setup_draft_association_persistance_for_children_of target_model, target_associations
+    end
+
+    def setup_associations_and_scopes_for(target_model, set_default_scope: false, allow_previous_versions_to_be_changed: true)
+      target_model.send       :include, Model::InstanceInterrogators unless target_model.method_defined?(:has_draft?)
+      target_model.send       :attr_accessor, :temporary_approved_object
+      target_model.send       :before_create, :before_create_draft if target_model.method_defined?(:before_create_draft)
+
+      target_model.const_set :ALLOW_PREVIOUS_VERSIONS_TO_BE_CHANGED, allow_previous_versions_to_be_changed.freeze
+      if target_model.tracks_approved_version_history?
+        target_model.belongs_to    :current_approved_version, class_name: target_model.name, optional: true
+        target_model.has_many      :previous_versions, -> { order(id: :desc) }, class_name: target_model.name, foreign_key: :current_approved_version_id
+        target_model.before_update :prevent_previous_versions_from_saving
+        target_model.send          :include, Model::PreviousVersionInstanceMethods
+      end
+
+      if set_default_scope
+        target_model.default_scope -> { approved }
+      end
+
+      return if target_model.reflect_on_association(:approved_version) || !target_model.column_names.include?('approved_version_id')
+      target_model.send       :include, Model::ActiveRecordInstanceMethods
+      target_model.send       :include, Model::DraftDiffInstanceMethods
+      target_model.belongs_to :approved_version, class_name: target_model.name, optional: true
+      target_model.scope      :approved, -> { where("#{target_model.quoted_table_name}.approved_version_id IS NULL") }
+      target_model.has_one    :draft, -> { unscope(where: :approved) }, class_name: target_model.name, foreign_key: :approved_version_id
+      target_model.scope      :draft, -> { unscoped.where("#{target_model.quoted_table_name}.approved_version_id IS NOT NULL") }
+    end
+
+    def setup_draft_association_persistance_for_children_of(target_model, target_associations=nil)
+      target_associations = target_model.draft_target_associations unless target_associations
+      target_reflections = target_associations.map do |assoc|
+        reflection = target_model.reflect_on_association(assoc.to_sym)
+        reflection.presence || (raise DraftPunk::ConfigurationError, "#{name} includes invalid association (#{assoc})")
+      end
+      target_reflections.select{|r| DraftPunk.is_relevant_association_type?(r) }.each do |assoc|
+        setup_amoeba_for assoc.klass
+      end
+    end
+
+    # Rejects the associations if the table hasn't been defined yet. This happens when
+    # running migrations which add that association's table.
+    def set_valid_associations(target_model, associations)
+      return target_model.const_get(:DRAFT_VALID_ASSOCIATIONS) if target_model.const_defined?(:DRAFT_VALID_ASSOCIATIONS)
+      associations = associations.map(&:to_sym)
+      valid_assocations = associations.select do |assoc|
+        reflection = target_model.reflect_on_association(assoc)
+        if reflection
+          table_name = reflection.klass.table_name
+          model_table_exists?(table_name)
+        else
+          false
+        end
+      end
+      target_model.const_set :DRAFT_VALID_ASSOCIATIONS, valid_assocations.freeze
+      valid_assocations
+    end
+	end
+end

data/lib/draft_punk.rb

@@ -1,15 +1,9 @@
 require 'draft_punk/version'
 require 'amoeba'
-require 'activerecord_class_methods'
-require 'helper_methods'
+require 'activerecord/macros'
+require 'helpers/helper_methods'
 
 module DraftPunk
-  module Model
-    def self.included(base)
-      base.send :extend, ActiveRecordClassMethods
-    end
-  end
-
   class ConfigurationError < RuntimeError
     def initialize(message)
     	@caller = caller[0]
@@ -46,9 +40,8 @@ module DraftPunk
       "the draft failed to be created: #{@message}"
     end
   end
-
 end
 
 ActiveSupport.on_load(:active_record) do
-  include DraftPunk::Model
+  extend DraftPunk::Model::Macros
 end

data/lib/draft_punk/version.rb

@@ -1,3 +1,3 @@
 module DraftPunk
-  VERSION = "0.3.0"
+  VERSION = "0.3.1"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: draft_punk
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.3.1
 platform: ruby
 authors:
 - Steve Hodges
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-09-28 00:00:00.000000000 Z
+date: 2018-12-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: amoeba
@@ -185,13 +185,15 @@ files:
 - draft_punk-0.2.0.gem
 - draft_punk-0.2.1.gem
 - draft_punk.gemspec
-- lib/activerecord_class_methods.rb
-- lib/activerecord_instance_methods.rb
-- lib/draft_diff_instance_methods.rb
+- lib/activerecord/activerecord_class_methods.rb
+- lib/activerecord/activerecord_instance_methods.rb
+- lib/activerecord/draft_diff_instance_methods.rb
+- lib/activerecord/macros.rb
+- lib/activerecord/previous_version_instance_methods.rb
+- lib/activerecord/setup_model.rb
 - lib/draft_punk.rb
 - lib/draft_punk/version.rb
-- lib/helper_methods.rb
-- lib/previous_version_instance_methods.rb
+- lib/helpers/helper_methods.rb
 homepage: https://github.com/stevehodges/draftpunk
 licenses:
 - MIT

data/lib/activerecord_class_methods.rb

@@ -1,212 +0,0 @@
-require 'activerecord_instance_methods'
-require 'draft_diff_instance_methods'
-require 'previous_version_instance_methods'
-
-module DraftPunk
-  module Model
-    module ActiveRecordClassMethods
-      # Call this method in your model to setup approval. It will recursively apply to its associations,
-      # thus does not need to be explicity called on its associated models (and will error if you try).
-      #
-      # This model must have an approved_version_id column (Integer), which will be used to track its draft
-      # 
-      # For instance, your Business model:
-      #  class Business << ActiveRecord::Base
-      #    has_many :employees
-      #    has_many :images
-      #    has_one  :address
-      #    has_many :vending_machines
-      #
-      #    requires_approval # When creating a business's draft, :employees, :vending_machines, :images, and :address will all have drafts created
-      #  end
-      #
-      # Optionally, specify which associations which the user will edit - associations which should have a draft created
-      # by defining a CREATES_NESTED_DRAFTS_FOR constant for this model.
-      #  
-      #  CREATES_NESTED_DRAFTS_FOR = [:address] # When creating a business's draft, only :address will have drafts created
-      #
-      # To disable drafts for all assocations for this model, simply pass an empty array:
-      # by defining a CREATES_NESTED_DRAFTS_FOR constant for this model.
-      #  CREATES_NESTED_DRAFTS_FOR = [] # When creating a business's draft, no associations will have drafts created
-      #
-      # WARNING: If you are setting associations via accepts_nested_attributes all changes to the draft, including associations, get set on the
-      # draft object (as expected). If your form includes associated objects which weren't defined in requires_approval, your save will fail since
-      # the draft object doesn't HAVE those associations to update! In this case, you should probably add that association to the
-      # +associations+ param here.
-      #
-      # If you want your draft associations to track their live version, add an :approved_version_id column
-      # to each association's table. You'll be able to access that associated object's live version, just
-      # like you can with the original model which called requires_approval. 
-      #
-      # @param nullify [Array] A list of attributes on this model to set to null on the draft when it is created. For
-      #    instance, the _id_ and _created_at_ columns are nullified by default, since you don't want Rails to try to
-      #    persist those on the draft.
-      # @param set_default_scope [Boolean] If true, set a default scope on this model for the approved scope; only approved objects
-      #    will be returned in ActiveRecord queries, unless you call Model.unscoped
-      # @param allow_previous_versions_to_be_changed [Boolean] If the model tracks approved version history, and this
-      #    param is false, previously-approved versions of the object cannot be saved (via a before_save callback )
-      # @param associations [Array] Use internally; set associations to create drafts for in the CREATES_NESTED_DRAFTS_FOR constant
-      # @return true
-      def requires_approval(associations: [], nullify: [], set_default_scope: false, allow_previous_versions_to_be_changed: true)
-        return unless draft_punk_table_exists?(table_name) # Short circuits if you're migrating
-
-        associations = draft_target_associations if associations.empty?
-        set_valid_associations(associations)
-
-        raise DraftPunk::ConfigurationError, "Cannot call requires_approval multiple times for #{name}" if const_defined?(:DRAFT_PUNK_IS_SETUP)
-        self.const_set :DRAFT_NULLIFY_ATTRIBUTES, Array(nullify).flatten.freeze
-
-        amoeba do
-          nullify nullify
-          # Note that the amoeba associations and customize options are being set in setup_associations_and_scopes_for
-        end
-        setup_amoeba_for self, set_default_scope: set_default_scope, allow_previous_versions_to_be_changed: allow_previous_versions_to_be_changed
-        true
-      end
-
-      # This will generally be only used in testing scenarios, in cases when requires_approval need to be
-      # called multiple times. Only the usage for that use case is supported. Use at your own risk for other
-      # use cases.
-      def disable_approval!
-        send(:remove_const, :DRAFT_PUNK_IS_SETUP)                   if const_defined? :DRAFT_PUNK_IS_SETUP
-        send(:remove_const, :DRAFT_NULLIFY_ATTRIBUTES)              if const_defined? :DRAFT_NULLIFY_ATTRIBUTES
-        send(:remove_const, :ALLOW_PREVIOUS_VERSIONS_TO_BE_CHANGED) if const_defined? :ALLOW_PREVIOUS_VERSIONS_TO_BE_CHANGED
-        fresh_amoeba do
-          disable
-        end
-      end
-
-      # List of association names this model is configured to have drafts for
-      # @return [Array]
-      def draft_target_associations
-        targets = if const_defined?(:CREATES_NESTED_DRAFTS_FOR) && const_get(:CREATES_NESTED_DRAFTS_FOR).is_a?(Array)
-          const_get(:CREATES_NESTED_DRAFTS_FOR).compact
-        else
-          default_draft_target_associations
-        end.map(&:to_sym)
-      end
-
-      # Whether this model is configured to track the approved version of a draft object.
-      # This will be true if the model has an approved_version_id column
-      #
-      # @return (Boolean)
-      def tracks_approved_version?
-        column_names.include? 'approved_version_id'
-      end
-
-      # Whether this model is configured to store previously-approved versions of the model.
-      # This will be true if the model has an current_approved_version_id column
-      #
-      # @return (Boolean)
-      def tracks_approved_version_history?
-        column_names.include?('current_approved_version_id')
-      end
-
-    protected #################################################################
-
-      def default_draft_target_associations
-        reflect_on_all_associations.select do |reflection|
-          is_relevant_association_type?(reflection) &&
-          !reflection.name.in?(%i(draft approved_version)) &&
-          reflection.class_name != name # Self referential associations!!! Don't do them!
-        end.map{|r| r.name.downcase.to_sym }
-      end
-
-      # Rejects the associations if the table hasn't been defined yet. This happens when
-      # running migrations which add that association's table.
-      def set_valid_associations(associations)
-        return const_get(:DRAFT_VALID_ASSOCIATIONS) if const_defined?(:DRAFT_VALID_ASSOCIATIONS)
-        associations = associations.map(&:to_sym)
-        valid_assocations = associations.select do |assoc|
-          reflection = reflect_on_association(assoc)
-          if reflection
-            table_name = reflection.klass.table_name
-            draft_punk_table_exists?(table_name)
-          else
-            false
-          end
-        end
-        self.const_set :DRAFT_VALID_ASSOCIATIONS, valid_assocations.freeze
-        valid_assocations
-      end
-
-    private ###################################################################
-
-      def setup_amoeba_for(target_class, options={})
-        return if target_class.const_defined?(:DRAFT_PUNK_IS_SETUP)
-        associations = target_class.draft_target_associations
-        associations = target_class.set_valid_associations(associations)
-        target_class.amoeba do
-          enable 
-          include_associations target_class.const_get(:DRAFT_VALID_ASSOCIATIONS) unless target_class.const_get(:DRAFT_VALID_ASSOCIATIONS).empty?
-          customize target_class.set_approved_version_id_callback
-        end
-        target_class.const_set :DRAFT_PUNK_IS_SETUP, true.freeze
-
-        setup_associations_and_scopes_for target_class, **options
-        setup_draft_association_persistance_for_children_of target_class, associations
-      end
-
-    public ###################################################################
-
-      def set_approved_version_id_callback
-        lambda do |live_obj, draft_obj|
-          draft_obj.approved_version_id = live_obj.id if draft_obj.respond_to?(:approved_version_id)
-          draft_obj.temporary_approved_object = live_obj
-        end
-      end
-
-    private ###################################################################
-
-      def setup_draft_association_persistance_for_children_of(target_class, associations=nil)
-        associations = target_class.draft_target_associations unless associations
-        target_reflections = associations.map do |assoc|
-          reflection = target_class.reflect_on_association(assoc.to_sym)
-          reflection.presence || (raise DraftPunk::ConfigurationError, "#{name} includes invalid association (#{assoc})")
-        end
-        target_reflections.select{|r| is_relevant_association_type?(r) }.each do |assoc|
-          setup_amoeba_for assoc.klass
-        end
-      end
-
-      def setup_associations_and_scopes_for(target_class, set_default_scope: false, allow_previous_versions_to_be_changed: true)
-        target_class.send       :include, InstanceInterrogators unless target_class.method_defined?(:has_draft?)
-        target_class.send       :attr_accessor, :temporary_approved_object
-        target_class.send       :before_create, :before_create_draft if target_class.method_defined?(:before_create_draft)
-
-        target_class.const_set :ALLOW_PREVIOUS_VERSIONS_TO_BE_CHANGED, allow_previous_versions_to_be_changed.freeze
-        if target_class.tracks_approved_version_history?
-          target_class.belongs_to    :current_approved_version, class_name: target_class.name, optional: true
-          target_class.has_many      :previous_versions, -> { order(id: :desc) }, class_name: target_class.name, foreign_key: :current_approved_version_id
-          target_class.before_update :prevent_previous_versions_from_saving
-          target_class.send          :include, PreviousVersionInstanceMethods
-        end
-
-        if set_default_scope
-          target_class.default_scope -> { approved }
-        end
-
-        return if target_class.reflect_on_association(:approved_version) || !target_class.column_names.include?('approved_version_id')
-        target_class.send       :include, ActiveRecordInstanceMethods
-        target_class.send       :include, DraftDiffInstanceMethods
-        target_class.belongs_to :approved_version, class_name: target_class.name, optional: true
-        target_class.scope      :approved, -> { where("#{target_class.quoted_table_name}.approved_version_id IS NULL") }
-        target_class.has_one    :draft, -> { unscope(where: :approved) }, class_name: target_class.name, foreign_key: :approved_version_id
-        target_class.scope      :draft, -> { unscoped.where("#{target_class.quoted_table_name}.approved_version_id IS NOT NULL") }
-      end
-
-      def is_relevant_association_type?(activerecord_reflection)
-        activerecord_reflection.macro.in? Amoeba::Config::DEFAULTS[:known_macros]
-      end
-
-      def draft_punk_table_exists?(table_name)
-        if ActiveRecord::Base.connection.respond_to? :data_source_exists?
-          ActiveRecord::Base.connection.data_source_exists?(table_name)
-        else
-          ActiveRecord::Base.connection.table_exists?(table_name)
-        end
-      end
-    end
-
-  end
-end