Use RubyVM.script_lines if available.

Author: ko1

lib/debug/source_repository.rb

@@ -4,72 +4,107 @@
 
 module DEBUGGER__
   class SourceRepository
-    SrcInfo = Struct.new(:src, :colored)
+    include Color
 
-    def initialize
-      @files = {} # filename => SrcInfo
-    end
+    if RubyVM.respond_to? :keep_script_lines
+      # Ruby 3.1 and later
+      RubyVM.keep_script_lines = true
+      require 'objspace'
 
-    def add iseq, src
-      if (path = iseq.absolute_path) && File.exist?(path)
-        add_path path
-      elsif src
-        add_iseq iseq, src
+      def initialize
+        # cache
+        @cmap = ObjectSpace::WeakMap.new
       end
-    end
 
-    def all_iseq iseq, rs = []
-      rs << iseq
-      iseq.each_child{|ci|
-        all_iseq(ci, rs)
-      }
-      rs
-    end
+      def add iseq, src
+        # do nothing
+      end
+
+      def get iseq
+        if lines = iseq.script_lines.map(&:chomp)
+          lines.join("\n")
+        else
+          nil
+        end
+      end
 
-    private def add_iseq iseq, src
-      line = iseq.first_line
-      if line > 1
-        src = ("\n" * (line - 1)) + src
+      def get_colored iseq
+        if lines = @cmap[iseq]
+          lines
+        else
+          if src = get(iseq)
+            @cmap[iseq] = colorize_code(src).lines
+          else
+            nil
+          end
+        end
       end
-      si = SrcInfo.new(src.lines)
+    else
+      # ruby 3.0 or erlier
+      SrcInfo = Struct.new(:src, :colored)
 
-      all_iseq(iseq).each{|e|
-        e.instance_variable_set(:@debugger_si, si)
-        e.freeze
-      }
-    end
+      def initialize
+        @files = {} # filename => SrcInfo
+      end
 
-    private def add_path path
-      src = File.read(path)
-      src = src.gsub("\r\n", "\n") # CRLF -> LF
-      @files[path] = SrcInfo.new(src.lines)
-    rescue SystemCallError
-    end
+      def add iseq, src
+        if (path = iseq.absolute_path) && File.exist?(path)
+          add_path path
+        elsif src
+          add_iseq iseq, src
+        end
+      end
 
-    private def get_si iseq
-      return unless iseq
+      private def all_iseq iseq, rs = []
+        rs << iseq
+        iseq.each_child{|ci|
+          all_iseq(ci, rs)
+        }
+        rs
+      end
 
-      if iseq.instance_variable_defined?(:@debugger_si)
-        iseq.instance_variable_get(:@debugger_si)
-      elsif @files.has_key?(path = iseq.absolute_path)
-        @files[path]
-      elsif path
-        add_path(path)
+      private def add_iseq iseq, src
+        line = iseq.first_line
+        if line > 1
+          src = ("\n" * (line - 1)) + src
+        end
+        si = SrcInfo.new(src.lines)
+        all_iseq(iseq).each{|e|
+          e.instance_variable_set(:@debugger_si, si)
+          e.freeze
+        }
       end
-    end
 
-    def get iseq
-      if si = get_si(iseq)
-        si.src
+      private def add_path path
+        src = File.read(path)
+        src = src.gsub("\r\n", "\n") # CRLF -> LF
+        @files[path] = SrcInfo.new(src.lines)
+      rescue SystemCallError
       end
-    end
 
-    include Color
+      private def get_si iseq
+        return unless iseq
+      
+        if iseq.instance_variable_defined?(:@debugger_si)
+          iseq.instance_variable_get(:@debugger_si)
+        elsif @files.has_key?(path = iseq.absolute_path)
+          @files[path]
+        elsif path
+          add_path(path)
+        end
+      end
+
+      def get iseq
+        if si = get_si(iseq)
+          si.src
+        end
+      end
 
-    def get_colored iseq
-      if si = get_si(iseq)
-        si.colored || begin
-          si.colored = colorize_code(si.src.join).lines
+      def get_colored iseq
+        if si = get_si(iseq)
+          si.colored || begin
+            si.colored = colorize_code(si.src.join).lines
+          end
         end
       end
     end