[ruby/prism] Prism::CallNode#full_message_loc

https://github.com/ruby/prism/commit/fa6fe9be84

1 files changed, 15 insertions, 0 deletions

diff --git a/lib/prism/node_ext.rb b/lib/prism/node_ext.rb
index d4c9b21079..13830dad3d 100644
--- a/lib/prism/node_ext.rb
+++ b/lib/prism/node_ext.rb
@@ -286,4 +286,19 @@ module Prism
       names
     end
   end
+
+  class CallNode < Node
+    # When a call node has the attribute_write flag set, it means that the call
+    # is using the attribute write syntax. This is either a method call to []=
+    # or a method call to a method that ends with =. Either way, the = sign is
+    # present in the source.
+    #
+    # Prism returns the message_loc _without_ the = sign attached, because there
+    # can be any amount of space between the message and the = sign. However,
+    # sometimes you want the location of the full message including the inner
+    # space and the = sign. This method provides that.
+    def full_message_loc
+      attribute_write? ? message_loc&.adjoin("=") : message_loc
+    end
+  end
 end