From d7e3310e5f6321f78f25d79e80603c182d5292b9 Mon Sep 17 00:00:00 2001
From: Andrew Kane <andrew@chartkick.com>
Date: Wed, 18 Mar 2020 05:46:30 -0700
Subject: [PATCH] Better model handling [skip ci]

---
 lib/searchkick/relation.rb | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/lib/searchkick/relation.rb b/lib/searchkick/relation.rb
index 6f2d63b..43aa170 100644
--- a/lib/searchkick/relation.rb
+++ b/lib/searchkick/relation.rb
@@ -222,11 +222,17 @@ module Searchkick
     end
 
     def models(*args)
+      raise ArgumentError, "Use Searchkick.search to search multiple models" if klass
       spawn.models!(*args)
     end
 
     def models!(*args)
-      options[:models] = Array(options[:models]) + args
+      # make Searchkick.search.models(Product) and Product.search equivalent
+      if !klass && args.size == 1 && !options[:models]
+        @klass = args.first
+      else
+        options[:models] = Array(options[:models]) + args
+      end
       self
     end
 
--
libgit2 0.21.0