mirror of
https://github.com/bcicen/ctop.git
synced 2024-08-30 18:23:19 +00:00
138 lines
2.6 KiB
Go
138 lines
2.6 KiB
Go
package widgets
|
|
|
|
import (
|
|
"sort"
|
|
|
|
ui "github.com/gizak/termui"
|
|
)
|
|
|
|
type Padding [2]int // x,y padding
|
|
|
|
type MenuItem struct {
|
|
Val string
|
|
Label string
|
|
}
|
|
|
|
// Use label as display text of item, if given
|
|
func (m MenuItem) Text() string {
|
|
if m.Label != "" {
|
|
return m.Label
|
|
}
|
|
return m.Val
|
|
}
|
|
|
|
type MenuItems []MenuItem
|
|
|
|
// Create new MenuItems from string array
|
|
func NewMenuItems(a []string) (items MenuItems) {
|
|
for _, s := range a {
|
|
items = append(items, MenuItem{Val: s})
|
|
}
|
|
return items
|
|
}
|
|
|
|
// Sort methods for MenuItems
|
|
func (m MenuItems) Len() int { return len(m) }
|
|
func (m MenuItems) Swap(a, b int) { m[a], m[b] = m[b], m[a] }
|
|
func (m MenuItems) Less(a, b int) bool {
|
|
return m[a].Text() < m[b].Text()
|
|
}
|
|
|
|
type Menu struct {
|
|
ui.Block
|
|
Items MenuItems
|
|
SortItems bool // enable automatic sorting of menu items
|
|
TextFgColor ui.Attribute
|
|
TextBgColor ui.Attribute
|
|
Selectable bool
|
|
CursorPos int
|
|
padding Padding
|
|
}
|
|
|
|
func NewMenu() *Menu {
|
|
return &Menu{
|
|
Block: *ui.NewBlock(),
|
|
TextFgColor: ui.ThemeAttr("par.text.fg"),
|
|
TextBgColor: ui.ThemeAttr("par.text.bg"),
|
|
CursorPos: 0,
|
|
padding: Padding{4, 2},
|
|
}
|
|
}
|
|
|
|
func (m *Menu) AddItems(items ...MenuItem) {
|
|
for _, i := range items {
|
|
m.Items = append(m.Items, i)
|
|
}
|
|
m.refresh()
|
|
}
|
|
|
|
// Sort menu items(if enabled) and re-calculate window size
|
|
func (m *Menu) refresh() {
|
|
if m.SortItems {
|
|
sort.Sort(m.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 item.Text() {
|
|
// 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 := i.Text()
|
|
if len(s) > m.Width {
|
|
m.Width = len(s)
|
|
}
|
|
}
|
|
|
|
m.Width += (m.padding[0] * 2)
|
|
m.Height = len(items) + (m.padding[1] * 2)
|
|
}
|