github.com/glycerine/xcryptossh@v7.0.4+incompatible/terminal/terminal.go (about) 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 5 package terminal 6 7 import ( 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. 16 type 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 24 var 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. 39 type 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 // "> "). 101 func 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 113 const ( 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 135 var ( 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. 143 func 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 221 func (t *Terminal) queue(data []rune) { 222 t.outBuf = append(t.outBuf, []byte(string(data))...) 223 } 224 225 var eraseUnderCursor = []rune{' ', keyEscape, '[', 'D'} 226 var space = []rune{' '} 227 228 func 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. 235 func (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 269 func (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 300 func (t *Terminal) clearLineToRight() { 301 op := []rune{keyEscape, '[', 'K'} 302 t.queue(op) 303 } 304 305 const maxLineLength = 4096 306 307 func (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 320 func (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 343 func (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. 368 func (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. 393 func (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. 411 func 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. 433 func (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. 570 func (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 586 func (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. 600 func 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 628 func (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. 675 func (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. 692 func (t *Terminal) ReadLine() (line string, err error) { 693 t.lock.Lock() 694 defer t.lock.Unlock() 695 696 return t.readLine() 697 } 698 699 func (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. 778 func (t *Terminal) SetPrompt(prompt string) { 779 t.lock.Lock() 780 defer t.lock.Unlock() 781 782 t.prompt = []rune(prompt) 783 } 784 785 func (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 806 func (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 859 type pasteIndicatorError struct{} 860 861 func (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. 869 var 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. 876 func (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. 885 type 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 895 func (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. 913 func (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. 927 func 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 }