fixed not working in screen #3

README.md

@@ -195,12 +195,17 @@ Users can change that in terminal simulator(i.e. iTerm2) to `Alt`+`B`
 # Tested with
 
 | Environment                   | $TERM  |
-|-------------------------------|-------|
+|-------------------------------|--------|
 | Mac OS X iTerm2               | xterm  |
 | Mac OS X default Terminal.app | xterm  |
+| Mac OS X iTerm2 Screen        | screen |
+| Mac OS X iTerm2 Tmux          | screen |
 | Ubuntu Server 14.04 LTS       | linux  |
 | Centos 7                      | linux  |
 
+### Notice:
+* `Ctrl`+`A` is not working as a Control Command in screen
+
 If you test it otherwhere, whether it works fine or not, please let me know!
 
 # Feedback

operation.go

@@ -24,7 +24,7 @@ func (w *wrapWriter) Write(b []byte) (int, error) {
 	buf.Clean()
 	n, err := w.target.Write(b)
 	if w.t.IsReading() {
-		w.r.buf.Refresh()
+		w.r.buf.Refresh(nil)
 	}
 	if w.r.IsSearchMode() {
 		w.r.SearchRefresh(-1)
@@ -61,7 +61,7 @@ func (o *Operation) ioloop() {
 				continue
 			}
 
-			o.buf.Refresh()
+			o.buf.Refresh(nil)
 			switch r {
 			case CharEnter, CharCtrlJ:
 				o.UpdateHistory(o.buf.Runes(), false)
@@ -75,14 +75,16 @@ func (o *Operation) ioloop() {
 		}
 
 		switch r {
+		case 'i':
+			o.buf.Clean()
 		case CharCancel:
 			if o.IsSearchMode() {
 				o.ExitSearchMode(true)
-				o.buf.Refresh()
+				o.buf.Refresh(nil)
 			}
 			if o.IsInCompleteMode() {
 				o.ExitCompleteMode(true)
-				o.buf.Refresh()
+				o.buf.Refresh(nil)
 			}
 		case CharTab:
 			if o.opCompleter == nil {
@@ -162,11 +164,11 @@ func (o *Operation) ioloop() {
 			if o.IsInCompleteMode() {
 				o.t.KickRead()
 				o.ExitCompleteMode(true)
-				o.buf.Refresh()
+				o.buf.Refresh(nil)
 				break
 			}
 			o.buf.MoveToLineEnd()
-			o.buf.Refresh()
+			o.buf.Refresh(nil)
 			o.buf.WriteString("^C\n")
 			o.outchan <- nil
 		default:
@@ -183,13 +185,13 @@ func (o *Operation) ioloop() {
 		}
 		if !keepInSearchMode && o.IsSearchMode() {
 			o.ExitSearchMode(false)
-			o.buf.Refresh()
+			o.buf.Refresh(nil)
 		} else if o.IsInCompleteMode() {
 			if !keepInCompleteMode {
 				o.ExitCompleteMode(false)
-				o.buf.Refresh()
+				o.buf.Refresh(nil)
 			} else {
-				o.buf.Refresh()
+				o.buf.Refresh(nil)
 				o.CompleteRefresh()
 			}
 		}
@@ -216,7 +218,7 @@ func (o *Operation) String() (string, error) {
 }
 
 func (o *Operation) Runes() ([]rune, error) {
-	o.buf.Refresh() // print prompt
+	o.buf.Refresh(nil) // print prompt
 	o.t.KickRead()
 	r := <-o.outchan
 	if r == nil {

runebuf.go

@@ -10,6 +10,8 @@ type RuneBuffer struct {
 	idx    int
 	prompt []rune
 	w      io.Writer
+
+	cleanInScreen bool
 }
 
 func NewRuneBuffer(w io.Writer, prompt string) *RuneBuffer {
@@ -57,16 +59,19 @@ func (r *RuneBuffer) MoveToLineStart() {
 	if r.idx == 0 {
 		return
 	}
+
+	r.Refresh(func() {
 		r.idx = 0
-	r.Refresh()
+	})
 }
 
 func (r *RuneBuffer) MoveBackward() {
 	if r.idx == 0 {
 		return
 	}
+	r.Refresh(func() {
 		r.idx--
-	r.Refresh()
+	})
 }
 
 func (r *RuneBuffer) WriteString(s string) {
@@ -78,26 +83,29 @@ func (r *RuneBuffer) WriteRune(s rune) {
 }
 
 func (r *RuneBuffer) WriteRunes(s []rune) {
+	r.Refresh(func() {
 		tail := append(s, r.buf[r.idx:]...)
 		r.buf = append(r.buf[:r.idx], tail...)
 		r.idx += len(s)
-	r.Refresh()
+	})
 }
 
 func (r *RuneBuffer) MoveForward() {
+	r.Refresh(func() {
 		if r.idx == len(r.buf) {
 			return
 		}
 		r.idx++
-	r.Refresh()
+	})
 }
 
 func (r *RuneBuffer) Delete() {
+	r.Refresh(func() {
 		if r.idx == len(r.buf) {
 			return
 		}
 		r.buf = append(r.buf[:r.idx], r.buf[r.idx+1:]...)
-	r.Refresh()
+	})
 }
 
 func (r *RuneBuffer) DeleteWord() {
@@ -110,8 +118,9 @@ func (r *RuneBuffer) DeleteWord() {
 	}
 	for i := init + 1; i < len(r.buf); i++ {
 		if !IsWordBreak(r.buf[i]) && IsWordBreak(r.buf[i-1]) {
+			r.Refresh(func() {
 				r.buf = append(r.buf[:r.idx], r.buf[i-1:]...)
-			r.Refresh()
+			})
 			return
 		}
 	}
@@ -124,13 +133,15 @@ func (r *RuneBuffer) MoveToPrevWord() {
 	}
 	for i := r.idx - 1; i > 0; i-- {
 		if !IsWordBreak(r.buf[i]) && IsWordBreak(r.buf[i-1]) {
+			r.Refresh(func() {
 				r.idx = i
-			r.Refresh()
+			})
 			return
 		}
 	}
+	r.Refresh(func() {
 		r.idx = 0
-	r.Refresh()
+	})
 }
 
 func (r *RuneBuffer) SetIdx(idx int) (change int) {
@@ -140,18 +151,21 @@ func (r *RuneBuffer) SetIdx(idx int) (change int) {
 }
 
 func (r *RuneBuffer) Kill() {
+	r.Refresh(func() {
 		r.buf = r.buf[:r.idx]
-	r.Refresh()
+	})
 }
 
 func (r *RuneBuffer) Transpose() {
 	if len(r.buf) < 2 {
 		if len(r.buf) == 1 {
+			r.Refresh(func() {
 				r.idx++
-			r.Refresh()
+			})
 		}
 		return
 	}
+	r.Refresh(func() {
 		if r.idx == 0 {
 			r.idx = 1
 		} else if r.idx >= len(r.buf) {
@@ -159,19 +173,21 @@ func (r *RuneBuffer) Transpose() {
 		}
 		r.buf[r.idx], r.buf[r.idx-1] = r.buf[r.idx-1], r.buf[r.idx]
 		r.idx++
-	r.Refresh()
+	})
 }
 
 func (r *RuneBuffer) MoveToNextWord() {
 	for i := r.idx + 1; i < len(r.buf); i++ {
 		if !IsWordBreak(r.buf[i]) && IsWordBreak(r.buf[i-1]) {
+			r.Refresh(func() {
 				r.idx = i
-			r.Refresh()
+			})
 			return
 		}
 	}
+	r.Refresh(func() {
 		r.idx = len(r.buf)
-	r.Refresh()
+	})
 }
 
 func (r *RuneBuffer) BackEscapeWord() {
@@ -180,33 +196,38 @@ func (r *RuneBuffer) BackEscapeWord() {
 	}
 	for i := r.idx - 1; i > 0; i-- {
 		if !IsWordBreak(r.buf[i]) && IsWordBreak(r.buf[i-1]) {
+			r.Refresh(func() {
 				r.buf = append(r.buf[:i], r.buf[r.idx:]...)
 				r.idx = i
-			r.Refresh()
+			})
 			return
 		}
 	}
 
+	r.Refresh(func() {
 		r.buf = r.buf[:0]
 		r.idx = 0
-	r.Refresh()
+	})
 }
 
 func (r *RuneBuffer) Backspace() {
 	if r.idx == 0 {
 		return
 	}
+	r.Refresh(func() {
 		r.idx--
 		r.buf = append(r.buf[:r.idx], r.buf[r.idx+1:]...)
-	r.Refresh()
+	})
 }
 
 func (r *RuneBuffer) MoveToLineEnd() {
 	if r.idx == len(r.buf) {
 		return
 	}
+
+	r.Refresh(func() {
 		r.idx = len(r.buf)
-	r.Refresh()
+	})
 }
 
 func (r *RuneBuffer) LineCount() int {
@@ -216,10 +237,9 @@ func (r *RuneBuffer) LineCount() int {
 func (r *RuneBuffer) IdxLine() int {
 	totalWidth := RunesWidth(r.buf[:r.idx]) + r.PromptLen()
 	w := getWidth()
-	line := 0
-	for totalWidth >= w {
-		totalWidth -= w
-		line++
+	line := totalWidth / w
+	if totalWidth%w == 0 {
+		line--
 	}
 	return line
 }
@@ -228,13 +248,17 @@ func (r *RuneBuffer) CursorLineCount() int {
 	return r.LineCount() - r.IdxLine()
 }
 
-func (r *RuneBuffer) Refresh() {
-	r.w.Write(r.Output())
+func (r *RuneBuffer) Refresh(f func()) {
+	r.Clean()
+	if f != nil {
+		f()
+	}
+	r.w.Write(r.output())
+	r.cleanInScreen = false
 }
 
-func (r *RuneBuffer) Output() []byte {
+func (r *RuneBuffer) output() []byte {
 	buf := bytes.NewBuffer(nil)
-	buf.Write(r.CleanOutput())
 	buf.WriteString(string(r.prompt))
 	buf.Write([]byte(string(r.buf)))
 	if len(r.buf) > r.idx {
@@ -247,14 +271,24 @@ func (r *RuneBuffer) CleanOutput() []byte {
 	buf := bytes.NewBuffer(nil)
 	buf.Write([]byte("\033[J")) // just like ^k :)
 
-	// TODO: calculate how many line before cursor.
-	for i := 0; i <= 100; i++ {
+	idxLine := r.IdxLine()
+	if idxLine == 0 {
+		buf.WriteString("\033[2K\r")
+		return buf.Bytes()
+	}
+
+	for i := 0; i < idxLine; i++ {
 		buf.WriteString("\033[2K\r\b")
 	}
+	buf.WriteString("\033[2K\r")
 	return buf.Bytes()
 }
 
 func (r *RuneBuffer) Clean() {
+	if r.cleanInScreen {
+		return
+	}
+	r.cleanInScreen = true
 	r.w.Write(r.CleanOutput())
 }
 
@@ -291,9 +325,10 @@ func (r *RuneBuffer) SetStyle(start, end int, style string) {
 }
 
 func (r *RuneBuffer) SetWithIdx(idx int, buf []rune) {
+	r.Refresh(func() {
 		r.buf = buf
 		r.idx = idx
-	r.Refresh()
+	})
 }
 
 func (r *RuneBuffer) Set(buf []rune) {