From 224de547b3b77e2675eefe31cfa4a983f628380b Mon Sep 17 00:00:00 2001
From: ridiculousfish <corydoras@ridiculousfish.com>
Date: Sun, 2 Dec 2012 11:53:34 -0800
Subject: [PATCH] Ensure we repaint the prompt after resetting it, when colors
 change. Fixes https://github.com/fish-shell/fish-shell/issues/418

---
 reader.cpp | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/reader.cpp b/reader.cpp
index cf5e44506..190d8dc22 100644
--- a/reader.cpp
+++ b/reader.cpp
@@ -758,13 +758,19 @@ void reader_repaint_needed()
 
 void reader_repaint_if_needed()
 {
-    if (data && data->screen_reset_needed)
+    if (data == NULL)
+        return;
+    
+    bool needs_reset = data->screen_reset_needed;
+    bool needs_repaint = needs_reset || data->repaint_needed;
+    
+    if (needs_reset)
     {
         s_reset(&data->screen, screen_reset_current_line_and_prompt);
         data->screen_reset_needed = false;
     }
 
-    if (data && data->repaint_needed)
+    if (needs_repaint)
     {
         reader_repaint();
         /* reader_repaint clears repaint_needed */
@@ -2848,6 +2854,7 @@ const wchar_t *reader_readline()
                     coalescing_repaints = true;
                     exec_prompt();
                     s_reset(&data->screen, screen_reset_current_line_and_prompt);
+                    data->screen_reset_needed = false;
                     reader_repaint();
                 }
                 break;