From 5e556bfd2855420e782e1753da13a4fcb798f2c7 Mon Sep 17 00:00:00 2001
From: zc he <blindfs19@gmail.com>
Date: Wed, 9 Oct 2024 21:50:35 +0800
Subject: [PATCH] Fix multiline prompt resize issue (#841)

---
 src/painting/painter.rs | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/painting/painter.rs b/src/painting/painter.rs
index 8365f072..bf2917ec 100644
--- a/src/painting/painter.rs
+++ b/src/painting/painter.rs
@@ -94,6 +94,7 @@ pub struct Painter {
     terminal_size: (u16, u16),
     last_required_lines: u16,
     large_buffer: bool,
+    just_resized: bool,
     after_cursor_lines: Option<String>,
 }
 
@@ -105,6 +106,7 @@ impl Painter {
             terminal_size: (0, 0),
             last_required_lines: 0,
             large_buffer: false,
+            just_resized: false,
             after_cursor_lines: None,
         }
     }
@@ -197,6 +199,15 @@ impl Painter {
         let screen_width = self.screen_width();
         let screen_height = self.screen_height();
 
+        // Handle resize for multi line prompt
+        if self.just_resized {
+            self.prompt_start_row = self.prompt_start_row.saturating_sub(
+                (lines.prompt_str_left.matches('\n').count()
+                    + lines.prompt_indicator.matches('\n').count()) as u16,
+            );
+            self.just_resized = false;
+        }
+
         // Lines and distance parameters
         let remaining_lines = self.remaining_lines();
         let required_lines = lines.required_lines(screen_width, menu);
@@ -488,6 +499,7 @@ impl Painter {
         // out yet.
         if let Ok(position) = cursor::position() {
             self.prompt_start_row = position.1;
+            self.just_resized = true;
         }
     }