package tui import ( "fmt" "os" "os/exec" "regexp" "strconv" "strings" "syscall" "time" "unicode/utf8" "github.com/junegunn/fzf/src/util" "golang.org/x/crypto/ssh/terminal" ) const ( defaultWidth = 80 defaultHeight = 24 defaultEscDelay = 100 escPollInterval = 5 offsetPollTries = 10 ) const consoleDevice string = "/dev/tty" var offsetRegexp *regexp.Regexp = regexp.MustCompile("(.*)\x1b\\[([0-9]+);([0-9]+)R") func openTtyIn() *os.File { in, err := os.OpenFile(consoleDevice, syscall.O_RDONLY, 0) if err != nil { tty := ttyname() if len(tty) > 0 { if in, err := os.OpenFile(tty, syscall.O_RDONLY, 0); err == nil { return in } } fmt.Fprintln(os.Stderr, "Failed to open "+consoleDevice) os.Exit(2) } return in } func (r *LightRenderer) stderr(str string) { r.stderrInternal(str, true) } // FIXME: Need better handling of non-displayable characters func (r *LightRenderer) stderrInternal(str string, allowNLCR bool) { bytes := []byte(str) runes := []rune{} for len(bytes) > 0 { r, sz := utf8.DecodeRune(bytes) nlcr := r == '\n' || r == '\r' if r >= 32 || r == '\x1b' || nlcr { if r == utf8.RuneError || nlcr && !allowNLCR { runes = append(runes, ' ') } else { runes = append(runes, r) } } bytes = bytes[sz:] } r.queued += string(runes) } func (r *LightRenderer) csi(code string) { r.stderr("\x1b[" + code) } func (r *LightRenderer) flush() { if len(r.queued) > 0 { fmt.Fprint(os.Stderr, r.queued) r.queued = "" } } // Light renderer type LightRenderer struct { theme *ColorTheme mouse bool forceBlack bool clearOnExit bool prevDownTime time.Time clickY []int ttyin *os.File buffer []byte origState *terminal.State width int height int yoffset int tabstop int escDelay int fullscreen bool upOneLine bool queued string y int x int maxHeightFunc func(int) int } type LightWindow struct { renderer *LightRenderer colored bool border BorderStyle top int left int width int height int posx int posy int tabstop int fg Color bg Color } func NewLightRenderer(theme *ColorTheme, forceBlack bool, mouse bool, tabstop int, clearOnExit bool, fullscreen bool, maxHeightFunc func(int) int) Renderer { r := LightRenderer{ theme: theme, forceBlack: forceBlack, mouse: mouse, clearOnExit: clearOnExit, ttyin: openTtyIn(), yoffset: 0, tabstop: tabstop, fullscreen: fullscreen, upOneLine: false, maxHeightFunc: maxHeightFunc} return &r } func (r *LightRenderer) fd() int { return int(r.ttyin.Fd()) } func (r *LightRenderer) defaultTheme() *ColorTheme { if strings.Contains(os.Getenv("TERM"), "256") { return Dark256 } colors, err := exec.Command("tput", "colors").Output() if err == nil && atoi(strings.TrimSpace(string(colors)), 16) > 16 { return Dark256 } return Default16 } func (r *LightRenderer) findOffset() (row int, col int) { r.csi("6n") r.flush() bytes := []byte{} for tries := 0; tries < offsetPollTries; tries++ { bytes = r.getBytesInternal(bytes, tries > 0) offsets := offsetRegexp.FindSubmatch(bytes) if len(offsets) > 3 { // add anything we skipped over to the input buffer r.buffer = append(r.buffer, offsets[1]...) return atoi(string(offsets[2]), 0) - 1, atoi(string(offsets[3]), 0) - 1 } } return -1, -1 } func repeat(r rune, times int) string { if times > 0 { return strings.Repeat(string(r), times) } return "" } func atoi(s string, defaultValue int) int { value, err := strconv.Atoi(s) if err != nil { return defaultValue } return value } func (r *LightRenderer) Init() { r.escDelay = atoi(os.Getenv("ESCDELAY"), defaultEscDelay) fd := r.fd() origState, err := terminal.GetState(fd) if err != nil { errorExit(err.Error()) } r.origState = origState terminal.MakeRaw(fd) r.updateTerminalSize() initTheme(r.theme, r.defaultTheme(), r.forceBlack) if r.fullscreen { r.smcup() } else { // We assume that --no-clear is used for repetitive relaunching of fzf. // So we do not clear the lower bottom of the screen. if r.clearOnExit { r.csi("J") } y, x := r.findOffset() r.mouse = r.mouse && y >= 0 // When --no-clear is used for repetitive relaunching, there is a small // time frame between fzf processes where the user keystrokes are not // captured by either of fzf process which can cause x offset to be // increased and we're left with unwanted extra new line. if x > 0 && r.clearOnExit { r.upOneLine = true r.makeSpace() } for i := 1; i < r.MaxY(); i++ { r.makeSpace() } } if r.mouse { r.csi("?1000h") } r.csi(fmt.Sprintf("%dA", r.MaxY()-1)) r.csi("G") r.csi("K") if !r.clearOnExit && !r.fullscreen { r.csi("s") } if !r.fullscreen && r.mouse { r.yoffset, _ = r.findOffset() } } func (r *LightRenderer) makeSpace() { r.stderr("\n") r.csi("G") } func (r *LightRenderer) move(y int, x int) { // w.csi("u") if r.y < y { r.csi(fmt.Sprintf("%dB", y-r.y)) } else if r.y > y { r.csi(fmt.Sprintf("%dA", r.y-y)) } r.stderr("\r") if x > 0 { r.csi(fmt.Sprintf("%dC", x)) } r.y = y r.x = x } func (r *LightRenderer) origin() { r.move(0, 0) } func getEnv(name string, defaultValue int) int { env := os.Getenv(name) if len(env) == 0 { return defaultValue } return atoi(env, defaultValue) } func (r *LightRenderer) updateTerminalSize() { width, height, err := terminal.GetSize(r.fd()) if err == nil { r.width = width r.height = r.maxHeightFunc(height) } else { r.width = getEnv("COLUMNS", defaultWidth) r.height = r.maxHeightFunc(getEnv("LINES", defaultHeight)) } } func (r *LightRenderer) getch(nonblock bool) (int, bool) { b := make([]byte, 1) fd := r.fd() util.SetNonblock(r.ttyin, nonblock) _, err := util.Read(fd, b) if err != nil { return 0, false } return int(b[0]), true } func (r *LightRenderer) getBytes() []byte { return r.getBytesInternal(r.buffer, false) } func (r *LightRenderer) getBytesInternal(buffer []byte, nonblock bool) []byte { c, ok := r.getch(nonblock) if !nonblock && !ok { r.Close() errorExit("Failed to read " + consoleDevice) } retries := 0 if c == ESC || nonblock { retries = r.escDelay / escPollInterval } buffer = append(buffer, byte(c)) pc := c for { c, ok = r.getch(true) if !ok { if retries > 0 { retries-- time.Sleep(escPollInterval * time.Millisecond) continue } break } else if c == ESC && pc != c { retries = r.escDelay / escPollInterval } else { retries = 0 } buffer = append(buffer, byte(c)) pc = c } return buffer } func (r *LightRenderer) GetChar() Event { if len(r.buffer) == 0 { r.buffer = r.getBytes() } if len(r.buffer) == 0 { panic("Empty buffer") } sz := 1 defer func() { r.buffer = r.buffer[sz:] }() switch r.buffer[0] { case CtrlC: return Event{CtrlC, 0, nil} case CtrlG: return Event{CtrlG, 0, nil} case CtrlQ: return Event{CtrlQ, 0, nil} case 127: return Event{BSpace, 0, nil} case 0: return Event{CtrlSpace, 0, nil} case 28: return Event{CtrlBackSlash, 0, nil} case 29: return Event{CtrlRightBracket, 0, nil} case 30: return Event{CtrlCaret, 0, nil} case 31: return Event{CtrlSlash, 0, nil} case ESC: ev := r.escSequence(&sz) // Second chance if ev.Type == Invalid { r.buffer = r.getBytes() ev = r.escSequence(&sz) } return ev } // CTRL-A ~ CTRL-Z if r.buffer[0] <= CtrlZ { return Event{int(r.buffer[0]), 0, nil} } char, rsz := utf8.DecodeRune(r.buffer) if char == utf8.RuneError { return Event{ESC, 0, nil} } sz = rsz return Event{Rune, char, nil} } func (r *LightRenderer) escSequence(sz *int) Event { if len(r.buffer) < 2 { return Event{ESC, 0, nil} } *sz = 2 if r.buffer[1] >= 1 && r.buffer[1] <= 'z'-'a'+1 { return Event{int(CtrlAltA + r.buffer[1] - 1), 0, nil} } alt := false if len(r.buffer) > 2 && r.buffer[1] == ESC { r.buffer = r.buffer[1:] alt = true } switch r.buffer[1] { case ESC: return Event{ESC, 0, nil} case 32: return Event{AltSpace, 0, nil} case 47: return Event{AltSlash, 0, nil} case 98: return Event{AltB, 0, nil} case 100: return Event{AltD, 0, nil} case 102: return Event{AltF, 0, nil} case 127: return Event{AltBS, 0, nil} case 91, 79: if len(r.buffer) < 3 { return Event{Invalid, 0, nil} } *sz = 3 switch r.buffer[2] { case 68: if alt { return Event{AltLeft, 0, nil} } return Event{Left, 0, nil} case 67: if alt { // Ugh.. return Event{AltRight, 0, nil} } return Event{Right, 0, nil} case 66: if alt { return Event{AltDown, 0, nil} } return Event{Down, 0, nil} case 65: if alt { return Event{AltUp, 0, nil} } return Event{Up, 0, nil} case 90: return Event{BTab, 0, nil} case 72: return Event{Home, 0, nil} case 70: return Event{End, 0, nil} case 77: return r.mouseSequence(sz) case 80: return Event{F1, 0, nil} case 81: return Event{F2, 0, nil} case 82: return Event{F3, 0, nil} case 83: return Event{F4, 0, nil} case 49, 50, 51, 52, 53, 54: if len(r.buffer) < 4 { return Event{Invalid, 0, nil} } *sz = 4 switch r.buffer[2] { case 50: if len(r.buffer) == 5 && r.buffer[4] == 126 { *sz = 5 switch r.buffer[3] { case 48: return Event{F9, 0, nil} case 49: return Event{F10, 0, nil} case 51: return Event{F11, 0, nil} case 52: return Event{F12, 0, nil} } } // Bracketed paste mode: \e[200~ ... \e[201~ if r.buffer[3] == '0' && (r.buffer[4] == '0' || r.buffer[4] == '1') && r.buffer[5] == '~' { // Immediately discard the sequence from the buffer and reread input r.buffer = r.buffer[6:] *sz = 0 return r.GetChar() } return Event{Invalid, 0, nil} // INS case 51: return Event{Del, 0, nil} case 52: return Event{End, 0, nil} case 53: return Event{PgUp, 0, nil} case 54: return Event{PgDn, 0, nil} case 49: switch r.buffer[3] { case 126: return Event{Home, 0, nil} case 53, 55, 56, 57: if len(r.buffer) == 5 && r.buffer[4] == 126 { *sz = 5 switch r.buffer[3] { case 53: return Event{F5, 0, nil} case 55: return Event{F6, 0, nil} case 56: return Event{F7, 0, nil} case 57: return Event{F8, 0, nil} } } return Event{Invalid, 0, nil} case ';': if len(r.buffer) != 6 { return Event{Invalid, 0, nil} } *sz = 6 switch r.buffer[4] { case '2', '5': switch r.buffer[5] { case 'A': return Event{SUp, 0, nil} case 'B': return Event{SDown, 0, nil} case 'C': return Event{SRight, 0, nil} case 'D': return Event{SLeft, 0, nil} } } // r.buffer[4] } // r.buffer[3] } // r.buffer[2] } // r.buffer[2] } // r.buffer[1] if r.buffer[1] >= 'a' && r.buffer[1] <= 'z' { return Event{AltA + int(r.buffer[1]) - 'a', 0, nil} } if r.buffer[1] >= '0' && r.buffer[1] <= '9' { return Event{Alt0 + int(r.buffer[1]) - '0', 0, nil} } return Event{Invalid, 0, nil} } func (r *LightRenderer) mouseSequence(sz *int) Event { if len(r.buffer) < 6 || !r.mouse { return Event{Invalid, 0, nil} } *sz = 6 switch r.buffer[3] { case 32, 34, 36, 40, 48, // mouse-down / shift / cmd / ctrl 35, 39, 43, 51: // mouse-up / shift / cmd / ctrl mod := r.buffer[3] >= 36 left := r.buffer[3] == 32 down := r.buffer[3]%2 == 0 x := int(r.buffer[4] - 33) y := int(r.buffer[5]-33) - r.yoffset double := false if down { now := time.Now() if !left { // Right double click is not allowed r.clickY = []int{} } else if now.Sub(r.prevDownTime) < doubleClickDuration { r.clickY = append(r.clickY, y) } else { r.clickY = []int{y} } r.prevDownTime = now } else { if len(r.clickY) > 1 && r.clickY[0] == r.clickY[1] && time.Since(r.prevDownTime) < doubleClickDuration { double = true } } return Event{Mouse, 0, &MouseEvent{y, x, 0, left, down, double, mod}} case 96, 100, 104, 112, // scroll-up / shift / cmd / ctrl 97, 101, 105, 113: // scroll-down / shift / cmd / ctrl mod := r.buffer[3] >= 100 s := 1 - int(r.buffer[3]%2)*2 x := int(r.buffer[4] - 33) y := int(r.buffer[5]-33) - r.yoffset return Event{Mouse, 0, &MouseEvent{y, x, s, false, false, false, mod}} } return Event{Invalid, 0, nil} } func (r *LightRenderer) smcup() { r.csi("?1049h") } func (r *LightRenderer) rmcup() { r.csi("?1049l") } func (r *LightRenderer) Pause(clear bool) { terminal.Restore(r.fd(), r.origState) if clear { if r.fullscreen { r.rmcup() } else { r.smcup() r.csi("H") } r.flush() } } func (r *LightRenderer) Resume(clear bool) { terminal.MakeRaw(r.fd()) if clear { if r.fullscreen { r.smcup() } else { r.rmcup() } r.flush() } else if !r.fullscreen && r.mouse { // NOTE: Resume(false) is only called on SIGCONT after SIGSTOP. // And It's highly likely that the offset we obtained at the beginning will // no longer be correct, so we simply disable mouse input. r.csi("?1000l") r.mouse = false } } func (r *LightRenderer) Clear() { if r.fullscreen { r.csi("H") } // r.csi("u") r.origin() r.csi("J") r.flush() } func (r *LightRenderer) RefreshWindows(windows []Window) { r.flush() } func (r *LightRenderer) Refresh() { r.updateTerminalSize() } func (r *LightRenderer) Close() { // r.csi("u") if r.clearOnExit { if r.fullscreen { r.rmcup() } else { r.origin() if r.upOneLine { r.csi("A") } r.csi("J") } } else if !r.fullscreen { r.csi("u") } if r.mouse { r.csi("?1000l") } r.flush() terminal.Restore(r.fd(), r.origState) } func (r *LightRenderer) MaxX() int { return r.width } func (r *LightRenderer) MaxY() int { return r.height } func (r *LightRenderer) DoesAutoWrap() bool { return false } func (r *LightRenderer) NewWindow(top int, left int, width int, height int, borderStyle BorderStyle) Window { w := &LightWindow{ renderer: r, colored: r.theme != nil, border: borderStyle, top: top, left: left, width: width, height: height, tabstop: r.tabstop, fg: colDefault, bg: colDefault} if r.theme != nil { w.fg = r.theme.Fg w.bg = r.theme.Bg } w.drawBorder() return w } func (w *LightWindow) drawBorder() { switch w.border.shape { case BorderAround: w.drawBorderAround() case BorderHorizontal: w.drawBorderHorizontal() } } func (w *LightWindow) drawBorderHorizontal() { w.Move(0, 0) w.CPrint(ColBorder, AttrRegular, repeat(w.border.horizontal, w.width)) w.Move(w.height-1, 0) w.CPrint(ColBorder, AttrRegular, repeat(w.border.horizontal, w.width)) } func (w *LightWindow) drawBorderAround() { w.Move(0, 0) w.CPrint(ColBorder, AttrRegular, string(w.border.topLeft)+repeat(w.border.horizontal, w.width-2)+string(w.border.topRight)) for y := 1; y < w.height-1; y++ { w.Move(y, 0) w.CPrint(ColBorder, AttrRegular, string(w.border.vertical)) w.cprint2(colDefault, w.bg, AttrRegular, repeat(' ', w.width-2)) w.CPrint(ColBorder, AttrRegular, string(w.border.vertical)) } w.Move(w.height-1, 0) w.CPrint(ColBorder, AttrRegular, string(w.border.bottomLeft)+repeat(w.border.horizontal, w.width-2)+string(w.border.bottomRight)) } func (w *LightWindow) csi(code string) { w.renderer.csi(code) } func (w *LightWindow) stderrInternal(str string, allowNLCR bool) { w.renderer.stderrInternal(str, allowNLCR) } func (w *LightWindow) Top() int { return w.top } func (w *LightWindow) Left() int { return w.left } func (w *LightWindow) Width() int { return w.width } func (w *LightWindow) Height() int { return w.height } func (w *LightWindow) Refresh() { } func (w *LightWindow) Close() { } func (w *LightWindow) X() int { return w.posx } func (w *LightWindow) Y() int { return w.posy } func (w *LightWindow) Enclose(y int, x int) bool { return x >= w.left && x < (w.left+w.width) && y >= w.top && y < (w.top+w.height) } func (w *LightWindow) Move(y int, x int) { w.posx = x w.posy = y w.renderer.move(w.Top()+y, w.Left()+x) } func (w *LightWindow) MoveAndClear(y int, x int) { w.Move(y, x) // We should not delete preview window on the right // csi("K") w.Print(repeat(' ', w.width-x)) w.Move(y, x) } func attrCodes(attr Attr) []string { codes := []string{} if (attr & Bold) > 0 { codes = append(codes, "1") } if (attr & Dim) > 0 { codes = append(codes, "2") } if (attr & Italic) > 0 { codes = append(codes, "3") } if (attr & Underline) > 0 { codes = append(codes, "4") } if (attr & Blink) > 0 { codes = append(codes, "5") } if (attr & Reverse) > 0 { codes = append(codes, "7") } return codes } func colorCodes(fg Color, bg Color) []string { codes := []string{} appendCode := func(c Color, offset int) { if c == colDefault { return } if c.is24() { r := (c >> 16) & 0xff g := (c >> 8) & 0xff b := (c) & 0xff codes = append(codes, fmt.Sprintf("%d;2;%d;%d;%d", 38+offset, r, g, b)) } else if c >= colBlack && c <= colWhite { codes = append(codes, fmt.Sprintf("%d", int(c)+30+offset)) } else if c > colWhite && c < 16 { codes = append(codes, fmt.Sprintf("%d", int(c)+90+offset-8)) } else if c >= 16 && c < 256 { codes = append(codes, fmt.Sprintf("%d;5;%d", 38+offset, c)) } } appendCode(fg, 0) appendCode(bg, 10) return codes } func (w *LightWindow) csiColor(fg Color, bg Color, attr Attr) bool { codes := append(attrCodes(attr), colorCodes(fg, bg)...) w.csi(";" + strings.Join(codes, ";") + "m") return len(codes) > 0 } func (w *LightWindow) Print(text string) { w.cprint2(colDefault, w.bg, AttrRegular, text) } func cleanse(str string) string { return strings.Replace(str, "\x1b", "", -1) } func (w *LightWindow) CPrint(pair ColorPair, attr Attr, text string) { if !w.colored { w.csiColor(colDefault, colDefault, attrFor(pair, attr)) } else { w.csiColor(pair.Fg(), pair.Bg(), attr) } w.stderrInternal(cleanse(text), false) w.csi("m") } func (w *LightWindow) cprint2(fg Color, bg Color, attr Attr, text string) { if w.csiColor(fg, bg, attr) { defer w.csi("m") } w.stderrInternal(cleanse(text), false) } type wrappedLine struct { text string displayWidth int } func wrapLine(input string, prefixLength int, max int, tabstop int) []wrappedLine { lines := []wrappedLine{} width := 0 line := "" for _, r := range input { w := util.Max(util.RuneWidth(r, prefixLength+width, 8), 1) width += w str := string(r) if r == '\t' { str = repeat(' ', w) } if prefixLength+width <= max { line += str } else { lines = append(lines, wrappedLine{string(line), width - w}) line = str prefixLength = 0 width = util.RuneWidth(r, prefixLength, 8) } } lines = append(lines, wrappedLine{string(line), width}) return lines } func (w *LightWindow) fill(str string, onMove func()) FillReturn { allLines := strings.Split(str, "\n") for i, line := range allLines { lines := wrapLine(line, w.posx, w.width, w.tabstop) for j, wl := range lines { if w.posx >= w.Width()-1 && wl.displayWidth == 0 { if w.posy < w.height-1 { w.Move(w.posy+1, 0) } return FillNextLine } w.stderrInternal(wl.text, false) w.posx += wl.displayWidth // Wrap line if j < len(lines)-1 || i < len(allLines)-1 { if w.posy+1 >= w.height { return FillSuspend } w.MoveAndClear(w.posy, w.posx) w.Move(w.posy+1, 0) onMove() } } } return FillContinue } func (w *LightWindow) setBg() { if w.bg != colDefault { w.csiColor(colDefault, w.bg, AttrRegular) } } func (w *LightWindow) Fill(text string) FillReturn { w.Move(w.posy, w.posx) w.setBg() return w.fill(text, w.setBg) } func (w *LightWindow) CFill(fg Color, bg Color, attr Attr, text string) FillReturn { w.Move(w.posy, w.posx) if fg == colDefault { fg = w.fg } if bg == colDefault { bg = w.bg } if w.csiColor(fg, bg, attr) { defer w.csi("m") return w.fill(text, func() { w.csiColor(fg, bg, attr) }) } return w.fill(text, w.setBg) } func (w *LightWindow) FinishFill() { w.MoveAndClear(w.posy, w.posx) for y := w.posy + 1; y < w.height; y++ { w.MoveAndClear(y, 0) } } func (w *LightWindow) Erase() { w.drawBorder() // We don't erase the window here to avoid flickering during scroll w.Move(0, 0) }