From ea4b94a7bf707d27af44701fa44ef5f916b84e01 Mon Sep 17 00:00:00 2001
From: Tomas D'Stefano <tomasdestefi@gmail.com>
Date: Tue, 31 May 2011 11:43:19 -0300
Subject: Destroy association habtm record before destroying the record itself.
 Fixes issue #402.

---
 .../builder/has_and_belongs_to_many.rb             | 22 ++++++++++------------
 activerecord/lib/active_record/persistence.rb      |  7 +++++++
 2 files changed, 17 insertions(+), 12 deletions(-)

(limited to 'activerecord/lib')

diff --git a/activerecord/lib/active_record/associations/builder/has_and_belongs_to_many.rb b/activerecord/lib/active_record/associations/builder/has_and_belongs_to_many.rb
index d7632b2ea9..30fc44b4c2 100644
--- a/activerecord/lib/active_record/associations/builder/has_and_belongs_to_many.rb
+++ b/activerecord/lib/active_record/associations/builder/has_and_belongs_to_many.rb
@@ -7,24 +7,22 @@ module ActiveRecord::Associations::Builder
     def build
       reflection = super
       check_validity(reflection)
-      define_after_destroy_method
+      define_destroy_hook
       reflection
     end
 
     private
 
-      def define_after_destroy_method
+      def define_destroy_hook
         name = self.name
-        model.send(:class_eval, <<-eoruby, __FILE__, __LINE__ + 1)
-          def #{after_destroy_method_name}
-            association(#{name.to_sym.inspect}).delete_all
-          end
-        eoruby
-        model.after_destroy after_destroy_method_name
-      end
-
-      def after_destroy_method_name
-        "has_and_belongs_to_many_after_destroy_for_#{name}"
+        model.send(:include, Module.new {
+          class_eval <<-RUBY, __FILE__, __LINE__ + 1
+            def destroy_associations
+              association(#{name.to_sym.inspect}).delete_all
+              super
+            end
+          RUBY
+        })
       end
 
       # TODO: These checks should probably be moved into the Reflection, and we should not be
diff --git a/activerecord/lib/active_record/persistence.rb b/activerecord/lib/active_record/persistence.rb
index f425118f59..d9cd7987b0 100644
--- a/activerecord/lib/active_record/persistence.rb
+++ b/activerecord/lib/active_record/persistence.rb
@@ -79,6 +79,8 @@ module ActiveRecord
     # Deletes the record in the database and freezes this instance to reflect
     # that no changes should be made (since they can't be persisted).
     def destroy
+      destroy_associations
+
       if persisted?
         IdentityMap.remove(self) if IdentityMap.enabled?
         pk         = self.class.primary_key
@@ -284,6 +286,11 @@ module ActiveRecord
     end
 
   private
+
+    # A hook to be overriden by association modules.
+    def destroy_associations
+    end
+
     def create_or_update
       raise ReadOnlyRecord if readonly?
       result = new_record? ? create : update
-- 
cgit v1.2.3