Allow IfMod and UnlessMod to know about heredocs moving their bounds.

Author: kddnewton

CHANGELOG.md

@@ -43,6 +43,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/) a
 - Only unescape forward slashes in regular expressions if converting from slash bounds to `%r` bounds.
 - Allow `When` nodes to grab trailing comments away from their statements lists.
 - Allow flip-flop operators to be formatted correctly within `IfMod` and `UnlessMod` nodes.
+- Allow `IfMod` and `UnlessMod` to know about heredocs moving their bounds.
 
 ### Removed
 

lib/syntax_tree.rb

@@ -6973,7 +6973,13 @@ def on_if_mod(predicate, statement)
     IfMod.new(
       statement: statement,
       predicate: predicate,
-      location: statement.location.to(predicate.location)
+      location:
+        Location.new(
+          start_line: statement.location.start_line,
+          start_char: statement.location.start_char,
+          end_line: [statement.location.end_line, predicate.location.end_line].max,
+          end_char: predicate.location.end_char
+        )
     )
   end
 
@@ -12127,7 +12133,13 @@ def on_unless_mod(predicate, statement)
     UnlessMod.new(
       statement: statement,
       predicate: predicate,
-      location: statement.location.to(predicate.location)
+      location:
+        Location.new(
+          start_line: statement.location.start_line,
+          start_char: statement.location.start_char,
+          end_line: [statement.location.end_line, predicate.location.end_line].max,
+          end_char: predicate.location.end_char
+        )
     )
   end