blob: 9a887598ff40437dae4a6a85dd8502e428917450 [file] [log] [blame]
khenaidooac637102019-01-14 15:44:34 -05001// 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++
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}