Remove \1 and \2 that escape CSI before render

ruby · May 26, 2019 · 64dc218 · 64dc218
1 parent c6b7cad
commit 64dc218
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/lib/reline/line_editor.rb b/lib/reline/line_editor.rb
@@ -81,6 +81,7 @@ module CompletionState
   MenuInfo = Struct.new('MenuInfo', :target, :list)
 
   CSI_REGEXP = /\e\[(?:\d+;?)*[ABCDEFGHJKSTfminsuhl]/
+  NON_PRINTING_ESCAPES = "\1\2"
 
   def initialize(config)
     @config = config
@@ -164,6 +165,8 @@ def multiline_off
     lines = [String.new(encoding: @encoding)]
     height = 1
     width = 0
+    prompt = prompt.tr(NON_PRINTING_ESCAPES, '')
+    str = str.tr(NON_PRINTING_ESCAPES, '')
     rest = "#{prompt}#{str}".encode(Encoding::UTF_8)
     loop do
       break if rest.empty?
@@ -803,6 +806,7 @@ def finish
   private def calculate_width(str, allow_csi = false)
     if allow_csi
       str = str.gsub(CSI_REGEXP, '')
+      str = str.tr(NON_PRINTING_ESCAPES, '')
     end
     str.encode(Encoding::UTF_8).grapheme_clusters.inject(0) { |width, gc|
       width + Reline::Unicode.get_mbchar_width(gc)