fzf/src/tui/tcell.go

622 lines
13 KiB
Go
Raw Normal View History

2016-10-24 03:45:45 +00:00
// +build tcell windows
package tui
import (
"os"
2016-10-24 03:45:45 +00:00
"time"
"unicode/utf8"
"runtime"
2017-06-01 23:27:17 +00:00
"github.com/gdamore/tcell"
"github.com/gdamore/tcell/encoding"
2017-06-01 23:27:17 +00:00
"github.com/mattn/go-runewidth"
2016-10-24 03:45:45 +00:00
)
2017-01-22 05:13:37 +00:00
func HasFullscreenRenderer() bool {
return true
}
2017-01-07 16:30:31 +00:00
func (p ColorPair) style() tcell.Style {
style := tcell.StyleDefault
return style.Foreground(tcell.Color(p.Fg())).Background(tcell.Color(p.Bg()))
}
2016-10-24 03:45:45 +00:00
2017-01-07 16:30:31 +00:00
type Attr tcell.Style
type TcellWindow struct {
color bool
preview bool
top int
left int
width int
height int
normal ColorPair
lastX int
lastY int
moveCursor bool
borderStyle BorderStyle
2016-10-24 03:45:45 +00:00
}
2017-01-07 16:30:31 +00:00
func (w *TcellWindow) Top() int {
return w.top
2016-10-24 03:45:45 +00:00
}
2017-01-07 16:30:31 +00:00
func (w *TcellWindow) Left() int {
return w.left
2016-10-24 03:45:45 +00:00
}
2017-01-07 16:30:31 +00:00
func (w *TcellWindow) Width() int {
return w.width
}
func (w *TcellWindow) Height() int {
return w.height
}
func (w *TcellWindow) Refresh() {
if w.moveCursor {
_screen.ShowCursor(w.left+w.lastX, w.top+w.lastY)
w.moveCursor = false
}
w.lastX = 0
w.lastY = 0
w.drawBorder()
2017-01-07 16:30:31 +00:00
}
2016-10-24 03:45:45 +00:00
2017-01-07 16:30:31 +00:00
func (w *TcellWindow) FinishFill() {
// NO-OP
2016-10-24 03:45:45 +00:00
}
const (
2016-11-15 14:57:32 +00:00
Bold Attr = Attr(tcell.AttrBold)
Dim = Attr(tcell.AttrDim)
Blink = Attr(tcell.AttrBlink)
Reverse = Attr(tcell.AttrReverse)
Underline = Attr(tcell.AttrUnderline)
2016-11-23 15:13:10 +00:00
Italic = Attr(tcell.AttrNone) // Not supported
2016-10-24 03:45:45 +00:00
)
const (
AttrRegular Attr = 0
)
2017-01-07 16:30:31 +00:00
func (r *FullscreenRenderer) defaultTheme() *ColorTheme {
2016-10-24 03:45:45 +00:00
if _screen.Colors() >= 256 {
return Dark256
}
return Default16
}
var (
_colorToAttribute = []tcell.Color{
tcell.ColorBlack,
tcell.ColorRed,
tcell.ColorGreen,
tcell.ColorYellow,
tcell.ColorBlue,
tcell.ColorDarkMagenta,
tcell.ColorLightCyan,
tcell.ColorWhite,
}
)
func (c Color) Style() tcell.Color {
if c <= colDefault {
return tcell.ColorDefault
} else if c >= colBlack && c <= colWhite {
return _colorToAttribute[int(c)]
} else {
return tcell.Color(c)
}
}
func (a Attr) Merge(b Attr) Attr {
return a | b
}
var (
_screen tcell.Screen
)
2017-01-07 16:30:31 +00:00
func (r *FullscreenRenderer) initScreen() {
2016-10-24 03:45:45 +00:00
s, e := tcell.NewScreen()
if e != nil {
errorExit(e.Error())
2016-10-24 03:45:45 +00:00
}
if e = s.Init(); e != nil {
errorExit(e.Error())
}
2017-01-07 16:30:31 +00:00
if r.mouse {
s.EnableMouse()
} else {
s.DisableMouse()
2016-10-24 03:45:45 +00:00
}
_screen = s
}
2017-01-07 16:30:31 +00:00
func (r *FullscreenRenderer) Init() {
if os.Getenv("TERM") == "cygwin" {
os.Setenv("TERM", "")
}
encoding.Register()
2017-01-07 16:30:31 +00:00
r.initScreen()
initTheme(r.theme, r.defaultTheme(), r.forceBlack)
}
2016-10-24 03:45:45 +00:00
2017-01-07 16:30:31 +00:00
func (r *FullscreenRenderer) MaxX() int {
2016-10-24 03:45:45 +00:00
ncols, _ := _screen.Size()
return int(ncols)
}
2017-01-07 16:30:31 +00:00
func (r *FullscreenRenderer) MaxY() int {
2016-10-24 03:45:45 +00:00
_, nlines := _screen.Size()
return int(nlines)
}
2017-01-07 16:30:31 +00:00
func (w *TcellWindow) X() int {
return w.lastX
}
func (w *TcellWindow) Y() int {
return w.lastY
}
2017-01-07 16:30:31 +00:00
func (r *FullscreenRenderer) DoesAutoWrap() bool {
return false
}
2017-01-07 16:30:31 +00:00
func (r *FullscreenRenderer) Clear() {
_screen.Sync()
2016-10-24 03:45:45 +00:00
_screen.Clear()
}
2017-01-07 16:30:31 +00:00
func (r *FullscreenRenderer) Refresh() {
2016-10-24 03:45:45 +00:00
// noop
}
2017-01-07 16:30:31 +00:00
func (r *FullscreenRenderer) GetChar() Event {
2016-10-24 03:45:45 +00:00
ev := _screen.PollEvent()
switch ev := ev.(type) {
case *tcell.EventResize:
return Event{Resize, 0, nil}
2016-10-24 03:45:45 +00:00
// process mouse events:
case *tcell.EventMouse:
x, y := ev.Position()
button := ev.Buttons()
mod := ev.Modifiers() != 0
if button&tcell.WheelDown != 0 {
return Event{Mouse, 0, &MouseEvent{y, x, -1, false, false, false, mod}}
2016-10-24 03:45:45 +00:00
} else if button&tcell.WheelUp != 0 {
return Event{Mouse, 0, &MouseEvent{y, x, +1, false, false, false, mod}}
2016-10-24 03:45:45 +00:00
} else if runtime.GOOS != "windows" {
// double and single taps on Windows don't quite work due to
// the console acting on the events and not allowing us
// to consume them.
left := button&tcell.Button1 != 0
down := left || button&tcell.Button3 != 0
2016-10-24 03:45:45 +00:00
double := false
if down {
now := time.Now()
if !left {
r.clickY = []int{}
} else if now.Sub(r.prevDownTime) < doubleClickDuration {
2017-01-07 16:30:31 +00:00
r.clickY = append(r.clickY, x)
2016-10-24 03:45:45 +00:00
} else {
2017-01-07 16:30:31 +00:00
r.clickY = []int{x}
r.prevDownTime = now
2016-10-24 03:45:45 +00:00
}
} else {
2017-01-07 16:30:31 +00:00
if len(r.clickY) > 1 && r.clickY[0] == r.clickY[1] &&
time.Now().Sub(r.prevDownTime) < doubleClickDuration {
2016-10-24 03:45:45 +00:00
double = true
}
}
return Event{Mouse, 0, &MouseEvent{y, x, 0, left, down, double, mod}}
2016-10-24 03:45:45 +00:00
}
// process keyboard:
case *tcell.EventKey:
alt := (ev.Modifiers() & tcell.ModAlt) > 0
keyfn := func(r rune) int {
if alt {
return CtrlAltA - 'a' + int(r)
}
return CtrlA - 'a' + int(r)
}
2016-10-24 03:45:45 +00:00
switch ev.Key() {
case tcell.KeyCtrlA:
return Event{keyfn('a'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlB:
return Event{keyfn('b'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlC:
return Event{keyfn('c'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlD:
return Event{keyfn('d'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlE:
return Event{keyfn('e'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlF:
return Event{keyfn('f'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlG:
return Event{keyfn('g'), 0, nil}
case tcell.KeyCtrlH:
return Event{keyfn('h'), 0, nil}
case tcell.KeyCtrlI:
return Event{keyfn('i'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlJ:
return Event{keyfn('j'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlK:
return Event{keyfn('k'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlL:
return Event{keyfn('l'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlM:
return Event{keyfn('m'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlN:
return Event{keyfn('n'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlO:
return Event{keyfn('o'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlP:
return Event{keyfn('p'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlQ:
return Event{keyfn('q'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlR:
return Event{keyfn('r'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlS:
return Event{keyfn('s'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlT:
return Event{keyfn('t'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlU:
return Event{keyfn('u'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlV:
return Event{keyfn('v'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlW:
return Event{keyfn('w'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlX:
return Event{keyfn('x'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlY:
return Event{keyfn('y'), 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyCtrlZ:
return Event{keyfn('z'), 0, nil}
case tcell.KeyCtrlSpace:
return Event{CtrlSpace, 0, nil}
2019-11-14 13:39:25 +00:00
case tcell.KeyCtrlBackslash:
return Event{CtrlBackSlash, 0, nil}
case tcell.KeyCtrlRightSq:
return Event{CtrlRightBracket, 0, nil}
case tcell.KeyCtrlUnderscore:
return Event{CtrlSlash, 0, nil}
2017-04-28 13:58:08 +00:00
case tcell.KeyBackspace2:
if alt {
return Event{AltBS, 0, nil}
}
2016-10-24 03:45:45 +00:00
return Event{BSpace, 0, nil}
case tcell.KeyUp:
if alt {
return Event{AltUp, 0, nil}
}
2016-10-24 03:45:45 +00:00
return Event{Up, 0, nil}
case tcell.KeyDown:
if alt {
return Event{AltDown, 0, nil}
}
2016-10-24 03:45:45 +00:00
return Event{Down, 0, nil}
case tcell.KeyLeft:
if alt {
return Event{AltLeft, 0, nil}
}
2016-10-24 03:45:45 +00:00
return Event{Left, 0, nil}
case tcell.KeyRight:
if alt {
return Event{AltRight, 0, nil}
}
2016-10-24 03:45:45 +00:00
return Event{Right, 0, nil}
case tcell.KeyInsert:
return Event{Insert, 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyHome:
return Event{Home, 0, nil}
case tcell.KeyDelete:
return Event{Del, 0, nil}
case tcell.KeyEnd:
return Event{End, 0, nil}
case tcell.KeyPgUp:
2016-10-24 03:45:45 +00:00
return Event{PgUp, 0, nil}
case tcell.KeyPgDn:
return Event{PgDn, 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyBacktab:
return Event{BTab, 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyF1:
return Event{F1, 0, nil}
case tcell.KeyF2:
return Event{F2, 0, nil}
case tcell.KeyF3:
return Event{F3, 0, nil}
case tcell.KeyF4:
return Event{F4, 0, nil}
case tcell.KeyF5:
return Event{F5, 0, nil}
case tcell.KeyF6:
return Event{F6, 0, nil}
case tcell.KeyF7:
return Event{F7, 0, nil}
case tcell.KeyF8:
return Event{F8, 0, nil}
case tcell.KeyF9:
return Event{F9, 0, nil}
case tcell.KeyF10:
return Event{F10, 0, nil}
case tcell.KeyF11:
return Event{F11, 0, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyF12:
return Event{F12, 0, nil}
2016-10-24 03:45:45 +00:00
// ev.Ch doesn't work for some reason for space:
case tcell.KeyRune:
r := ev.Rune()
if alt {
switch r {
case ' ':
return Event{AltSpace, 0, nil}
case '/':
return Event{AltSlash, 0, nil}
}
if r >= 'a' && r <= 'z' {
return Event{AltA + int(r) - 'a', 0, nil}
}
if r >= '0' && r <= '9' {
return Event{Alt0 + int(r) - '0', 0, nil}
}
}
return Event{Rune, r, nil}
2016-10-24 03:45:45 +00:00
case tcell.KeyEsc:
return Event{ESC, 0, nil}
}
}
return Event{Invalid, 0, nil}
}
func (r *FullscreenRenderer) Pause(clear bool) {
if clear {
_screen.Fini()
}
}
func (r *FullscreenRenderer) Resume(clear bool) {
if clear {
r.initScreen()
}
2016-10-24 03:45:45 +00:00
}
2017-01-07 16:30:31 +00:00
func (r *FullscreenRenderer) Close() {
2016-10-24 03:45:45 +00:00
_screen.Fini()
}
2017-01-07 16:30:31 +00:00
func (r *FullscreenRenderer) RefreshWindows(windows []Window) {
2016-10-24 03:45:45 +00:00
// TODO
for _, w := range windows {
2017-01-07 16:30:31 +00:00
w.Refresh()
2016-10-24 03:45:45 +00:00
}
_screen.Show()
}
func (r *FullscreenRenderer) NewWindow(top int, left int, width int, height int, preview bool, borderStyle BorderStyle) Window {
normal := ColNormal
if preview {
normal = ColPreview
}
2017-01-07 16:30:31 +00:00
return &TcellWindow{
color: r.theme != nil,
preview: preview,
top: top,
left: left,
width: width,
height: height,
normal: normal,
borderStyle: borderStyle}
2016-10-24 03:45:45 +00:00
}
2017-01-07 16:30:31 +00:00
func (w *TcellWindow) Close() {
2016-10-24 03:45:45 +00:00
// TODO
}
func fill(x, y, w, h int, n ColorPair, r rune) {
2016-10-24 03:45:45 +00:00
for ly := 0; ly <= h; ly++ {
for lx := 0; lx <= w; lx++ {
_screen.SetContent(x+lx, y+ly, r, nil, n.style())
2016-10-24 03:45:45 +00:00
}
}
}
2017-01-07 16:30:31 +00:00
func (w *TcellWindow) Erase() {
fill(w.left-1, w.top, w.width+1, w.height, w.normal, ' ')
2016-10-24 03:45:45 +00:00
}
2017-01-07 16:30:31 +00:00
func (w *TcellWindow) Enclose(y int, x int) bool {
return x >= w.left && x < (w.left+w.width) &&
y >= w.top && y < (w.top+w.height)
2016-10-24 03:45:45 +00:00
}
2017-01-07 16:30:31 +00:00
func (w *TcellWindow) Move(y int, x int) {
w.lastX = x
w.lastY = y
w.moveCursor = true
2016-10-24 03:45:45 +00:00
}
2017-01-07 16:30:31 +00:00
func (w *TcellWindow) MoveAndClear(y int, x int) {
2016-10-24 03:45:45 +00:00
w.Move(y, x)
2017-01-07 16:30:31 +00:00
for i := w.lastX; i < w.width; i++ {
_screen.SetContent(i+w.left, w.lastY+w.top, rune(' '), nil, w.normal.style())
2016-10-24 03:45:45 +00:00
}
2017-01-07 16:30:31 +00:00
w.lastX = x
2016-10-24 03:45:45 +00:00
}
2017-01-07 16:30:31 +00:00
func (w *TcellWindow) Print(text string) {
w.printString(text, w.normal, 0)
2016-10-24 03:45:45 +00:00
}
2017-01-07 16:30:31 +00:00
func (w *TcellWindow) printString(text string, pair ColorPair, a Attr) {
2016-10-24 03:45:45 +00:00
t := text
lx := 0
2016-11-08 16:45:06 +00:00
var style tcell.Style
2017-01-07 16:30:31 +00:00
if w.color {
2016-11-08 16:45:06 +00:00
style = pair.style().
Reverse(a&Attr(tcell.AttrReverse) != 0).
Underline(a&Attr(tcell.AttrUnderline) != 0)
} else {
style = w.normal.style().
2016-11-08 16:45:06 +00:00
Reverse(a&Attr(tcell.AttrReverse) != 0 || pair == ColCurrent || pair == ColCurrentMatch).
Underline(a&Attr(tcell.AttrUnderline) != 0 || pair == ColMatch || pair == ColCurrentMatch)
}
style = style.
2016-10-24 03:45:45 +00:00
Blink(a&Attr(tcell.AttrBlink) != 0).
Bold(a&Attr(tcell.AttrBold) != 0).
2016-11-08 16:45:06 +00:00
Dim(a&Attr(tcell.AttrDim) != 0)
2016-10-24 03:45:45 +00:00
for {
if len(t) == 0 {
break
}
r, size := utf8.DecodeRuneInString(t)
t = t[size:]
if r < rune(' ') { // ignore control characters
continue
}
if r == '\n' {
2017-01-07 16:30:31 +00:00
w.lastY++
2016-10-24 03:45:45 +00:00
lx = 0
} else {
if r == '\u000D' { // skip carriage return
continue
}
2017-01-07 16:30:31 +00:00
var xPos = w.left + w.lastX + lx
var yPos = w.top + w.lastY
if xPos < (w.left+w.width) && yPos < (w.top+w.height) {
2016-10-24 03:45:45 +00:00
_screen.SetContent(xPos, yPos, r, nil, style)
}
lx += runewidth.RuneWidth(r)
2016-10-24 03:45:45 +00:00
}
}
2017-01-07 16:30:31 +00:00
w.lastX += lx
2016-10-24 03:45:45 +00:00
}
2017-01-07 16:30:31 +00:00
func (w *TcellWindow) CPrint(pair ColorPair, attr Attr, text string) {
w.printString(text, pair, attr)
2016-10-24 03:45:45 +00:00
}
func (w *TcellWindow) fillString(text string, pair ColorPair, a Attr) FillReturn {
2016-10-24 03:45:45 +00:00
lx := 0
2016-11-08 16:45:06 +00:00
var style tcell.Style
2017-01-07 16:30:31 +00:00
if w.color {
2016-11-08 16:45:06 +00:00
style = pair.style()
} else {
style = w.normal.style()
2016-11-08 16:45:06 +00:00
}
style = style.
2016-10-24 03:45:45 +00:00
Blink(a&Attr(tcell.AttrBlink) != 0).
Bold(a&Attr(tcell.AttrBold) != 0).
Dim(a&Attr(tcell.AttrDim) != 0).
Reverse(a&Attr(tcell.AttrReverse) != 0).
Underline(a&Attr(tcell.AttrUnderline) != 0)
for _, r := range text {
if r == '\n' {
2017-01-07 16:30:31 +00:00
w.lastY++
w.lastX = 0
2016-10-24 03:45:45 +00:00
lx = 0
} else {
2017-01-07 16:30:31 +00:00
var xPos = w.left + w.lastX + lx
2016-10-24 03:45:45 +00:00
// word wrap:
2017-01-07 16:30:31 +00:00
if xPos >= (w.left + w.width) {
w.lastY++
w.lastX = 0
2016-10-24 03:45:45 +00:00
lx = 0
2017-01-07 16:30:31 +00:00
xPos = w.left
2016-10-24 03:45:45 +00:00
}
2017-01-07 16:30:31 +00:00
var yPos = w.top + w.lastY
2016-10-24 03:45:45 +00:00
2017-01-07 16:30:31 +00:00
if yPos >= (w.top + w.height) {
return FillSuspend
2016-10-24 03:45:45 +00:00
}
_screen.SetContent(xPos, yPos, r, nil, style)
lx += runewidth.RuneWidth(r)
2016-10-24 03:45:45 +00:00
}
}
2017-01-07 16:30:31 +00:00
w.lastX += lx
2016-10-24 03:45:45 +00:00
return FillContinue
2016-10-24 03:45:45 +00:00
}
func (w *TcellWindow) Fill(str string) FillReturn {
return w.fillString(str, w.normal, 0)
2016-10-24 03:45:45 +00:00
}
func (w *TcellWindow) CFill(fg Color, bg Color, a Attr, str string) FillReturn {
if fg == colDefault {
fg = w.normal.Fg()
}
if bg == colDefault {
bg = w.normal.Bg()
}
return w.fillString(str, NewColorPair(fg, bg), a)
2016-10-24 03:45:45 +00:00
}
func (w *TcellWindow) drawBorder() {
if w.borderStyle.shape == BorderNone {
return
}
2017-01-07 16:30:31 +00:00
left := w.left
right := left + w.width
top := w.top
bot := top + w.height
2016-10-24 03:45:45 +00:00
2016-11-08 16:45:06 +00:00
var style tcell.Style
2017-01-07 16:30:31 +00:00
if w.color {
if w.preview {
style = ColPreviewBorder.style()
} else {
style = ColBorder.style()
}
2016-11-08 16:45:06 +00:00
} else {
style = w.normal.style()
2016-11-08 16:45:06 +00:00
}
2016-10-24 03:45:45 +00:00
for x := left; x < right; x++ {
_screen.SetContent(x, top, w.borderStyle.horizontal, nil, style)
_screen.SetContent(x, bot-1, w.borderStyle.horizontal, nil, style)
2016-10-24 03:45:45 +00:00
}
if w.borderStyle.shape != BorderHorizontal {
for y := top; y < bot; y++ {
_screen.SetContent(left, y, w.borderStyle.vertical, nil, style)
_screen.SetContent(right-1, y, w.borderStyle.vertical, nil, style)
}
2016-10-24 03:45:45 +00:00
_screen.SetContent(left, top, w.borderStyle.topLeft, nil, style)
_screen.SetContent(right-1, top, w.borderStyle.topRight, nil, style)
_screen.SetContent(left, bot-1, w.borderStyle.bottomLeft, nil, style)
_screen.SetContent(right-1, bot-1, w.borderStyle.bottomRight, nil, style)
}
2016-10-24 03:45:45 +00:00
}