is_paranoid_ext 0.0.1 → 0.0.2

data/is_paranoid_ext.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name = %q{is_paranoid_ext}
-  s.version = "0.0.1"
+  s.version = "0.0.2"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Alex Neill"]

data/lib/is_paranoid_ext.rb

@@ -15,10 +15,29 @@ module IsParanoidExt
   private
   
   def merge_paranoid_options!(association_id, options)
-    if (klass = association_id.to_s.classify.constantize) && klass.respond_to?(:restore)
-      options.merge!(:conditions => ["#{klass.quoted_table_name}.#{klass.destroyed_field} IS ?", klass.field_not_destroyed])
+    through_id = options[:through]
+    @paranoid_conditions, @paranoid_values = nil, nil
+    apply_paranoid_conditions_for(options[:class_name] || association_id)
+    apply_paranoid_conditions_for through_id if through_id
+    options.merge!(:conditions => merged_paranoid_conditions) unless @paranoid_conditions.nil?
+  end
+  
+  def apply_paranoid_conditions_for(association_id)
+    reflection = reflections[association_id.to_sym]
+    association_klass = ( reflection.try(:klass) || association_id.to_s.classify.constantize )
+    if association_klass && association_klass.respond_to?(:restore)
+      @paranoid_conditions ||= []
+      @paranoid_conditions << "#{association_klass.quoted_table_name}.#{association_klass.destroyed_field} IS ?"
+      @paranoid_values ||= []
+      @paranoid_values << association_klass.field_not_destroyed
     end
   end
+  
+  def merged_paranoid_conditions
+    @merged_paranoid_conditions ||= if @paranoid_conditions.any? && @paranoid_values.any?
+      [@paranoid_conditions.map{|c|"(#{c})"}.join(' AND ')] + @paranoid_values
+    else nil end
+  end
 end
 
 ActiveRecord::Base.send(:extend, IsParanoidExt)

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 0
-  - 1
-  version: 0.0.1
+  - 2
+  version: 0.0.2
 platform: ruby
 authors: 
 - Alex Neill