only count the last line of promt when caculate prompt width

1 files changed, 4 insertions, 0 deletions

diff --git a/screen.c b/screen.c
index fb173ce6..0046e0cf 100644
--- a/screen.c
+++ b/screen.c
@@ -236,6 +236,10 @@ static int calc_prompt_width( wchar_t *prompt )
 		{
 			res = next_tab_stop( res );
 		}
+		else if( prompt[j] == L'\n' )
+		{
+			res = 0;
+		}
 		else
 		{
 			/*