mirror of
https://github.com/aditya-K2/gspt.git
synced 2026-01-09 13:58:05 -05:00
move files to ui package
This commit is contained in:
aditya-K2
217
ui/iview.go
Normal file
217
ui/iview.go
Normal file
@@ -0,0 +1,217 @@
|
||||
package ui
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"strings"
|
||||
|
||||
"github.com/gdamore/tcell/v2"
|
||||
"github.com/rivo/tview"
|
||||
)
|
||||
|
||||
var (
|
||||
berr = errors.New("Couldn't Get Base Selection in Interactive View")
|
||||
defaultfg = tcell.ColorGreen
|
||||
defaultbg = tcell.ColorDefault
|
||||
defaultstyle = tcell.StyleDefault.
|
||||
Foreground(defaultfg).
|
||||
Background(defaultbg)
|
||||
)
|
||||
|
||||
type _range struct {
|
||||
Start int
|
||||
End int
|
||||
}
|
||||
|
||||
type interactiveView struct {
|
||||
visual bool
|
||||
vrange *_range
|
||||
baseSel int
|
||||
View *tview.Table
|
||||
}
|
||||
|
||||
func NewInteractiveView() *interactiveView {
|
||||
view := tview.NewTable()
|
||||
view.SetSelectable(true, false)
|
||||
view.SetBackgroundColor(tcell.ColorDefault)
|
||||
|
||||
i := &interactiveView{
|
||||
View: view,
|
||||
vrange: &_range{},
|
||||
visual: false,
|
||||
}
|
||||
|
||||
view.SetDrawFunc(func(s tcell.Screen,
|
||||
x, y, width, height int) (int, int, int, int) {
|
||||
i.update()
|
||||
return i.View.GetInnerRect()
|
||||
})
|
||||
view.SetInputCapture(i.capture)
|
||||
return i
|
||||
}
|
||||
|
||||
func (i *interactiveView) exitVisualMode() {
|
||||
if i.vrange.Start < i.baseSel {
|
||||
i.View.Select(i.vrange.Start, -1)
|
||||
} else if i.vrange.End > i.baseSel {
|
||||
i.View.Select(i.vrange.End, -1)
|
||||
}
|
||||
i.baseSel = -1
|
||||
}
|
||||
|
||||
func (i *interactiveView) enterVisualMode() {
|
||||
row, _ := i.View.GetSelection()
|
||||
i.baseSel = row
|
||||
i.vrange.Start, i.vrange.End = row, row
|
||||
}
|
||||
|
||||
func (i *interactiveView) toggleVisualMode() {
|
||||
if i.visual {
|
||||
i.exitVisualMode()
|
||||
} else if !i.visual {
|
||||
i.enterVisualMode()
|
||||
}
|
||||
i.visual = !i.visual
|
||||
}
|
||||
|
||||
func (i *interactiveView) getHandler(s string) func(e *tcell.EventKey) *tcell.EventKey {
|
||||
vr := i.vrange
|
||||
check := func() {
|
||||
if vr.Start <= -1 {
|
||||
vr.Start = 0
|
||||
}
|
||||
if vr.End <= -1 {
|
||||
vr.End = 0
|
||||
}
|
||||
if vr.End >= i.View.GetRowCount() {
|
||||
vr.End = i.View.GetRowCount() - 1
|
||||
}
|
||||
if vr.Start >= i.View.GetRowCount() {
|
||||
vr.Start = i.View.GetRowCount() - 1
|
||||
}
|
||||
}
|
||||
funcMap := map[string]func(e *tcell.EventKey) *tcell.EventKey{
|
||||
"up": func(e *tcell.EventKey) *tcell.EventKey {
|
||||
if i.visual {
|
||||
check()
|
||||
if vr.End > i.baseSel {
|
||||
vr.End--
|
||||
} else if vr.Start <= i.baseSel {
|
||||
vr.Start--
|
||||
}
|
||||
if i.baseSel == -1 {
|
||||
panic(berr)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
return e
|
||||
},
|
||||
"down": func(e *tcell.EventKey) *tcell.EventKey {
|
||||
if i.visual {
|
||||
check()
|
||||
if vr.Start < i.baseSel {
|
||||
vr.Start++
|
||||
} else if vr.Start == i.baseSel {
|
||||
vr.End++
|
||||
}
|
||||
if i.baseSel == -1 {
|
||||
panic(berr)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
return e
|
||||
},
|
||||
"exitvisual": func(e *tcell.EventKey) *tcell.EventKey {
|
||||
if i.visual {
|
||||
i.exitVisualMode()
|
||||
i.visual = false
|
||||
return nil
|
||||
}
|
||||
return e
|
||||
},
|
||||
"top": func(e *tcell.EventKey) *tcell.EventKey {
|
||||
if i.visual {
|
||||
i.vrange.Start = 0
|
||||
i.vrange.End = i.baseSel
|
||||
i.View.ScrollToBeginning()
|
||||
return nil
|
||||
}
|
||||
return e
|
||||
},
|
||||
"bottom": func(e *tcell.EventKey) *tcell.EventKey {
|
||||
if i.visual {
|
||||
i.vrange.Start = i.baseSel
|
||||
i.vrange.End = i.View.GetRowCount() - 1
|
||||
i.View.ScrollToEnd()
|
||||
return nil
|
||||
}
|
||||
return e
|
||||
},
|
||||
}
|
||||
if val, ok := funcMap[s]; ok {
|
||||
return val
|
||||
} else {
|
||||
return nil
|
||||
}
|
||||
}
|
||||
|
||||
func (i *interactiveView) capture(e *tcell.EventKey) *tcell.EventKey {
|
||||
switch e.Rune() {
|
||||
case 'j':
|
||||
{
|
||||
return i.getHandler("down")(e)
|
||||
}
|
||||
case 'k':
|
||||
{
|
||||
return i.getHandler("up")(e)
|
||||
}
|
||||
case 'v':
|
||||
{
|
||||
i.toggleVisualMode()
|
||||
return nil
|
||||
}
|
||||
case 'g':
|
||||
{
|
||||
return i.getHandler("top")(e)
|
||||
}
|
||||
case 'G':
|
||||
{
|
||||
return i.getHandler("bottom")(e)
|
||||
}
|
||||
case 'C':
|
||||
{
|
||||
return i.getHandler("openContextMenu")(e)
|
||||
}
|
||||
default:
|
||||
{
|
||||
if e.Key() == tcell.KeyEscape {
|
||||
return i.getHandler("exitvisual")(e)
|
||||
}
|
||||
return e
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func GetCell(text string, st tcell.Style) *tview.TableCell {
|
||||
return tview.NewTableCell(text).
|
||||
SetAlign(tview.AlignLeft).
|
||||
SetStyle(st)
|
||||
}
|
||||
|
||||
func (i *interactiveView) update() {
|
||||
s := strings.Split("orem ipsum dolor sit amet, consectetur adipiscing elit. Nunc nec leo a tellus gravida convallis. Curabitur tempus purus nisi. Proin non enim convallis augue porta aliquet.", " ")
|
||||
i.View.Clear()
|
||||
for j := range s {
|
||||
b := ""
|
||||
if i.visual && (j >= i.vrange.Start && j <= i.vrange.End) {
|
||||
b = "[blue::]█[::]"
|
||||
}
|
||||
i.View.SetCell(j, 0,
|
||||
GetCell(b, defaultstyle))
|
||||
i.View.SetCell(j, 1,
|
||||
GetCell(s[j], defaultstyle))
|
||||
i.View.SetCell(j, 2,
|
||||
GetCell(s[j], defaultstyle.Foreground(tcell.ColorBlue)))
|
||||
i.View.SetCell(j, 3,
|
||||
GetCell(s[j], defaultstyle.Foreground(tcell.ColorYellow)))
|
||||
}
|
||||
}
|
||||
110
ui/root.go
Normal file
110
ui/root.go
Normal file
@@ -0,0 +1,110 @@
|
||||
package ui
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"github.com/gdamore/tcell/v2"
|
||||
"github.com/rivo/tview"
|
||||
)
|
||||
|
||||
type Main struct {
|
||||
Root *tview.Pages
|
||||
}
|
||||
|
||||
func NewMain() *Main {
|
||||
m := &Main{}
|
||||
|
||||
iv := NewInteractiveView()
|
||||
Root := tview.NewPages()
|
||||
Root.AddPage("iview", iv.View, true, true)
|
||||
|
||||
m.Root = Root
|
||||
return m
|
||||
}
|
||||
|
||||
func (m *Main) OpenContextMenu(title string,
|
||||
ctxContentHandler func() []string, ctxSelectHandler func(s string)) {
|
||||
ctxMenu := tview.NewTable()
|
||||
|
||||
_, _, w, h := m.Root.GetRect()
|
||||
cslice := ctxContentHandler()
|
||||
cwidth := 30
|
||||
cheight := len(cslice) + 3
|
||||
currentTime := time.Now().String()
|
||||
epx := 4
|
||||
closec := make(chan bool)
|
||||
|
||||
closeCtx := func() {
|
||||
m.Root.RemovePage(currentTime)
|
||||
}
|
||||
|
||||
drawCtx := func() {
|
||||
m.Root.AddPage(currentTime, ctxMenu, false, true)
|
||||
ctxMenu.SetRect(w/2-(cwidth/2+epx), (h/2 - (cheight/2 + epx)), cwidth, cheight)
|
||||
}
|
||||
|
||||
redraw := func() {
|
||||
closeCtx()
|
||||
drawCtx()
|
||||
}
|
||||
|
||||
deleteCtx := func() {
|
||||
closeCtx()
|
||||
closec <- true
|
||||
}
|
||||
|
||||
resizeHandler := func() {
|
||||
dur := 500
|
||||
tck := time.NewTicker(time.Duration(dur) * time.Millisecond)
|
||||
go func() {
|
||||
for {
|
||||
select {
|
||||
case <-tck.C:
|
||||
{
|
||||
_, _, _w, _h := m.Root.GetRect()
|
||||
if _w != w || _h != h {
|
||||
w = _w
|
||||
h = _h
|
||||
redraw()
|
||||
}
|
||||
}
|
||||
case <-closec:
|
||||
{
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
resizeHandler()
|
||||
|
||||
ctxMenu.SetBorder(true)
|
||||
ctxMenu.SetSelectable(true, false)
|
||||
capture := func(e *tcell.EventKey) *tcell.EventKey {
|
||||
if e.Key() == tcell.KeyEscape {
|
||||
deleteCtx()
|
||||
return nil
|
||||
} else if e.Key() == tcell.KeyEnter {
|
||||
ctxSelectHandler(
|
||||
ctxMenu.GetCell(
|
||||
ctxMenu.GetSelection()).Text)
|
||||
deleteCtx()
|
||||
return nil
|
||||
}
|
||||
return e
|
||||
}
|
||||
|
||||
ctxMenu.SetInputCapture(capture)
|
||||
|
||||
ctxMenu.SetCell(0, 0,
|
||||
GetCell(title, tcell.StyleDefault.
|
||||
Foreground(tcell.ColorWhite).
|
||||
Bold(true)).SetSelectable(false))
|
||||
for k := range cslice {
|
||||
ctxMenu.SetCell(k+1, 0,
|
||||
GetCell(cslice[k], defaultstyle))
|
||||
}
|
||||
|
||||
drawCtx()
|
||||
}
|
||||
Reference in New Issue
Block a user