Properly report performance rules in macros

closes #102

spec/ameba/rule/performance/any_after_filter_spec.cr

@@ -48,6 +48,15 @@ module Ameba::Rule::Performance
       end
     end
 
+    context "macro" do
+      it "reports in macro scope" do
+        source = Source.new %(
+          {{ [1, 2, 3].reject { |e| e > 2  }.any? }}
+        )
+        subject.catch(source).should_not be_valid
+      end
+    end
+
     it "reports rule, pos and message" do
       s = Source.new %(
         [1, 2, 3].reject { |e| e > 2 }.any?

spec/ameba/rule/performance/first_last_after_filter_spec.cr

@@ -107,6 +107,15 @@ module Ameba::Rule::Performance
       issue.message.should eq "Use `reverse_each.find {...}` instead of `select {...}.last`"
     end
 
+    context "macro" do
+      it "doesn't report in macro scope" do
+        source = Source.new %(
+          {{[1, 2, 3].select { |e| e > 2  }.last }}
+        )
+        subject.catch(source).should be_valid
+      end
+    end
+
     it "reports a correct message for last?" do
       s = Source.new %(
         [1, 2, 3].select { |e| e > 2 }.last?

spec/ameba/rule/performance/size_after_filter_spec.cr

@@ -49,6 +49,15 @@ module Ameba::Rule::Performance
       end
     end
 
+    context "macro" do
+      it "doesn't report in macro scope" do
+        source = Source.new %(
+          {{[1, 2, 3].select { |v| v > 1 }.size}}
+        )
+        subject.catch(source).should be_valid
+      end
+    end
+
     it "reports rule, pos and message" do
       s = Source.new %(
         lines.split("\n").reject(&.empty?).size

src/ameba/ast/visitors/node_visitor.cr

@@ -34,6 +34,13 @@ module Ameba::AST
   # ```
   #
   class NodeVisitor < BaseVisitor
+    @skip : Array(Crystal::ASTNode.class)?
+
+    def initialize(@rule, @source, skip = nil)
+      @skip = skip.try &.map { |el| el.as(Crystal::ASTNode.class) }
+      super @rule, @source
+    end
+
     {% for name in NODES %}
       # A visit callback for `Crystal::{{name}}` node.
       # Returns true meaning that child nodes will be traversed as well.
@@ -42,5 +49,10 @@ module Ameba::AST
         true
       end
     {% end %}
+
+    def visit(node)
+      return true unless (skip = @skip)
+      !skip.includes?(node.class)
+    end
   end
 end

src/ameba/rule/performance/first_last_after_filter.cr

@@ -35,7 +35,12 @@ module Ameba::Rule::Performance
     end
 
     def test(source)
-      AST::NodeVisitor.new self, source
+      AST::NodeVisitor.new self, source, skip: [
+        Crystal::Macro,
+        Crystal::MacroExpression,
+        Crystal::MacroIf,
+        Crystal::MacroFor,
+      ]
     end
 
     def test(source, node : Crystal::Call)

src/ameba/rule/performance/size_after_filter.cr

@@ -41,7 +41,12 @@ module Ameba::Rule::Performance
     end
 
     def test(source)
-      AST::NodeVisitor.new self, source
+      AST::NodeVisitor.new self, source, skip: [
+        Crystal::Macro,
+        Crystal::MacroExpression,
+        Crystal::MacroIf,
+        Crystal::MacroFor,
+      ]
     end
 
     def test(source, node : Crystal::Call)