[Fix rubocop#12221] Fix a false positive for `Layout/EmptyLineAfterGu…

…ardClause`

Fixes rubocop#12221.

This PR fixes a false positive for `Layout/EmptyLineAfterGuardClause`
when using `return` before guard condition with heredoc.

changelog/fix_a_false_positive_for_layout_empty_line_after_guard_clause.md

@@ -0,0 +1 @@
+* [#12221](https://github.com/rubocop/rubocop/issues/12221): Fix a false positive for `Layout/EmptyLineAfterGuardClause` when using `return` before guard condition with heredoc. ([@koic][])

lib/rubocop/cop/layout/empty_line_after_guard_clause.rb

@@ -164,6 +164,8 @@ def last_heredoc_argument_node(node)
 
           if node.if_branch.and_type?
             node.if_branch.children.first
+          elsif use_heredoc_in_condition?(node.condition)
+            node.condition
           else
             node.if_branch.children.last
           end
@@ -180,6 +182,12 @@ def heredoc?(node)
           node.respond_to?(:heredoc?) && node.heredoc?
         end
 
+        def use_heredoc_in_condition?(condition)
+          condition.descendants.any? do |descendant|
+            descendant.respond_to?(:heredoc?) && descendant.heredoc?
+          end
+        end
+
         def offense_location(node)
           if node.loc.respond_to?(:end) && node.loc.end
             node.loc.end

spec/rubocop/cop/layout/empty_line_after_guard_clause_spec.rb

@@ -482,6 +482,30 @@ def foo
     RUBY
   end
 
+  it 'does not register an offense and corrects when using `return` before guard condition with heredoc' do
+    expect_no_offenses(<<~RUBY)
+      def foo
+        return true if <<~TEXT.length > bar
+          hi
+        TEXT
+
+        false
+      end
+    RUBY
+  end
+
+  it 'does not register an offense and corrects when using `raise` before guard condition with heredoc' do
+    expect_no_offenses(<<~RUBY)
+      def foo
+        raise if <<~TEXT.length > bar
+          hi
+        TEXT
+
+        baz
+      end
+    RUBY
+  end
+
   it 'registers an offense and corrects a guard clause that is a ternary operator' do
     expect_offense(<<~RUBY)
       def foo