From 8fa18435234335afbf17bda88324a620e15a4323 Mon Sep 17 00:00:00 2001
From: Gopal Patel <nixme@stillhope.com>
Date: Tue, 9 Jan 2024 21:13:46 -0800
Subject: [PATCH] [ruby/prism] Relax Location#source to be optional

https://github.com/ruby/prism/commit/9f00fe7510
---
 lib/prism/parse_result.rb            | 31 ++++++++++++++++------------
 prism/templates/lib/prism/dsl.rb.erb |  1 -
 2 files changed, 18 insertions(+), 14 deletions(-)

diff --git a/lib/prism/parse_result.rb b/lib/prism/parse_result.rb
index c55b7b6685..c00059fbfe 100644
--- a/lib/prism/parse_result.rb
+++ b/lib/prism/parse_result.rb
@@ -168,13 +168,13 @@ module Prism
 
     # The source code that this location represents.
     def slice
-      source.slice(start_offset, length)
+      source!.slice(start_offset, length)
     end
 
     # The character offset from the beginning of the source where this location
     # starts.
     def start_character_offset
-      source.character_offset(start_offset)
+      source!.character_offset(start_offset)
     end
 
     # The offset from the start of the file in code units of the given encoding.
@@ -190,7 +190,7 @@ module Prism
     # The character offset from the beginning of the source where this location
     # ends.
     def end_character_offset
-      source.character_offset(end_offset)
+      source!.character_offset(end_offset)
     end
 
     # The offset from the start of the file in code units of the given encoding.
@@ -200,30 +200,30 @@ module Prism
 
     # The line number where this location starts.
     def start_line
-      source.line(start_offset)
+      source!.line(start_offset)
     end
 
     # The content of the line where this location starts before this location.
     def start_line_slice
-      offset = source.line_start(start_offset)
-      source.slice(offset, start_offset - offset)
+      offset = source!.line_start(start_offset)
+      source!.slice(offset, start_offset - offset)
     end
 
     # The line number where this location ends.
     def end_line
-      source.line(end_offset)
+      source!.line(end_offset)
     end
 
     # The column number in bytes where this location starts from the start of
     # the line.
     def start_column
-      source.column(start_offset)
+      source!.column(start_offset)
     end
 
     # The column number in characters where this location ends from the start of
     # the line.
     def start_character_column
-      source.character_column(start_offset)
+      source!.character_column(start_offset)
     end
 
     # The column number in code units of the given encoding where this location
@@ -235,13 +235,13 @@ module Prism
     # The column number in bytes where this location ends from the start of the
     # line.
     def end_column
-      source.column(end_offset)
+      source!.column(end_offset)
     end
 
     # The column number in characters where this location ends from the start of
     # the line.
     def end_character_column
-      source.character_column(end_offset)
+      source!.character_column(end_offset)
     end
 
     # The column number in code units of the given encoding where this location
@@ -281,8 +281,13 @@ module Prism
     # the beginning of the file. Useful for when you want a location object but
     # do not care where it points.
     def self.null
-      source = nil #: Source
-      new(source, 0, 0)
+      new(nil, 0, 0)
+    end
+
+    private
+
+    def source!
+      source or raise "Missing source"
     end
   end
 
diff --git a/prism/templates/lib/prism/dsl.rb.erb b/prism/templates/lib/prism/dsl.rb.erb
index 4c139ca281..7c55fb10bc 100644
--- a/prism/templates/lib/prism/dsl.rb.erb
+++ b/prism/templates/lib/prism/dsl.rb.erb
@@ -36,7 +36,6 @@ module Prism
 
     # Create a new Location object
     def Location(source = nil, start_offset = 0, length = 0)
-      # @type var source: Source
       Location.new(source, start_offset, length)
     end
     <%- nodes.each do |node| -%>