fix: progress bars in virtual text (#77)

benlubas · Dec 10, 2023 · 3b886c1 · 3b886c1
1 parent 6b2660d
commit 3b886c1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/rplugin/python3/molten/outputbuffer.py b/rplugin/python3/molten/outputbuffer.py
@@ -366,7 +366,6 @@ def handle_progress_bars(line_str: str) -> List[str]:
         last = parts[-1]
         if last != "":
             actual_lines.append(last)
-            lines = actual_lines
 
     return actual_lines
 

diff --git a/rplugin/python3/molten/outputchunks.py b/rplugin/python3/molten/outputchunks.py
@@ -75,14 +75,17 @@ def place(
             if hard_wrap:
                 lines = []
                 splits = []
+                # Assume this is a progress bar, or similar, we shouldn't try to wrap it
+                if text.find("\r") != -1:
+                    return text, 0
                 for line in text.split("\n"):
                     index = 0
                     if len(line) + col > win_width:
                         splits.append(line[: win_width - col])
                         line = line[win_width - col :]
 
                     for _ in range(len(line) // win_width):
-                        splits.append(line[index * win_width : (index+1)*win_width])
+                        splits.append(line[index * win_width : (index + 1) * win_width])
                         index += 1
                     else:
                         splits.append(line[index * win_width :])