fix issue #349 (replace all bug)

Author: SilenZcience

babi/file.py

@@ -185,15 +185,21 @@ def __init__(
         self.wrapped = False
         self._start_x = file.buf.x + offset
         self._start_y = file.buf.y
+        self.original_line_change_diff = 0
 
     def __iter__(self) -> _SearchIter:
         return self
 
+    def calc_past_original_offset(self, y: int, match: Match[str], replacement: str):
+        if not self.wrapped or y != self._start_y:
+            return
+        self.original_line_change_diff += len(replacement) - (match.end()-match.start())
+
     def _stop_if_past_original(self, y: int, match: Match[str]) -> Found:
         if (
                 self.wrapped and (
                     y > self._start_y or
-                    y == self._start_y and match.start() >= self._start_x
+                    y == self._start_y and match.start() >= self._start_x + self.original_line_change_diff
                 )
         ):
             raise StopIteration()
@@ -487,6 +493,7 @@ def replace(
                 count += 1
                 with self.edit_action_context('replace', final=True):
                     replaced = match.expand(replace)
+                    search.calc_past_original_offset(line_y, match, replaced)
                     line = screen.file.buf[line_y]
                     if '\n' in replaced:
                         replaced_lines = replaced.split('\n')