From 4ae9d7e0fd1ee6d96f67b8e08a42b6e0e84da638 Mon Sep 17 00:00:00 2001
From: Cheney <me@chenye.org>
Date: Sat, 5 Mar 2016 15:27:12 +0800
Subject: [PATCH] fix multiline

---
 runebuf.go | 65 ++++++++++++++++++++++++++++++++++++------------------
 utils.go   | 24 ++++++++++++++++++--
 2 files changed, 66 insertions(+), 23 deletions(-)

diff --git a/runebuf.go b/runebuf.go
index a137fb9..1d06a31 100644
--- a/runebuf.go
+++ b/runebuf.go
@@ -2,6 +2,7 @@ package readline
 
 import (
 	"bytes"
+	"fmt"
 	"io"
 	"strings"
 
@@ -293,7 +294,8 @@ func (r *RuneBuffer) MoveToLineEnd() {
 }
 
 func (r *RuneBuffer) LineCount() int {
-	return LineCount(r.cfg.FuncGetWidth, runes.WidthAll(r.buf)+r.PromptLen())
+	return LineCount(r.cfg.FuncGetWidth(),
+		runes.WidthAll(r.buf)+r.PromptLen())
 }
 
 func (r *RuneBuffer) MoveTo(ch rune, prevChar, reverse bool) (success bool) {
@@ -326,22 +328,9 @@ func (r *RuneBuffer) MoveTo(ch rune, prevChar, reverse bool) (success bool) {
 }
 
 func (r *RuneBuffer) IdxLine() int {
-	totalWidth := runes.WidthAll(r.buf[:r.idx]) + r.PromptLen()
-	w := r.cfg.FuncGetWidth()
-	if w <= 0 {
-		return -1
-	}
-	line := totalWidth / w
-
-	// if cursor is in last colmun and not any character behind it
-	// the cursor will in the first line, otherwise will in the second line
-	// this situation only occurs in golang's Stdout
-	// TODO: figure out why
-	if totalWidth%w == 0 && len(r.buf) == r.idx && !isWindows {
-		line--
-	}
-
-	return line
+	sw := r.cfg.FuncGetWidth()
+	sp := SplitByLine(r.PromptLen(), sw, r.buf[:r.idx])
+	return len(sp) - 1
 }
 
 func (r *RuneBuffer) CursorLineCount() int {
@@ -373,11 +362,45 @@ func (r *RuneBuffer) output() []byte {
 		} else {
 			buf.Write([]byte(string(r.cfg.MaskRune)))
 		}
+		if len(r.buf) > r.idx {
+			buf.Write(runes.Backspace(r.buf[r.idx:]))
+		}
+
 	} else {
-		buf.Write([]byte(string(r.buf)))
-	}
-	if len(r.buf) > r.idx {
-		buf.Write(runes.Backspace(r.buf[r.idx:]))
+		sw := r.cfg.FuncGetWidth()
+		sp := SplitByLine(r.PromptLen(), sw, r.buf)
+		written := 0
+		idxInLine := 0
+		for idx, s := range sp {
+			buf.Write([]byte(s))
+			if r.idx > written && r.idx < written+len(s) {
+				idxInLine = r.idx - written
+			}
+			written += len(s)
+
+			if idx < len(sp)-1 && !isWindows {
+				buf.Write([]byte{'\n'})
+			}
+		}
+		if len(r.buf) > r.idx {
+			targetLine := r.IdxLine()
+			currentLine := len(sp) - 1
+			// assert currentLine >= targetLine
+			if targetLine == 0 {
+				idxInLine += r.PromptLen()
+			}
+			buf.WriteString("\r")
+			if currentLine > targetLine {
+				buf.WriteString(fmt.Sprintf(
+					"\033[%vA", currentLine-targetLine,
+				))
+			}
+			if idxInLine > 0 {
+				buf.WriteString(fmt.Sprintf(
+					"\033[%vC", idxInLine,
+				))
+			}
+		}
 	}
 	return buf.Bytes()
 }
diff --git a/utils.go b/utils.go
index a270952..01488a7 100644
--- a/utils.go
+++ b/utils.go
@@ -2,9 +2,12 @@ package readline
 
 import (
 	"bufio"
+	"bytes"
 	"strconv"
 	"syscall"
 
+	"github.com/chzyer/readline/runes"
+
 	"golang.org/x/crypto/ssh/terminal"
 )
 
@@ -86,9 +89,26 @@ func escapeKey(r rune) rune {
 	return r
 }
 
+func SplitByLine(start, screenWidth int, rs []rune) []string {
+	var ret []string
+	buf := bytes.NewBuffer(nil)
+	currentWidth := start
+	for _, r := range rs {
+		w := runes.Width(r)
+		currentWidth += w
+		buf.WriteRune(r)
+		if currentWidth >= screenWidth {
+			ret = append(ret, buf.String())
+			buf.Reset()
+			currentWidth = 0
+		}
+	}
+	ret = append(ret, buf.String())
+	return ret
+}
+
 // calculate how many lines for N character
-func LineCount(getWidth func() int, w int) int {
-	screenWidth := getWidth()
+func LineCount(screenWidth, w int) int {
 	r := w / screenWidth
 	if w%screenWidth != 0 {
 		r++