165 lines
3.7 KiB
Go
165 lines
3.7 KiB
Go
package ui
|
|
|
|
import (
|
|
"path"
|
|
"strings"
|
|
|
|
"github.com/derailed/k9s/internal/resource"
|
|
"github.com/derailed/tview"
|
|
"github.com/gdamore/tcell"
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
// Selectable represents a table with selections.
|
|
type SelectTable struct {
|
|
*tview.Table
|
|
|
|
ActiveNS string
|
|
selectedItem string
|
|
selectedRow int
|
|
selectedFn func(string) string
|
|
selListeners []SelectedRowFunc
|
|
marks map[string]bool
|
|
}
|
|
|
|
// ClearSelection reset selected row.
|
|
func (s *SelectTable) ClearSelection() {
|
|
s.Select(0, 0)
|
|
s.ScrollToBeginning()
|
|
}
|
|
|
|
// SelectFirstRow select first data row if any.
|
|
func (s *SelectTable) SelectFirstRow() {
|
|
if s.GetRowCount() > 0 {
|
|
s.Select(1, 0)
|
|
}
|
|
}
|
|
|
|
// GetSelectedItems return currently marked or selected items names.
|
|
func (s *SelectTable) GetSelectedItems() []string {
|
|
if len(s.marks) == 0 {
|
|
return []string{s.GetSelectedItem()}
|
|
}
|
|
|
|
var items []string
|
|
for item, marked := range s.marks {
|
|
if marked {
|
|
items = append(items, item)
|
|
}
|
|
}
|
|
|
|
return items
|
|
}
|
|
|
|
// GetSelectedItem returns the currently selected item name.
|
|
func (s *SelectTable) GetSelectedItem() string {
|
|
if s.selectedFn != nil {
|
|
return s.selectedFn(s.selectedItem)
|
|
}
|
|
return s.selectedItem
|
|
}
|
|
|
|
// GetSelectedCell returns the content of a cell for the currently selected row.
|
|
func (s *SelectTable) GetSelectedCell(col int) string {
|
|
return TrimCell(s, s.selectedRow, col)
|
|
}
|
|
|
|
// SetSelectedFn defines a function that cleanse the current selection.
|
|
func (s *SelectTable) SetSelectedFn(f func(string) string) {
|
|
s.selectedFn = f
|
|
}
|
|
|
|
// GetSelectedRow fetch the currently selected row index.
|
|
func (s *SelectTable) GetSelectedRowIndex() int {
|
|
return s.selectedRow
|
|
}
|
|
|
|
// RowSelected checks if there is an active row selection.
|
|
func (s *SelectTable) RowSelected() bool {
|
|
return s.selectedItem != ""
|
|
}
|
|
|
|
// GetRow retrieves the entire selected row.
|
|
func (s *SelectTable) GetRow() resource.Row {
|
|
r := make(resource.Row, s.GetColumnCount())
|
|
for i := 0; i < s.GetColumnCount(); i++ {
|
|
c := s.GetCell(s.selectedRow, i)
|
|
r[i] = strings.TrimSpace(c.Text)
|
|
}
|
|
return r
|
|
}
|
|
|
|
func (s *SelectTable) updateSelectedItem(r int) {
|
|
if r == 0 || s.GetCell(r, 0) == nil {
|
|
s.selectedItem = ""
|
|
return
|
|
}
|
|
|
|
col0 := TrimCell(s, r, 0)
|
|
switch s.ActiveNS {
|
|
case resource.NotNamespaced:
|
|
s.selectedItem = col0
|
|
case resource.AllNamespace, resource.AllNamespaces:
|
|
s.selectedItem = path.Join(col0, TrimCell(s, r, 1))
|
|
default:
|
|
s.selectedItem = path.Join(s.ActiveNS, col0)
|
|
}
|
|
}
|
|
|
|
// SelectRow select a given row by index.
|
|
func (s *SelectTable) SelectRow(r int, broadcast bool) {
|
|
if !broadcast {
|
|
s.SetSelectionChangedFunc(nil)
|
|
}
|
|
defer s.SetSelectionChangedFunc(s.selChanged)
|
|
s.Select(r, 0)
|
|
s.updateSelectedItem(r)
|
|
}
|
|
|
|
// UpdateSelection refresh selected row.
|
|
func (s *SelectTable) updateSelection(broadcast bool) {
|
|
s.SelectRow(s.selectedRow, broadcast)
|
|
}
|
|
|
|
func (s *SelectTable) selChanged(r, c int) {
|
|
s.selectedRow = r
|
|
s.updateSelectedItem(r)
|
|
if r == 0 {
|
|
return
|
|
}
|
|
|
|
if s.marks[s.GetSelectedItem()] {
|
|
s.SetSelectedStyle(tcell.ColorBlack, tcell.ColorCadetBlue, tcell.AttrBold)
|
|
} else {
|
|
cell := s.GetCell(r, c)
|
|
s.SetSelectedStyle(tcell.ColorBlack, cell.Color, tcell.AttrBold)
|
|
}
|
|
|
|
for _, f := range s.selListeners {
|
|
f(r, c)
|
|
}
|
|
}
|
|
|
|
// ToggleMark toggles marked row
|
|
func (s *SelectTable) ToggleMark() {
|
|
s.marks[s.GetSelectedItem()] = !s.marks[s.GetSelectedItem()]
|
|
if !s.marks[s.GetSelectedItem()] {
|
|
return
|
|
}
|
|
log.Debug().Msgf("YO!!!!")
|
|
s.SetSelectedStyle(
|
|
tcell.ColorBlack,
|
|
tcell.ColorViolet,
|
|
tcell.AttrBold,
|
|
)
|
|
}
|
|
|
|
func (s *Table) IsMarked(item string) bool {
|
|
return s.marks[item]
|
|
}
|
|
|
|
// AddSelectedRowListener add a new selected row listener.
|
|
func (s *SelectTable) AddSelectedRowListener(f SelectedRowFunc) {
|
|
s.selListeners = append(s.selListeners, f)
|
|
}
|