mirror of
https://github.com/bcicen/ctop.git
synced 2024-08-30 18:23:19 +00:00
116 lines
2.1 KiB
Go
116 lines
2.1 KiB
Go
package widgets
|
|
|
|
import (
|
|
ui "github.com/gizak/termui"
|
|
)
|
|
|
|
type Padding [2]int // x,y padding
|
|
|
|
type MenuItem struct {
|
|
Val string
|
|
Text string
|
|
}
|
|
|
|
type Menu struct {
|
|
ui.Block
|
|
Items []MenuItem
|
|
TextFgColor ui.Attribute
|
|
TextBgColor ui.Attribute
|
|
Selectable bool
|
|
CursorPos int
|
|
padding Padding
|
|
}
|
|
|
|
func NewMenu(items []string) *Menu {
|
|
var mItems []MenuItem
|
|
for _, s := range items {
|
|
mItems = append(mItems, MenuItem{Val: s})
|
|
}
|
|
m := &Menu{
|
|
Block: *ui.NewBlock(),
|
|
Items: mItems,
|
|
TextFgColor: ui.ThemeAttr("par.text.fg"),
|
|
TextBgColor: ui.ThemeAttr("par.text.bg"),
|
|
Selectable: false,
|
|
CursorPos: 0,
|
|
padding: Padding{4, 2},
|
|
}
|
|
m.calcSize()
|
|
return m
|
|
}
|
|
|
|
func (m *Menu) SetItems(items []MenuItem) {
|
|
m.Items = items
|
|
m.calcSize()
|
|
}
|
|
|
|
func (m *Menu) SelectedItem() MenuItem {
|
|
return m.Items[m.CursorPos]
|
|
}
|
|
|
|
func (m *Menu) Buffer() ui.Buffer {
|
|
var cell ui.Cell
|
|
buf := m.Block.Buffer()
|
|
|
|
for n, item := range m.Items {
|
|
x := m.padding[0]
|
|
for _, ch := range getDisplayText(item) {
|
|
// invert bg/fg colors on currently selected row
|
|
if m.Selectable && n == m.CursorPos {
|
|
cell = ui.Cell{Ch: ch, Fg: m.TextBgColor, Bg: m.TextFgColor}
|
|
} else {
|
|
cell = ui.Cell{Ch: ch, Fg: m.TextFgColor, Bg: m.TextBgColor}
|
|
}
|
|
buf.Set(x, n+m.padding[1], cell)
|
|
x++
|
|
}
|
|
}
|
|
|
|
return buf
|
|
}
|
|
|
|
func (m *Menu) Up(ui.Event) {
|
|
if m.CursorPos > 0 {
|
|
m.CursorPos--
|
|
ui.Render(m)
|
|
}
|
|
}
|
|
|
|
func (m *Menu) Down(ui.Event) {
|
|
if m.CursorPos < (len(m.Items) - 1) {
|
|
m.CursorPos++
|
|
ui.Render(m)
|
|
}
|
|
}
|
|
|
|
// Setup some default handlers for menu navigation
|
|
func (m *Menu) NavigationHandlers() {
|
|
ui.Handle("/sys/kbd/<up>", m.Up)
|
|
ui.Handle("/sys/kbd/<down>", m.Down)
|
|
ui.Handle("/sys/kbd/q", func(ui.Event) { ui.StopLoop() })
|
|
}
|
|
|
|
// Set width and height based on menu items
|
|
func (m *Menu) calcSize() {
|
|
m.Width = 8 // minimum width
|
|
|
|
items := m.Items
|
|
for _, i := range m.Items {
|
|
s := getDisplayText(i)
|
|
if len(s) > m.Width {
|
|
m.Width = len(s)
|
|
}
|
|
}
|
|
|
|
m.Width += (m.padding[0] * 2)
|
|
m.Height = len(items) + (m.padding[1] * 2)
|
|
}
|
|
|
|
// override display text of item, if given
|
|
func getDisplayText(m MenuItem) string {
|
|
if m.Text != "" {
|
|
return m.Text
|
|
}
|
|
return m.Val
|
|
}
|