multi_search: Use collection to infer class & index

Author: ellnix

lib/meilisearch/rails/multi_search.rb

@@ -5,7 +5,8 @@ module Rails
     class << self
       def multi_search(searches)
         search_parameters = searches.map do |(index_target, options)|
-          index_target = options.delete(:index_uid) || index_target
+          collection_class = options[:collection].respond_to?(:model) ? options[:collection].model : options[:collection]
+          index_target = options.delete(:index_uid) || collection_class || index_target
 
           paginate(options) if pagination_enabled?
           normalize(options, index_target)

lib/meilisearch/rails/multi_search/result.rb

@@ -12,6 +12,8 @@ def initialize(searches, raw_results)
                             search_options[:class_name].constantize
                           elsif target.instance_of?(Class)
                             target
+                          elsif search_options[:collection]
+                            search_options[:collection]
                           end
 
           @results[target] = results_class ? load_results(results_class, result, collection: search_options[:collection]) : result['hits']
@@ -69,7 +71,9 @@ def to_h
 
       private
 
-      def load_results(klass, result, collection: klass)
+      def load_results(klass, result, collection:)
+        collection ||= klass
+
         pk_method = klass.ms_primary_key_method
         pk_method = pk_method.in if Utilities.mongo_model?(klass)
 

spec/multi_search_spec.rb

@@ -192,5 +192,21 @@ def reset_indexes
         black, palm_pixi_plus
       )
     end
+
+    it 'infers the model' do
+      results = MeiliSearch::Rails.multi_search(
+        'colors' => { q: 'bl', collection: Color.all, index_uid: Color.index.uid }
+      )
+
+      expect(results.to_h['colors']).to contain_exactly(blue, black)
+    end
+
+    it 'infers the index as well as the model' do
+      results = MeiliSearch::Rails.multi_search(
+        'colors' => { q: 'bl', collection: Color }
+      )
+
+      expect(results.to_h['colors']).to contain_exactly(blue, black)
+    end
   end
 end