Allow passing collections to scope multi search

Author: ellnix

lib/meilisearch/rails/multi_search.rb

@@ -18,7 +18,7 @@ def multi_search(searches)
 
       def normalize(options, index_target)
         options
-          .except(:class_name)
+          .except(:class_name, :collection)
           .merge!(index_uid: index_uid_from_target(index_target))
       end
 

lib/meilisearch/rails/multi_search/result.rb

@@ -14,7 +14,7 @@ def initialize(searches, raw_results)
                             target
                           end
 
-          @results[target] = results_class ? load_results(results_class, result) : result['hits']
+          @results[target] = results_class ? load_results(results_class, result, collection: search_options[:collection]) : result['hits']
 
           @metadata[target] = result.except('hits')
         end
@@ -69,7 +69,7 @@ def to_h
 
       private
 
-      def load_results(klass, result)
+      def load_results(klass, result, collection: klass)
         pk_method = klass.ms_primary_key_method
         pk_method = pk_method.in if Utilities.mongo_model?(klass)
 
@@ -78,7 +78,7 @@ def load_results(klass, result)
         hits_by_id =
           result['hits'].index_by { |hit| hit[condition_key.to_s] }
 
-        records = klass.where(condition_key => hits_by_id.keys)
+        records = collection.where(condition_key => hits_by_id.keys)
 
         if records.respond_to? :in_order_of
           records.in_order_of(condition_key, hits_by_id.keys).each do |record|

spec/multi_search_spec.rb

@@ -180,4 +180,17 @@ def reset_indexes
       MeiliSearch::Rails.configuration[:pagination_backend] = nil
     end
   end
+
+  context 'with collections' do
+    it 'fetches items from the given collection' do
+      results = MeiliSearch::Rails.multi_search(
+        Product.index.uid => { q: 'palm', class_name: 'Product', collection: Product.where('tags LIKE "%terrible%"') },
+        Color => { q: 'bl', collection: Color.where(short_name: 'bla') }
+      )
+
+      expect(results).to contain_exactly(
+        black, palm_pixi_plus
+      )
+    end
+  end
 end