Add code_units_cache helper

Shopify · Oct 10, 2024 · 86925ea · 86925ea
1 parent 7b95a3f
commit 86925ea
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/lib/ruby_lsp/erb_document.rb b/lib/ruby_lsp/erb_document.rb
@@ -64,6 +64,16 @@ def inside_host_language?(char_position)
       char && char != " "
     end
 
+    sig do
+      returns(T.any(
+        T.proc.params(arg0: Integer).returns(Integer),
+        Prism::CodeUnitsCache,
+      ))
+    end
+    def code_units_cache
+      @parse_result.code_units_cache(@encoding)
+    end
+
     class ERBScanner
       extend T::Sig
 

diff --git a/lib/ruby_lsp/ruby_document.rb b/lib/ruby_lsp/ruby_document.rb
@@ -218,5 +218,15 @@ def locate_node(position, node_types: [])
         encoding: @encoding,
       )
     end
+
+    sig do
+      returns(T.any(
+        T.proc.params(arg0: Integer).returns(Integer),
+        Prism::CodeUnitsCache,
+      ))
+    end
+    def code_units_cache
+      @parse_result.code_units_cache(@encoding)
+    end
   end
 end