where_exists 2.0.1 → 2.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 70bb8d0cc4affb99051fe748c883841abd8635f78faab07572208dd495ae9e0e
-  data.tar.gz: 56be70dc7847f0beddf4a19284485e8e50c38709e2b0db4f64a0250c967b0a33
+  metadata.gz: d9957b5787ab2fda2505560b01ffeca8f2e1d24bf0d0d38e09f4f25969a30ab6
+  data.tar.gz: 4718917be7477f54afb0e322a588adadb4d2d375c6b7938d922b89ef0629ea27
 SHA512:
-  metadata.gz: e86824a1c85aa3a4006bc20035444fe36d1895c7cb17fd3019e5da5431753ba45fd735bbd41ae903f807bc7e45e36d415c887390d2fab8805d44e8afdcd37f2c
-  data.tar.gz: 5a4814a20b678da5ee28b9cdbccd99ef653f4ea6284fd42f5203d0074301d35b3627524e00e89623000595fdb7f9c110873224f163edc2b0a112ae0deeba18b9
+  metadata.gz: 839de9969500c67df98512ee5f5043b4d34a8e337263e3dcc79dd081e6730a0b1bb6bce3c632a764372d6dabe143f9b3c7ffcbc55092f55727d8746d40eafaf3
+  data.tar.gz: a4c38490f6caf945889d2f92075e1069cef92706524c299bd4d800c4b856fd62fda39738539fee816daa955a5ef3908e49c7aed9ed4e3be97af6bf8c419f3b07

data/lib/where_exists/version.rb

@@ -1,3 +1,3 @@
 module WhereExists
-  VERSION = "2.0.1"
+  VERSION = "2.0.2"
 end

data/lib/where_exists.rb

@@ -36,11 +36,11 @@ module WhereExists
 
     case association.macro
     when :belongs_to
-      queries = where_exists_for_belongs_to_query(association, where_parameters)
+      queries = where_exists_for_belongs_to_query(association, where_parameters, &block)
     when :has_many, :has_one
-      queries = where_exists_for_has_many_query(association, where_parameters)
+      queries = where_exists_for_has_many_query(association, where_parameters, &block)
     when :has_and_belongs_to_many
-      queries = where_exists_for_habtm_query(association, where_parameters)
+      queries = where_exists_for_habtm_query(association, where_parameters, &block)
     else
       inspection = nil
       begin
@@ -53,13 +53,12 @@ module WhereExists
 
     queries_sql =
       queries.map do |query|
-        query = yield query if block_given?
         "EXISTS (" + query.to_sql + ")"
       end
     queries_sql.join(" OR ")
   end
 
-  def where_exists_for_belongs_to_query(association, where_parameters)
+  def where_exists_for_belongs_to_query(association, where_parameters, &block)
     polymorphic = association.options[:polymorphic].present?
 
     association_scope = association.scope
@@ -93,13 +92,14 @@ module WhereExists
 
         query = query.where("#{self_type} IN (?)", other_types.uniq)
       end
+      query = yield query if block_given?
       queries.push query
     end
 
     queries
   end
 
-  def where_exists_for_has_many_query(association, where_parameters, next_association = {})
+  def where_exists_for_has_many_query(association, where_parameters, next_association = {}, &block)
     if association.through_reflection
       raise ArgumentError.new(association) unless association.source_reflection
       next_association = {
@@ -111,9 +111,9 @@ module WhereExists
 
       case association.macro
       when :has_many, :has_one
-        return where_exists_for_has_many_query(association, {}, next_association)
+        return where_exists_for_has_many_query(association, {}, next_association, &block)
       when :has_and_belongs_to_many
-        return where_exists_for_habtm_query(association, {}, next_association)
+        return where_exists_for_habtm_query(association, {}, next_association, &block)
       else
         inspection = nil
         begin
@@ -148,17 +148,18 @@ module WhereExists
     end
 
     if next_association[:association]
-      return loop_nested_association(result, next_association)
+      return loop_nested_association(result, next_association, &block)
     end
 
     if where_parameters != []
       result = result.where(*where_parameters)
     end
 
+    result = yield result if block_given?
     [result]
   end
 
-  def where_exists_for_habtm_query(association, where_parameters, next_association = {})
+  def where_exists_for_habtm_query(association, where_parameters, next_association = {}, &block)
     association_scope = association.scope
 
     associated_model = association.klass
@@ -182,7 +183,7 @@ module WhereExists
       where("#{join_ids} = #{self_ids}")
 
     if next_association[:association]
-      return loop_nested_association(result, next_association)
+      return loop_nested_association(result, next_association, &block)
     end
 
     if where_parameters != []
@@ -193,15 +194,18 @@ module WhereExists
       result = result.instance_exec(&association_scope)
     end
 
+    result = yield result if block_given?
+
     [result]
   end
 
-  def loop_nested_association(query, next_association = {}, nested = false)
+  def loop_nested_association(query, next_association = {}, nested = false, &block)
     str = query.klass.build_exists_string(
       next_association[:association].name,
       *[
         *next_association[:params]
       ],
+      &block
     )
 
     if next_association[:next_association] && next_association[:next_association][:association]
@@ -210,7 +214,8 @@ module WhereExists
         "(#{subq} AND (#{loop_nested_association(
           next_association[:association],
           next_association[:next_association],
-          true
+          true,
+          &block
         )}))"
       end
     end

data/test/has_many_through_test.rb

@@ -52,8 +52,6 @@ class Blob < ActiveRecord::Base
   end
 end
 
-
-
 class Project < ActiveRecord::Base
   has_many :tasks
   has_many :invoices, :through => :tasks
@@ -99,8 +97,6 @@ class HasManyThroughTest < Minitest::Test
   end
 
   def test_one_level_through
-    ActiveRecord::Base.descendants.each(&:delete_all)
-
     project = Project.create!
     irrelevant_project = Project.create!
 
@@ -121,8 +117,6 @@ class HasManyThroughTest < Minitest::Test
   end
 
   def test_deep_through
-    ActiveRecord::Base.descendants.each(&:delete_all)
-
     project = Project.create! name: 'relevant'
     irrelevant_project = Project.create! name: 'irrelevant'
 
@@ -188,6 +182,14 @@ class HasManyThroughTest < Minitest::Test
     result = Project.where_not_exists(:blobs)
 
     assert_equal 0, result.length
+  end
 
+  def test_with_yield
+    project = Project.create! name: 'example_project'
+    task = Task.create!(project: project)
+    line_item = LineItem.create!(name: 'example_line_item', task: task)
+    result = Project.where_exists(:project_line_items) { |scope| scope.where(name: 'example_line_item') }
+
+    assert_equal 1, result.length
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: where_exists
 version: !ruby/object:Gem::Version
-  version: 2.0.1
+  version: 2.0.2
 platform: ruby
 authors:
 - Eugene Zolotarev
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-07-11 00:00:00.000000000 Z
+date: 2022-10-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -114,7 +114,6 @@ files:
 - lib/where_exists/version.rb
 - test/belongs_to_polymorphic_test.rb
 - test/belongs_to_test.rb
-- test/db/test.db
 - test/documentation_test.rb
 - test/has_and_belongs_to_many.rb
 - test/has_many_polymorphic_test.rb
@@ -149,7 +148,6 @@ test_files:
 - test/belongs_to_polymorphic_test.rb
 - test/belongs_to_test.rb
 - test/has_many_polymorphic_test.rb
-- test/db/test.db
 - test/has_many_through_test.rb
 - test/test_helper.rb
 - test/has_and_belongs_to_many.rb