eager_eye 0.2.2 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6329285c3bdd39b8c99bbf665ffef121ac0dc5530b26b8e125d71ca04239a7a4
-  data.tar.gz: 78a608b043a06a58b22a54553d0c0b4ee459c2ee921405f01c5a216a77e801fb
+  metadata.gz: 251f5ec97df15b5f85cbea7b1be3b7c95961f6ed133f4a69b721d76ebd7ad40e
+  data.tar.gz: ce06033abc65e06113af34789bf17a97badcfe8076675a45bf5ee59f5e19af79
 SHA512:
-  metadata.gz: 5d985342e6d3bc4a09dab284b153807b4cda7a7dce898c4dca9ca3e0223e6fb2f575b369426a2a04c30f4d04b557749be8091ee4ed23de4c4a86522d410867ec
-  data.tar.gz: 18ba07590dce273befec6295f0ce05702241ee46a727c05825dd573d44e3019839655f0da14dc9674f5ded7d56b7ee21d802b6264a2b2a85c5f6859928c887da
+  metadata.gz: 9f24578b9ec1a121b1698e3d53d78ec570b12b3f5092289ca42adc421c89cd1a94566e5de6ff45309803c0be2b8bafc76b1ea3e9f78599da32aef87503c51c38
+  data.tar.gz: 3f9dd3a3a3d4116840d8de2392e06afa78486ae049505f8706a7e5c9303d25e7efcf0ba7522b256bb9e504c1d8b709d93e06e92b1d0391cfd0e3bafdb7e00536

data/lib/eager_eye/detectors/custom_method_query.rb

@@ -21,6 +21,9 @@ module EagerEye
         maximum
       ].freeze
 
+      # Array-only methods that should not be flagged when collection is clearly an array
+      ARRAY_METHODS = %i[first last take].freeze
+
       ITERATION_METHODS = %i[each map select find_all reject collect detect find_index flat_map].freeze
 
       def self.detector_name
@@ -33,8 +36,9 @@ module EagerEye
         @issues = []
         @file_path = file_path
 
-        find_iteration_blocks(ast) do |block_body, block_var|
-          check_block_for_query_methods(block_body, block_var)
+        find_iteration_blocks(ast) do |block_body, block_var, collection|
+          is_array_collection = collection_is_array?(collection)
+          check_block_for_query_methods(block_body, block_var, is_array_collection)
         end
 
         @issues
@@ -48,7 +52,8 @@ module EagerEye
         if iteration_block?(node)
           block_var = extract_block_variable(node)
           block_body = extract_block_body(node)
-          yield(block_body, block_var) if block_var && block_body
+          collection = extract_collection(node)
+          yield(block_body, block_var, collection) if block_var && block_body
         end
 
         node.children.each do |child|
@@ -66,26 +71,40 @@ module EagerEye
         ITERATION_METHODS.include?(method_name)
       end
 
-      def check_block_for_query_methods(node, block_var)
+      def check_block_for_query_methods(node, block_var, is_array_collection = false) # rubocop:disable Style/OptionalBooleanParameter
         return unless node.is_a?(Parser::AST::Node)
 
-        add_issue(node) if query_chain_on_association?(node, block_var)
+        add_issue(node) if query_chain_on_association?(node, block_var, is_array_collection)
 
         node.children.each do |child|
-          check_block_for_query_methods(child, block_var)
+          check_block_for_query_methods(child, block_var, is_array_collection)
         end
       end
 
-      def query_chain_on_association?(node, block_var)
+      def query_chain_on_association?(node, block_var, is_array_collection = false) # rubocop:disable Style/OptionalBooleanParameter
         return false unless node.type == :send
 
         method_name = node.children[1]
         return false unless QUERY_METHODS.include?(method_name)
 
+        # Skip array-only methods when collection is clearly an array (.map result)
+        # AND the receiver is only the block variable (not chained)
+        if is_array_collection && ARRAY_METHODS.include?(method_name) &&
+           receiver_is_only_block_var?(node.children[0], block_var)
+          return false
+        end
+
         receiver = node.children[0]
         receiver_chain_starts_with?(receiver, block_var)
       end
 
+      def receiver_is_only_block_var?(node, block_var)
+        # Returns true only if receiver is EXACTLY the block variable, not a chain
+        node.is_a?(Parser::AST::Node) &&
+          node.type == :lvar &&
+          node.children[0] == block_var
+      end
+
       def receiver_chain_starts_with?(node, block_var)
         return false unless node.is_a?(Parser::AST::Node)
 
@@ -113,6 +132,31 @@ module EagerEye
         block_node.children[2]
       end
 
+      def extract_collection(block_node)
+        # Extract the collection being iterated on
+        # For: collection.each { |item| ... }
+        # Returns: the send node representing the collection method call
+        block_node.children[0]
+      end
+
+      def collection_is_array?(collection_node)
+        return false unless collection_node.is_a?(Parser::AST::Node)
+
+        case collection_node.type
+        when :array
+          # Literal array: [1, 2, 3].each { |item| ... }
+          true
+        when :send
+          # Only consider these methods as definitely returning arrays when iterating
+          method_name = collection_node.children[1]
+          # map, select, collect, etc. on anything return arrays for iteration
+          %i[map select collect flat_map to_a uniq compact].include?(method_name)
+        else
+          # Block variable itself won't tell us if it's an array
+          false
+        end
+      end
+
       def add_issue(node)
         method_name = node.children[1]
         association_chain = reconstruct_chain(node.children[0])

data/lib/eager_eye/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module EagerEye
-  VERSION = "0.2.2"
+  VERSION = "0.2.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: eager_eye
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.2.3
 platform: ruby
 authors:
 - hamzagedikkaya