1// Copyright 2011 The Go Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style
3// license that can be found in the LICENSE file.
4
5package terminal
6
7import (
8	"bytes"
9	"io"
10	"sync"
11	"unicode/utf8"
12)
13
14// EscapeCodes contains escape sequences that can be written to the terminal in
15// order to achieve different styles of text.
16type EscapeCodes struct {
17	// Foreground colors
18	Black, Red, Green, Yellow, Blue, Magenta, Cyan, White []byte
19
20	// Reset all attributes
21	Reset []byte
22}
23
24var vt100EscapeCodes = EscapeCodes{
25	Black:   []byte{keyEscape, '[', '3', '0', 'm'},
26	Red:     []byte{keyEscape, '[', '3', '1', 'm'},
27	Green:   []byte{keyEscape, '[', '3', '2', 'm'},
28	Yellow:  []byte{keyEscape, '[', '3', '3', 'm'},
29	Blue:    []byte{keyEscape, '[', '3', '4', 'm'},
30	Magenta: []byte{keyEscape, '[', '3', '5', 'm'},
31	Cyan:    []byte{keyEscape, '[', '3', '6', 'm'},
32	White:   []byte{keyEscape, '[', '3', '7', 'm'},
33
34	Reset: []byte{keyEscape, '[', '0', 'm'},
35}
36
37// Terminal contains the state for running a VT100 terminal that is capable of
38// reading lines of input.
39type Terminal struct {
40	// AutoCompleteCallback, if non-null, is called for each keypress with
41	// the full input line and the current position of the cursor (in
42	// bytes, as an index into |line|). If it returns ok=false, the key
43	// press is processed normally. Otherwise it returns a replacement line
44	// and the new cursor position.
45	AutoCompleteCallback func(line string, pos int, key rune) (newLine string, newPos int, ok bool)
46
47	// Escape contains a pointer to the escape codes for this terminal.
48	// It's always a valid pointer, although the escape codes themselves
49	// may be empty if the terminal doesn't support them.
50	Escape *EscapeCodes
51
52	// lock protects the terminal and the state in this object from
53	// concurrent processing of a key press and a Write() call.
54	lock sync.Mutex
55
56	c      io.ReadWriter
57	prompt []rune
58
59	// line is the current line being entered.
60	line []rune
61	// pos is the logical position of the cursor in line
62	pos int
63	// echo is true if local echo is enabled
64	echo bool
65	// pasteActive is true iff there is a bracketed paste operation in
66	// progress.
67	pasteActive bool
68
69	// cursorX contains the current X value of the cursor where the left
70	// edge is 0. cursorY contains the row number where the first row of
71	// the current line is 0.
72	cursorX, cursorY int
73	// maxLine is the greatest value of cursorY so far.
74	maxLine int
75
76	termWidth, termHeight int
77
78	// outBuf contains the terminal data to be sent.
79	outBuf []byte
80	// remainder contains the remainder of any partial key sequences after
81	// a read. It aliases into inBuf.
82	remainder []byte
83	inBuf     [256]byte
84
85	// history contains previously entered commands so that they can be
86	// accessed with the up and down keys.
87	history stRingBuffer
88	// historyIndex stores the currently accessed history entry, where zero
89	// means the immediately previous entry.
90	historyIndex int
91	// When navigating up and down the history it's possible to return to
92	// the incomplete, initial line. That value is stored in
93	// historyPending.
94	historyPending string
95}
96
97// NewTerminal runs a VT100 terminal on the given ReadWriter. If the ReadWriter is
98// a local terminal, that terminal must first have been put into raw mode.
99// prompt is a string that is written at the start of each input line (i.e.
100// "> ").
101func NewTerminal(c io.ReadWriter, prompt string) *Terminal {
102	return &Terminal{
103		Escape:       &vt100EscapeCodes,
104		c:            c,
105		prompt:       []rune(prompt),
106		termWidth:    80,
107		termHeight:   24,
108		echo:         true,
109		historyIndex: -1,
110	}
111}
112
113const (
114	keyCtrlD     = 4
115	keyCtrlU     = 21
116	keyEnter     = '\r'
117	keyEscape    = 27
118	keyBackspace = 127
119	keyUnknown   = 0xd800 /* UTF-16 surrogate area */ + iota
120	keyUp
121	keyDown
122	keyLeft
123	keyRight
124	keyAltLeft
125	keyAltRight
126	keyHome
127	keyEnd
128	keyDeleteWord
129	keyDeleteLine
130	keyClearScreen
131	keyPasteStart
132	keyPasteEnd
133)
134
135var (
136	crlf       = []byte{'\r', '\n'}
137	pasteStart = []byte{keyEscape, '[', '2', '0', '0', '~'}
138	pasteEnd   = []byte{keyEscape, '[', '2', '0', '1', '~'}
139)
140
141// bytesToKey tries to parse a key sequence from b. If successful, it returns
142// the key and the remainder of the input. Otherwise it returns utf8.RuneError.
143func bytesToKey(b []byte, pasteActive bool) (rune, []byte) {
144	if len(b) == 0 {
145		return utf8.RuneError, nil
146	}
147
148	if !pasteActive {
149		switch b[0] {
150		case 1: // ^A
151			return keyHome, b[1:]
152		case 5: // ^E
153			return keyEnd, b[1:]
154		case 8: // ^H
155			return keyBackspace, b[1:]
156		case 11: // ^K
157			return keyDeleteLine, b[1:]
158		case 12: // ^L
159			return keyClearScreen, b[1:]
160		case 23: // ^W
161			return keyDeleteWord, b[1:]
162		}
163	}
164
165	if b[0] != keyEscape {
166		if !utf8.FullRune(b) {
167			return utf8.RuneError, b
168		}
169		r, l := utf8.DecodeRune(b)
170		return r, b[l:]
171	}
172
173	if !pasteActive && len(b) >= 3 && b[0] == keyEscape && b[1] == '[' {
174		switch b[2] {
175		case 'A':
176			return keyUp, b[3:]
177		case 'B':
178			return keyDown, b[3:]
179		case 'C':
180			return keyRight, b[3:]
181		case 'D':
182			return keyLeft, b[3:]
183		case 'H':
184			return keyHome, b[3:]
185		case 'F':
186			return keyEnd, b[3:]
187		}
188	}
189
190	if !pasteActive && len(b) >= 6 && b[0] == keyEscape && b[1] == '[' && b[2] == '1' && b[3] == ';' && b[4] == '3' {
191		switch b[5] {
192		case 'C':
193			return keyAltRight, b[6:]
194		case 'D':
195			return keyAltLeft, b[6:]
196		}
197	}
198
199	if !pasteActive && len(b) >= 6 && bytes.Equal(b[:6], pasteStart) {
200		return keyPasteStart, b[6:]
201	}
202
203	if pasteActive && len(b) >= 6 && bytes.Equal(b[:6], pasteEnd) {
204		return keyPasteEnd, b[6:]
205	}
206
207	// If we get here then we have a key that we don't recognise, or a
208	// partial sequence. It's not clear how one should find the end of a
209	// sequence without knowing them all, but it seems that [a-zA-Z~] only
210	// appears at the end of a sequence.
211	for i, c := range b[0:] {
212		if c >= 'a' && c <= 'z' || c >= 'A' && c <= 'Z' || c == '~' {
213			return keyUnknown, b[i+1:]
214		}
215	}
216
217	return utf8.RuneError, b
218}
219
220// queue appends data to the end of t.outBuf
221func (t *Terminal) queue(data []rune) {
222	t.outBuf = append(t.outBuf, []byte(string(data))...)
223}
224
225var eraseUnderCursor = []rune{' ', keyEscape, '[', 'D'}
226var space = []rune{' '}
227
228func isPrintable(key rune) bool {
229	isInSurrogateArea := key >= 0xd800 && key <= 0xdbff
230	return key >= 32 && !isInSurrogateArea
231}
232
233// moveCursorToPos appends data to t.outBuf which will move the cursor to the
234// given, logical position in the text.
235func (t *Terminal) moveCursorToPos(pos int) {
236	if !t.echo {
237		return
238	}
239
240	x := visualLength(t.prompt) + pos
241	y := x / t.termWidth
242	x = x % t.termWidth
243
244	up := 0
245	if y < t.cursorY {
246		up = t.cursorY - y
247	}
248
249	down := 0
250	if y > t.cursorY {
251		down = y - t.cursorY
252	}
253
254	left := 0
255	if x < t.cursorX {
256		left = t.cursorX - x
257	}
258
259	right := 0
260	if x > t.cursorX {
261		right = x - t.cursorX
262	}
263
264	t.cursorX = x
265	t.cursorY = y
266	t.move(up, down, left, right)
267}
268
269func (t *Terminal) move(up, down, left, right int) {
270	movement := make([]rune, 3*(up+down+left+right))
271	m := movement
272	for i := 0; i < up; i++ {
273		m[0] = keyEscape
274		m[1] = '['
275		m[2] = 'A'
276		m = m[3:]
277	}
278	for i := 0; i < down; i++ {
279		m[0] = keyEscape
280		m[1] = '['
281		m[2] = 'B'
282		m = m[3:]
283	}
284	for i := 0; i < left; i++ {
285		m[0] = keyEscape
286		m[1] = '['
287		m[2] = 'D'
288		m = m[3:]
289	}
290	for i := 0; i < right; i++ {
291		m[0] = keyEscape
292		m[1] = '['
293		m[2] = 'C'
294		m = m[3:]
295	}
296
297	t.queue(movement)
298}
299
300func (t *Terminal) clearLineToRight() {
301	op := []rune{keyEscape, '[', 'K'}
302	t.queue(op)
303}
304
305const maxLineLength = 4096
306
307func (t *Terminal) setLine(newLine []rune, newPos int) {
308	if t.echo {
309		t.moveCursorToPos(0)
310		t.writeLine(newLine)
311		for i := len(newLine); i < len(t.line); i++ {
312			t.writeLine(space)
313		}
314		t.moveCursorToPos(newPos)
315	}
316	t.line = newLine
317	t.pos = newPos
318}
319
320func (t *Terminal) advanceCursor(places int) {
321	t.cursorX += places
322	t.cursorY += t.cursorX / t.termWidth
323	if t.cursorY > t.maxLine {
324		t.maxLine = t.cursorY
325	}
326	t.cursorX = t.cursorX % t.termWidth
327
328	if places > 0 && t.cursorX == 0 {
329		// Normally terminals will advance the current position
330		// when writing a character. But that doesn't happen
331		// for the last character in a line. However, when
332		// writing a character (except a new line) that causes
333		// a line wrap, the position will be advanced two
334		// places.
335		//
336		// So, if we are stopping at the end of a line, we
337		// need to write a newline so that our cursor can be
338		// advanced to the next line.
339		t.outBuf = append(t.outBuf, '\r', '\n')
340	}
341}
342
343func (t *Terminal) eraseNPreviousChars(n int) {
344	if n == 0 {
345		return
346	}
347
348	if t.pos < n {
349		n = t.pos
350	}
351	t.pos -= n
352	t.moveCursorToPos(t.pos)
353
354	copy(t.line[t.pos:], t.line[n+t.pos:])
355	t.line = t.line[:len(t.line)-n]
356	if t.echo {
357		t.writeLine(t.line[t.pos:])
358		for i := 0; i < n; i++ {
359			t.queue(space)
360		}
361		t.advanceCursor(n)
362		t.moveCursorToPos(t.pos)
363	}
364}
365
366// countToLeftWord returns then number of characters from the cursor to the
367// start of the previous word.
368func (t *Terminal) countToLeftWord() int {
369	if t.pos == 0 {
370		return 0
371	}
372
373	pos := t.pos - 1
374	for pos > 0 {
375		if t.line[pos] != ' ' {
376			break
377		}
378		pos--
379	}
380	for pos > 0 {
381		if t.line[pos] == ' ' {
382			pos++
383			break
384		}
385		pos--
386	}
387
388	return t.pos - pos
389}
390
391// countToRightWord returns then number of characters from the cursor to the
392// start of the next word.
393func (t *Terminal) countToRightWord() int {
394	pos := t.pos
395	for pos < len(t.line) {
396		if t.line[pos] == ' ' {
397			break
398		}
399		pos++
400	}
401	for pos < len(t.line) {
402		if t.line[pos] != ' ' {
403			break
404		}
405		pos++
406	}
407	return pos - t.pos
408}
409
410// visualLength returns the number of visible glyphs in s.
411func visualLength(runes []rune) int {
412	inEscapeSeq := false
413	length := 0
414
415	for _, r := range runes {
416		switch {
417		case inEscapeSeq:
418			if (r >= 'a' && r <= 'z') || (r >= 'A' && r <= 'Z') {
419				inEscapeSeq = false
420			}
421		case r == '\x1b':
422			inEscapeSeq = true
423		default:
424			length++
425		}
426	}
427
428	return length
429}
430
431// handleKey processes the given key and, optionally, returns a line of text
432// that the user has entered.
433func (t *Terminal) handleKey(key rune) (line string, ok bool) {
434	if t.pasteActive && key != keyEnter {
435		t.addKeyToLine(key)
436		return
437	}
438
439	switch key {
440	case keyBackspace:
441		if t.pos == 0 {
442			return
443		}
444		t.eraseNPreviousChars(1)
445	case keyAltLeft:
446		// move left by a word.
447		t.pos -= t.countToLeftWord()
448		t.moveCursorToPos(t.pos)
449	case keyAltRight:
450		// move right by a word.
451		t.pos += t.countToRightWord()
452		t.moveCursorToPos(t.pos)
453	case keyLeft:
454		if t.pos == 0 {
455			return
456		}
457		t.pos--
458		t.moveCursorToPos(t.pos)
459	case keyRight:
460		if t.pos == len(t.line) {
461			return
462		}
463		t.pos++
464		t.moveCursorToPos(t.pos)
465	case keyHome:
466		if t.pos == 0 {
467			return
468		}
469		t.pos = 0
470		t.moveCursorToPos(t.pos)
471	case keyEnd:
472		if t.pos == len(t.line) {
473			return
474		}
475		t.pos = len(t.line)
476		t.moveCursorToPos(t.pos)
477	case keyUp:
478		entry, ok := t.history.NthPreviousEntry(t.historyIndex + 1)
479		if !ok {
480			return "", false
481		}
482		if t.historyIndex == -1 {
483			t.historyPending = string(t.line)
484		}
485		t.historyIndex++
486		runes := []rune(entry)
487		t.setLine(runes, len(runes))
488	case keyDown:
489		switch t.historyIndex {
490		case -1:
491			return
492		case 0:
493			runes := []rune(t.historyPending)
494			t.setLine(runes, len(runes))
495			t.historyIndex--
496		default:
497			entry, ok := t.history.NthPreviousEntry(t.historyIndex - 1)
498			if ok {
499				t.historyIndex--
500				runes := []rune(entry)
501				t.setLine(runes, len(runes))
502			}
503		}
504	case keyEnter:
505		t.moveCursorToPos(len(t.line))
506		t.queue([]rune("\r\n"))
507		line = string(t.line)
508		ok = true
509		t.line = t.line[:0]
510		t.pos = 0
511		t.cursorX = 0
512		t.cursorY = 0
513		t.maxLine = 0
514	case keyDeleteWord:
515		// Delete zero or more spaces and then one or more characters.
516		t.eraseNPreviousChars(t.countToLeftWord())
517	case keyDeleteLine:
518		// Delete everything from the current cursor position to the
519		// end of line.
520		for i := t.pos; i < len(t.line); i++ {
521			t.queue(space)
522			t.advanceCursor(1)
523		}
524		t.line = t.line[:t.pos]
525		t.moveCursorToPos(t.pos)
526	case keyCtrlD:
527		// Erase the character under the current position.
528		// The EOF case when the line is empty is handled in
529		// readLine().
530		if t.pos < len(t.line) {
531			t.pos++
532			t.eraseNPreviousChars(1)
533		}
534	case keyCtrlU:
535		t.eraseNPreviousChars(t.pos)
536	case keyClearScreen:
537		// Erases the screen and moves the cursor to the home position.
538		t.queue([]rune("\x1b[2J\x1b[H"))
539		t.queue(t.prompt)
540		t.cursorX, t.cursorY = 0, 0
541		t.advanceCursor(visualLength(t.prompt))
542		t.setLine(t.line, t.pos)
543	default:
544		if t.AutoCompleteCallback != nil {
545			prefix := string(t.line[:t.pos])
546			suffix := string(t.line[t.pos:])
547
548			t.lock.Unlock()
549			newLine, newPos, completeOk := t.AutoCompleteCallback(prefix+suffix, len(prefix), key)
550			t.lock.Lock()
551
552			if completeOk {
553				t.setLine([]rune(newLine), utf8.RuneCount([]byte(newLine)[:newPos]))
554				return
555			}
556		}
557		if !isPrintable(key) {
558			return
559		}
560		if len(t.line) == maxLineLength {
561			return
562		}
563		t.addKeyToLine(key)
564	}
565	return
566}
567
568// addKeyToLine inserts the given key at the current position in the current
569// line.
570func (t *Terminal) addKeyToLine(key rune) {
571	if len(t.line) == cap(t.line) {
572		newLine := make([]rune, len(t.line), 2*(1+len(t.line)))
573		copy(newLine, t.line)
574		t.line = newLine
575	}
576	t.line = t.line[:len(t.line)+1]
577	copy(t.line[t.pos+1:], t.line[t.pos:])
578	t.line[t.pos] = key
579	if t.echo {
580		t.writeLine(t.line[t.pos:])
581	}
582	t.pos++
583	t.moveCursorToPos(t.pos)
584}
585
586func (t *Terminal) writeLine(line []rune) {
587	for len(line) != 0 {
588		remainingOnLine := t.termWidth - t.cursorX
589		todo := len(line)
590		if todo > remainingOnLine {
591			todo = remainingOnLine
592		}
593		t.queue(line[:todo])
594		t.advanceCursor(visualLength(line[:todo]))
595		line = line[todo:]
596	}
597}
598
599// writeWithCRLF writes buf to w but replaces all occurrences of \n with \r\n.
600func writeWithCRLF(w io.Writer, buf []byte) (n int, err error) {
601	for len(buf) > 0 {
602		i := bytes.IndexByte(buf, '\n')
603		todo := len(buf)
604		if i >= 0 {
605			todo = i
606		}
607
608		var nn int
609		nn, err = w.Write(buf[:todo])
610		n += nn
611		if err != nil {
612			return n, err
613		}
614		buf = buf[todo:]
615
616		if i >= 0 {
617			if _, err = w.Write(crlf); err != nil {
618				return n, err
619			}
620			n += 1
621			buf = buf[1:]
622		}
623	}
624
625	return n, nil
626}
627
628func (t *Terminal) Write(buf []byte) (n int, err error) {
629	t.lock.Lock()
630	defer t.lock.Unlock()
631
632	if t.cursorX == 0 && t.cursorY == 0 {
633		// This is the easy case: there's nothing on the screen that we
634		// have to move out of the way.
635		return writeWithCRLF(t.c, buf)
636	}
637
638	// We have a prompt and possibly user input on the screen. We
639	// have to clear it first.
640	t.move(0 /* up */, 0 /* down */, t.cursorX /* left */, 0 /* right */)
641	t.cursorX = 0
642	t.clearLineToRight()
643
644	for t.cursorY > 0 {
645		t.move(1 /* up */, 0, 0, 0)
646		t.cursorY--
647		t.clearLineToRight()
648	}
649
650	if _, err = t.c.Write(t.outBuf); err != nil {
651		return
652	}
653	t.outBuf = t.outBuf[:0]
654
655	if n, err = writeWithCRLF(t.c, buf); err != nil {
656		return
657	}
658
659	t.writeLine(t.prompt)
660	if t.echo {
661		t.writeLine(t.line)
662	}
663
664	t.moveCursorToPos(t.pos)
665
666	if _, err = t.c.Write(t.outBuf); err != nil {
667		return
668	}
669	t.outBuf = t.outBuf[:0]
670	return
671}
672
673// ReadPassword temporarily changes the prompt and reads a password, without
674// echo, from the terminal.
675func (t *Terminal) ReadPassword(prompt string) (line string, err error) {
676	t.lock.Lock()
677	defer t.lock.Unlock()
678
679	oldPrompt := t.prompt
680	t.prompt = []rune(prompt)
681	t.echo = false
682
683	line, err = t.readLine()
684
685	t.prompt = oldPrompt
686	t.echo = true
687
688	return
689}
690
691// ReadLine returns a line of input from the terminal.
692func (t *Terminal) ReadLine() (line string, err error) {
693	t.lock.Lock()
694	defer t.lock.Unlock()
695
696	return t.readLine()
697}
698
699func (t *Terminal) readLine() (line string, err error) {
700	// t.lock must be held at this point
701
702	if t.cursorX == 0 && t.cursorY == 0 {
703		t.writeLine(t.prompt)
704		t.c.Write(t.outBuf)
705		t.outBuf = t.outBuf[:0]
706	}
707
708	lineIsPasted := t.pasteActive
709
710	for {
711		rest := t.remainder
712		lineOk := false
713		for !lineOk {
714			var key rune
715			key, rest = bytesToKey(rest, t.pasteActive)
716			if key == utf8.RuneError {
717				break
718			}
719			if !t.pasteActive {
720				if key == keyCtrlD {
721					if len(t.line) == 0 {
722						return "", io.EOF
723					}
724				}
725				if key == keyPasteStart {
726					t.pasteActive = true
727					if len(t.line) == 0 {
728						lineIsPasted = true
729					}
730					continue
731				}
732			} else if key == keyPasteEnd {
733				t.pasteActive = false
734				continue
735			}
736			if !t.pasteActive {
737				lineIsPasted = false
738			}
739			line, lineOk = t.handleKey(key)
740		}
741		if len(rest) > 0 {
742			n := copy(t.inBuf[:], rest)
743			t.remainder = t.inBuf[:n]
744		} else {
745			t.remainder = nil
746		}
747		t.c.Write(t.outBuf)
748		t.outBuf = t.outBuf[:0]
749		if lineOk {
750			if t.echo {
751				t.historyIndex = -1
752				t.history.Add(line)
753			}
754			if lineIsPasted {
755				err = ErrPasteIndicator
756			}
757			return
758		}
759
760		// t.remainder is a slice at the beginning of t.inBuf
761		// containing a partial key sequence
762		readBuf := t.inBuf[len(t.remainder):]
763		var n int
764
765		t.lock.Unlock()
766		n, err = t.c.Read(readBuf)
767		t.lock.Lock()
768
769		if err != nil {
770			return
771		}
772
773		t.remainder = t.inBuf[:n+len(t.remainder)]
774	}
775}
776
777// SetPrompt sets the prompt to be used when reading subsequent lines.
778func (t *Terminal) SetPrompt(prompt string) {
779	t.lock.Lock()
780	defer t.lock.Unlock()
781
782	t.prompt = []rune(prompt)
783}
784
785func (t *Terminal) clearAndRepaintLinePlusNPrevious(numPrevLines int) {
786	// Move cursor to column zero at the start of the line.
787	t.move(t.cursorY, 0, t.cursorX, 0)
788	t.cursorX, t.cursorY = 0, 0
789	t.clearLineToRight()
790	for t.cursorY < numPrevLines {
791		// Move down a line
792		t.move(0, 1, 0, 0)
793		t.cursorY++
794		t.clearLineToRight()
795	}
796	// Move back to beginning.
797	t.move(t.cursorY, 0, 0, 0)
798	t.cursorX, t.cursorY = 0, 0
799
800	t.queue(t.prompt)
801	t.advanceCursor(visualLength(t.prompt))
802	t.writeLine(t.line)
803	t.moveCursorToPos(t.pos)
804}
805
806func (t *Terminal) SetSize(width, height int) error {
807	t.lock.Lock()
808	defer t.lock.Unlock()
809
810	if width == 0 {
811		width = 1
812	}
813
814	oldWidth := t.termWidth
815	t.termWidth, t.termHeight = width, height
816
817	switch {
818	case width == oldWidth:
819		// If the width didn't change then nothing else needs to be
820		// done.
821		return nil
822	case len(t.line) == 0 && t.cursorX == 0 && t.cursorY == 0:
823		// If there is nothing on current line and no prompt printed,
824		// just do nothing
825		return nil
826	case width < oldWidth:
827		// Some terminals (e.g. xterm) will truncate lines that were
828		// too long when shinking. Others, (e.g. gnome-terminal) will
829		// attempt to wrap them. For the former, repainting t.maxLine
830		// works great, but that behaviour goes badly wrong in the case
831		// of the latter because they have doubled every full line.
832
833		// We assume that we are working on a terminal that wraps lines
834		// and adjust the cursor position based on every previous line
835		// wrapping and turning into two. This causes the prompt on
836		// xterms to move upwards, which isn't great, but it avoids a
837		// huge mess with gnome-terminal.
838		if t.cursorX >= t.termWidth {
839			t.cursorX = t.termWidth - 1
840		}
841		t.cursorY *= 2
842		t.clearAndRepaintLinePlusNPrevious(t.maxLine * 2)
843	case width > oldWidth:
844		// If the terminal expands then our position calculations will
845		// be wrong in the future because we think the cursor is
846		// |t.pos| chars into the string, but there will be a gap at
847		// the end of any wrapped line.
848		//
849		// But the position will actually be correct until we move, so
850		// we can move back to the beginning and repaint everything.
851		t.clearAndRepaintLinePlusNPrevious(t.maxLine)
852	}
853
854	_, err := t.c.Write(t.outBuf)
855	t.outBuf = t.outBuf[:0]
856	return err
857}
858
859type pasteIndicatorError struct{}
860
861func (pasteIndicatorError) Error() string {
862	return "terminal: ErrPasteIndicator not correctly handled"
863}
864
865// ErrPasteIndicator may be returned from ReadLine as the error, in addition
866// to valid line data. It indicates that bracketed paste mode is enabled and
867// that the returned line consists only of pasted data. Programs may wish to
868// interpret pasted data more literally than typed data.
869var ErrPasteIndicator = pasteIndicatorError{}
870
871// SetBracketedPasteMode requests that the terminal bracket paste operations
872// with markers. Not all terminals support this but, if it is supported, then
873// enabling this mode will stop any autocomplete callback from running due to
874// pastes. Additionally, any lines that are completely pasted will be returned
875// from ReadLine with the error set to ErrPasteIndicator.
876func (t *Terminal) SetBracketedPasteMode(on bool) {
877	if on {
878		io.WriteString(t.c, "\x1b[?2004h")
879	} else {
880		io.WriteString(t.c, "\x1b[?2004l")
881	}
882}
883
884// stRingBuffer is a ring buffer of strings.
885type stRingBuffer struct {
886	// entries contains max elements.
887	entries []string
888	max     int
889	// head contains the index of the element most recently added to the ring.
890	head int
891	// size contains the number of elements in the ring.
892	size int
893}
894
895func (s *stRingBuffer) Add(a string) {
896	if s.entries == nil {
897		const defaultNumEntries = 100
898		s.entries = make([]string, defaultNumEntries)
899		s.max = defaultNumEntries
900	}
901
902	s.head = (s.head + 1) % s.max
903	s.entries[s.head] = a
904	if s.size < s.max {
905		s.size++
906	}
907}
908
909// NthPreviousEntry returns the value passed to the nth previous call to Add.
910// If n is zero then the immediately prior value is returned, if one, then the
911// next most recent, and so on. If such an element doesn't exist then ok is
912// false.
913func (s *stRingBuffer) NthPreviousEntry(n int) (value string, ok bool) {
914	if n >= s.size {
915		return "", false
916	}
917	index := s.head - n
918	if index < 0 {
919		index += s.max
920	}
921	return s.entries[index], true
922}
923
924// readPasswordLine reads from reader until it finds \n or io.EOF.
925// The slice returned does not include the \n.
926// readPasswordLine also ignores any \r it finds.
927func readPasswordLine(reader io.Reader) ([]byte, error) {
928	var buf [1]byte
929	var ret []byte
930
931	for {
932		n, err := reader.Read(buf[:])
933		if n > 0 {
934			switch buf[0] {
935			case '\n':
936				return ret, nil
937			case '\r':
938				// remove \r from passwords on Windows
939			default:
940				ret = append(ret, buf[0])
941			}
942			continue
943		}
944		if err != nil {
945			if err == io.EOF && len(ret) > 0 {
946				return ret, nil
947			}
948			return ret, err
949		}
950	}
951}
952