mirror of
https://github.com/Llewellynvdm/fzf.git
synced 2024-11-13 08:56:27 +00:00
618 lines
13 KiB
Go
618 lines
13 KiB
Go
// +build tcell windows
|
|
|
|
package tui
|
|
|
|
import (
|
|
"os"
|
|
"time"
|
|
"unicode/utf8"
|
|
|
|
"runtime"
|
|
|
|
"github.com/gdamore/tcell"
|
|
"github.com/gdamore/tcell/encoding"
|
|
|
|
"github.com/mattn/go-runewidth"
|
|
)
|
|
|
|
func HasFullscreenRenderer() bool {
|
|
return true
|
|
}
|
|
|
|
func (p ColorPair) style() tcell.Style {
|
|
style := tcell.StyleDefault
|
|
return style.Foreground(tcell.Color(p.Fg())).Background(tcell.Color(p.Bg()))
|
|
}
|
|
|
|
type Attr tcell.Style
|
|
|
|
type TcellWindow struct {
|
|
color bool
|
|
top int
|
|
left int
|
|
width int
|
|
height int
|
|
normal ColorPair
|
|
lastX int
|
|
lastY int
|
|
moveCursor bool
|
|
borderStyle BorderStyle
|
|
}
|
|
|
|
func (w *TcellWindow) Top() int {
|
|
return w.top
|
|
}
|
|
|
|
func (w *TcellWindow) Left() int {
|
|
return w.left
|
|
}
|
|
|
|
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()
|
|
}
|
|
|
|
func (w *TcellWindow) FinishFill() {
|
|
// NO-OP
|
|
}
|
|
|
|
const (
|
|
Bold Attr = Attr(tcell.AttrBold)
|
|
Dim = Attr(tcell.AttrDim)
|
|
Blink = Attr(tcell.AttrBlink)
|
|
Reverse = Attr(tcell.AttrReverse)
|
|
Underline = Attr(tcell.AttrUnderline)
|
|
Italic = Attr(tcell.AttrNone) // Not supported
|
|
)
|
|
|
|
const (
|
|
AttrRegular Attr = 0
|
|
)
|
|
|
|
func (r *FullscreenRenderer) defaultTheme() *ColorTheme {
|
|
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
|
|
)
|
|
|
|
func (r *FullscreenRenderer) initScreen() {
|
|
s, e := tcell.NewScreen()
|
|
if e != nil {
|
|
errorExit(e.Error())
|
|
}
|
|
if e = s.Init(); e != nil {
|
|
errorExit(e.Error())
|
|
}
|
|
if r.mouse {
|
|
s.EnableMouse()
|
|
} else {
|
|
s.DisableMouse()
|
|
}
|
|
_screen = s
|
|
}
|
|
|
|
func (r *FullscreenRenderer) Init() {
|
|
if os.Getenv("TERM") == "cygwin" {
|
|
os.Setenv("TERM", "")
|
|
}
|
|
encoding.Register()
|
|
|
|
r.initScreen()
|
|
initTheme(r.theme, r.defaultTheme(), r.forceBlack)
|
|
}
|
|
|
|
func (r *FullscreenRenderer) MaxX() int {
|
|
ncols, _ := _screen.Size()
|
|
return int(ncols)
|
|
}
|
|
|
|
func (r *FullscreenRenderer) MaxY() int {
|
|
_, nlines := _screen.Size()
|
|
return int(nlines)
|
|
}
|
|
|
|
func (w *TcellWindow) X() int {
|
|
return w.lastX
|
|
}
|
|
|
|
func (w *TcellWindow) Y() int {
|
|
return w.lastY
|
|
}
|
|
|
|
func (r *FullscreenRenderer) DoesAutoWrap() bool {
|
|
return false
|
|
}
|
|
|
|
func (r *FullscreenRenderer) Clear() {
|
|
_screen.Sync()
|
|
_screen.Clear()
|
|
}
|
|
|
|
func (r *FullscreenRenderer) Refresh() {
|
|
// noop
|
|
}
|
|
|
|
func (r *FullscreenRenderer) GetChar() Event {
|
|
ev := _screen.PollEvent()
|
|
switch ev := ev.(type) {
|
|
case *tcell.EventResize:
|
|
return Event{Resize, 0, nil}
|
|
|
|
// 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}}
|
|
} else if button&tcell.WheelUp != 0 {
|
|
return Event{Mouse, 0, &MouseEvent{y, x, +1, false, false, false, mod}}
|
|
} 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
|
|
double := false
|
|
if down {
|
|
now := time.Now()
|
|
if !left {
|
|
r.clickY = []int{}
|
|
} else if now.Sub(r.prevDownTime) < doubleClickDuration {
|
|
r.clickY = append(r.clickY, x)
|
|
} else {
|
|
r.clickY = []int{x}
|
|
r.prevDownTime = now
|
|
}
|
|
} else {
|
|
if len(r.clickY) > 1 && r.clickY[0] == r.clickY[1] &&
|
|
time.Now().Sub(r.prevDownTime) < doubleClickDuration {
|
|
double = true
|
|
}
|
|
}
|
|
|
|
return Event{Mouse, 0, &MouseEvent{y, x, 0, left, down, double, mod}}
|
|
}
|
|
|
|
// 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)
|
|
}
|
|
switch ev.Key() {
|
|
case tcell.KeyCtrlA:
|
|
return Event{keyfn('a'), 0, nil}
|
|
case tcell.KeyCtrlB:
|
|
return Event{keyfn('b'), 0, nil}
|
|
case tcell.KeyCtrlC:
|
|
return Event{keyfn('c'), 0, nil}
|
|
case tcell.KeyCtrlD:
|
|
return Event{keyfn('d'), 0, nil}
|
|
case tcell.KeyCtrlE:
|
|
return Event{keyfn('e'), 0, nil}
|
|
case tcell.KeyCtrlF:
|
|
return Event{keyfn('f'), 0, nil}
|
|
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}
|
|
case tcell.KeyCtrlJ:
|
|
return Event{keyfn('j'), 0, nil}
|
|
case tcell.KeyCtrlK:
|
|
return Event{keyfn('k'), 0, nil}
|
|
case tcell.KeyCtrlL:
|
|
return Event{keyfn('l'), 0, nil}
|
|
case tcell.KeyCtrlM:
|
|
return Event{keyfn('m'), 0, nil}
|
|
case tcell.KeyCtrlN:
|
|
return Event{keyfn('n'), 0, nil}
|
|
case tcell.KeyCtrlO:
|
|
return Event{keyfn('o'), 0, nil}
|
|
case tcell.KeyCtrlP:
|
|
return Event{keyfn('p'), 0, nil}
|
|
case tcell.KeyCtrlQ:
|
|
return Event{keyfn('q'), 0, nil}
|
|
case tcell.KeyCtrlR:
|
|
return Event{keyfn('r'), 0, nil}
|
|
case tcell.KeyCtrlS:
|
|
return Event{keyfn('s'), 0, nil}
|
|
case tcell.KeyCtrlT:
|
|
return Event{keyfn('t'), 0, nil}
|
|
case tcell.KeyCtrlU:
|
|
return Event{keyfn('u'), 0, nil}
|
|
case tcell.KeyCtrlV:
|
|
return Event{keyfn('v'), 0, nil}
|
|
case tcell.KeyCtrlW:
|
|
return Event{keyfn('w'), 0, nil}
|
|
case tcell.KeyCtrlX:
|
|
return Event{keyfn('x'), 0, nil}
|
|
case tcell.KeyCtrlY:
|
|
return Event{keyfn('y'), 0, nil}
|
|
case tcell.KeyCtrlZ:
|
|
return Event{keyfn('z'), 0, nil}
|
|
case tcell.KeyCtrlSpace:
|
|
return Event{CtrlSpace, 0, nil}
|
|
case tcell.KeyCtrlBackslash:
|
|
return Event{CtrlBackSlash, 0, nil}
|
|
case tcell.KeyCtrlRightSq:
|
|
return Event{CtrlRightBracket, 0, nil}
|
|
case tcell.KeyCtrlUnderscore:
|
|
return Event{CtrlSlash, 0, nil}
|
|
case tcell.KeyBackspace2:
|
|
if alt {
|
|
return Event{AltBS, 0, nil}
|
|
}
|
|
return Event{BSpace, 0, nil}
|
|
|
|
case tcell.KeyUp:
|
|
if alt {
|
|
return Event{AltUp, 0, nil}
|
|
}
|
|
return Event{Up, 0, nil}
|
|
case tcell.KeyDown:
|
|
if alt {
|
|
return Event{AltDown, 0, nil}
|
|
}
|
|
return Event{Down, 0, nil}
|
|
case tcell.KeyLeft:
|
|
if alt {
|
|
return Event{AltLeft, 0, nil}
|
|
}
|
|
return Event{Left, 0, nil}
|
|
case tcell.KeyRight:
|
|
if alt {
|
|
return Event{AltRight, 0, nil}
|
|
}
|
|
return Event{Right, 0, nil}
|
|
|
|
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:
|
|
return Event{PgUp, 0, nil}
|
|
case tcell.KeyPgDn:
|
|
return Event{PgDn, 0, nil}
|
|
|
|
case tcell.KeyBacktab:
|
|
return Event{BTab, 0, nil}
|
|
|
|
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}
|
|
case tcell.KeyF12:
|
|
return Event{F12, 0, nil}
|
|
|
|
// 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}
|
|
|
|
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()
|
|
}
|
|
}
|
|
|
|
func (r *FullscreenRenderer) Close() {
|
|
_screen.Fini()
|
|
}
|
|
|
|
func (r *FullscreenRenderer) RefreshWindows(windows []Window) {
|
|
// TODO
|
|
for _, w := range windows {
|
|
w.Refresh()
|
|
}
|
|
_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
|
|
}
|
|
return &TcellWindow{
|
|
color: r.theme != nil,
|
|
top: top,
|
|
left: left,
|
|
width: width,
|
|
height: height,
|
|
normal: normal,
|
|
borderStyle: borderStyle}
|
|
}
|
|
|
|
func (w *TcellWindow) Close() {
|
|
// TODO
|
|
}
|
|
|
|
func fill(x, y, w, h int, n ColorPair, r rune) {
|
|
for ly := 0; ly <= h; ly++ {
|
|
for lx := 0; lx <= w; lx++ {
|
|
_screen.SetContent(x+lx, y+ly, r, nil, n.style())
|
|
}
|
|
}
|
|
}
|
|
|
|
func (w *TcellWindow) Erase() {
|
|
fill(w.left-1, w.top, w.width+1, w.height, w.normal, ' ')
|
|
}
|
|
|
|
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)
|
|
}
|
|
|
|
func (w *TcellWindow) Move(y int, x int) {
|
|
w.lastX = x
|
|
w.lastY = y
|
|
w.moveCursor = true
|
|
}
|
|
|
|
func (w *TcellWindow) MoveAndClear(y int, x int) {
|
|
w.Move(y, x)
|
|
for i := w.lastX; i < w.width; i++ {
|
|
_screen.SetContent(i+w.left, w.lastY+w.top, rune(' '), nil, w.normal.style())
|
|
}
|
|
w.lastX = x
|
|
}
|
|
|
|
func (w *TcellWindow) Print(text string) {
|
|
w.printString(text, w.normal, 0)
|
|
}
|
|
|
|
func (w *TcellWindow) printString(text string, pair ColorPair, a Attr) {
|
|
t := text
|
|
lx := 0
|
|
|
|
var style tcell.Style
|
|
if w.color {
|
|
style = pair.style().
|
|
Reverse(a&Attr(tcell.AttrReverse) != 0).
|
|
Underline(a&Attr(tcell.AttrUnderline) != 0)
|
|
} else {
|
|
style = w.normal.style().
|
|
Reverse(a&Attr(tcell.AttrReverse) != 0 || pair == ColCurrent || pair == ColCurrentMatch).
|
|
Underline(a&Attr(tcell.AttrUnderline) != 0 || pair == ColMatch || pair == ColCurrentMatch)
|
|
}
|
|
style = style.
|
|
Blink(a&Attr(tcell.AttrBlink) != 0).
|
|
Bold(a&Attr(tcell.AttrBold) != 0).
|
|
Dim(a&Attr(tcell.AttrDim) != 0)
|
|
|
|
for {
|
|
if len(t) == 0 {
|
|
break
|
|
}
|
|
r, size := utf8.DecodeRuneInString(t)
|
|
t = t[size:]
|
|
|
|
if r < rune(' ') { // ignore control characters
|
|
continue
|
|
}
|
|
|
|
if r == '\n' {
|
|
w.lastY++
|
|
lx = 0
|
|
} else {
|
|
|
|
if r == '\u000D' { // skip carriage return
|
|
continue
|
|
}
|
|
|
|
var xPos = w.left + w.lastX + lx
|
|
var yPos = w.top + w.lastY
|
|
if xPos < (w.left+w.width) && yPos < (w.top+w.height) {
|
|
_screen.SetContent(xPos, yPos, r, nil, style)
|
|
}
|
|
lx += runewidth.RuneWidth(r)
|
|
}
|
|
}
|
|
w.lastX += lx
|
|
}
|
|
|
|
func (w *TcellWindow) CPrint(pair ColorPair, attr Attr, text string) {
|
|
w.printString(text, pair, attr)
|
|
}
|
|
|
|
func (w *TcellWindow) fillString(text string, pair ColorPair, a Attr) FillReturn {
|
|
lx := 0
|
|
|
|
var style tcell.Style
|
|
if w.color {
|
|
style = pair.style()
|
|
} else {
|
|
style = w.normal.style()
|
|
}
|
|
style = style.
|
|
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' {
|
|
w.lastY++
|
|
w.lastX = 0
|
|
lx = 0
|
|
} else {
|
|
var xPos = w.left + w.lastX + lx
|
|
|
|
// word wrap:
|
|
if xPos >= (w.left + w.width) {
|
|
w.lastY++
|
|
w.lastX = 0
|
|
lx = 0
|
|
xPos = w.left
|
|
}
|
|
var yPos = w.top + w.lastY
|
|
|
|
if yPos >= (w.top + w.height) {
|
|
return FillSuspend
|
|
}
|
|
|
|
_screen.SetContent(xPos, yPos, r, nil, style)
|
|
lx += runewidth.RuneWidth(r)
|
|
}
|
|
}
|
|
w.lastX += lx
|
|
|
|
return FillContinue
|
|
}
|
|
|
|
func (w *TcellWindow) Fill(str string) FillReturn {
|
|
return w.fillString(str, w.normal, 0)
|
|
}
|
|
|
|
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)
|
|
}
|
|
|
|
func (w *TcellWindow) drawBorder() {
|
|
if w.borderStyle.shape == BorderNone {
|
|
return
|
|
}
|
|
|
|
left := w.left
|
|
right := left + w.width
|
|
top := w.top
|
|
bot := top + w.height
|
|
|
|
var style tcell.Style
|
|
if w.color {
|
|
if w.borderStyle.shape == BorderAround {
|
|
style = ColPreviewBorder.style()
|
|
} else {
|
|
style = ColBorder.style()
|
|
}
|
|
} else {
|
|
style = w.normal.style()
|
|
}
|
|
|
|
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)
|
|
}
|
|
|
|
if w.borderStyle.shape == BorderAround {
|
|
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)
|
|
}
|
|
|
|
_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)
|
|
}
|
|
}
|