require 'activerecord'

module IsParanoidExt
  
  private

    def construct_finder_sql_with_included_associations(options, join_dependency)
      preload_options = options.dup
      join_dependency.reflections.each do |reflection|
        preload_options = paranoid_preload_options(reflection, preload_options)
      end
      super(preload_options, join_dependency)
    end
    
    def preload_has_one_association(records, reflection, preload_options = {})
      super(records, reflection, paranoid_preload_options(reflection, preload_options))
    end
    
    def preload_has_many_association(records, reflection, preload_options = {})
      super(records, reflection, paranoid_preload_options(reflection, preload_options))
    end
    
    def preload_has_and_belongs_to_many_association(records, reflection, preload_options = {})
      super(records, reflection, paranoid_preload_options(reflection, preload_options))
    end
    
    def paranoid_preload_options(reflection, preload_options)
      paranoid_preload_opts = preload_options.dup
      if reflection.klass.respond_to?(:restore)
        sql_array  = ["#{reflection.klass.quoted_table_name}.#{reflection.klass.destroyed_field} IS ?"]
        sql_array << reflection.klass.field_not_destroyed
        preload_conditions = paranoid_preload_opts.delete(:conditions) || []
        paranoid_preload_opts.merge!(:conditions => merge_conditions(sql_array, preload_conditions))
      end
      paranoid_preload_opts
    end

end

ActiveRecord::Base.send(:extend, IsParanoidExt)