70 lines
1.2 KiB
Go
70 lines
1.2 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/gdamore/tcell"
|
||
|
|
runewidth "github.com/mattn/go-runewidth"
|
||
|
|
"github.com/rivo/tview"
|
||
|
|
)
|
||
|
|
|
||
|
|
type menuItem struct {
|
||
|
|
Text string
|
||
|
|
Shortcut tcell.Key
|
||
|
|
Selected func()
|
||
|
|
}
|
||
|
|
|
||
|
|
type Menu struct {
|
||
|
|
*tview.Box
|
||
|
|
|
||
|
|
Items []menuItem
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewMenu(app *tview.Application) *Menu {
|
||
|
|
m := &Menu{
|
||
|
|
Box: tview.NewBox(),
|
||
|
|
}
|
||
|
|
|
||
|
|
app.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
|
||
|
|
for _, i := range m.Items {
|
||
|
|
if event.Key() == i.Shortcut {
|
||
|
|
if i.Selected != nil {
|
||
|
|
i.Selected()
|
||
|
|
}
|
||
|
|
break
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return event
|
||
|
|
})
|
||
|
|
|
||
|
|
return m
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *Menu) Draw(screen tcell.Screen) {
|
||
|
|
x, y, width, _ := m.GetInnerRect()
|
||
|
|
x++
|
||
|
|
|
||
|
|
for _, i := range m.Items {
|
||
|
|
tview.Print(screen, tcell.KeyNames[i.Shortcut], x, y, width-2, tview.AlignLeft, tcell.ColorYellow)
|
||
|
|
x += 3
|
||
|
|
|
||
|
|
tview.Print(screen, i.Text, x, y, width, tview.AlignLeft, tcell.ColorTeal)
|
||
|
|
x += runewidth.StringWidth(i.Text) + 2
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *Menu) AddItem(text string, shortcut tcell.Key, selected func()) *Menu {
|
||
|
|
m.Items = append(m.Items, menuItem{
|
||
|
|
Text: text,
|
||
|
|
Shortcut: shortcut,
|
||
|
|
Selected: selected,
|
||
|
|
})
|
||
|
|
|
||
|
|
return m
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *Menu) Clear() *Menu {
|
||
|
|
m.Items = nil
|
||
|
|
|
||
|
|
return m
|
||
|
|
}
|