Feature: Filter on process list

src/tui/proc-filter.go

@@ -0,0 +1,67 @@
+package tui
+
+import (
+	"github.com/gdamore/tcell/v2"
+	"github.com/rivo/tview"
+)
+
+func (pv *pcView) showProcFilter() {
+	const fieldWidth = 50
+	f := tview.NewForm()
+	f.SetCancelFunc(func() {
+		pv.pages.RemovePage(PageDialog)
+	})
+	f.SetItemPadding(1)
+	f.SetBorder(true)
+	f.SetFieldBackgroundColor(tcell.ColorLightSkyBlue)
+	f.SetFieldTextColor(tcell.ColorBlack)
+	f.SetButtonsAlign(tview.AlignCenter)
+	f.SetTitle("Search Process")
+
+	f.AddInputField("Search For", "", fieldWidth, nil, nil)
+	f.AddCheckbox("Case Sensitive", false, nil)
+	f.AddCheckbox("Regex", false, nil)
+
+	searchFunc := func(searchTerm string) {
+		caseSensitive := f.GetFormItem(1).(*tview.Checkbox).IsChecked()
+		isRegex := f.GetFormItem(2).(*tview.Checkbox).IsChecked()
+
+		pv.searchProcess(searchTerm, isRegex, caseSensitive)
+	}
+	f.GetFormItem(0).(*tview.InputField).SetChangedFunc(func(text string) {
+		searchFunc(text)
+	})
+
+	f.GetFormItem(0).(*tview.InputField).SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
+		if event.Key() == tcell.KeyEnter {
+			pv.pages.RemovePage(PageDialog)
+			return nil
+		}
+
+		return event
+	})
+
+	f.AddButton("Search", func() {
+		pv.pages.RemovePage(PageDialog)
+	})
+	f.AddButton("Cancel", func() {
+		pv.resetProcessSearch()
+		pv.pages.RemovePage(PageDialog)
+	})
+
+	f.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
+		switch event.Key() {
+		case tcell.KeyEsc:
+			pv.resetProcessSearch()
+			pv.pages.RemovePage(PageDialog)
+		default:
+			return event
+		}
+		return nil
+	})
+	f.SetFocus(0)
+	pv.resetProcessSearch()
+	// Display and focus the dialog
+	pv.pages.AddPage(PageDialog, createDialogPage(f, fieldWidth+20, 11), true, true)
+	pv.appView.SetFocus(f)
+}

src/tui/proc-table.go

@@ -3,6 +3,7 @@ package tui
 import (
 	"context"
 	"fmt"
+	"regexp"
 	"github.com/f1bonacc1/process-compose/src/types"
 	"github.com/gdamore/tcell/v2"
 	"github.com/rivo/tview"
@@ -46,13 +47,24 @@ func (pv *pcView) fillTableData() {
 			pv.procTable.RemoveRow(row)
 			continue
 		}
+
+		if !pv.matchProcRegex(state.Name) {
+			pv.procTable.RemoveRow(row)
+			continue
+		}
+
 		rowVals := getTableRowValues(state)
 		setRowValues(pv.procTable, row, rowVals)
 		if state.IsRunning {
 			runningProcCount += 1
 		}
 		row += 1
 	}
+	if row == 1 {
+		pv.procTable.SetSelectable(false, false)
+	} else {
+		pv.procTable.SetSelectable(true, false)
+	}
 
 	// remove unnecessary rows, don't forget the title row (-1)
 	if pv.procTable.GetRowCount()-1 > row {
@@ -209,6 +221,54 @@ func (pv *pcView) setTableSorter(sortBy ColumnID) {
 	}
 }
 
+func (pv *pcView) setProcRegex(reg *regexp.Regexp) {
+	pv.procRegexMtx.Lock()
+	defer pv.procRegexMtx.Unlock()
+	pv.procRegex = reg
+}
+
+func (pv *pcView) matchProcRegex(procName string) bool {
+	pv.procRegexMtx.Lock()
+	defer pv.procRegexMtx.Unlock()
+	if pv.procRegex != nil {
+		return pv.procRegex.MatchString(procName)
+	}
+	// if no search string is present, match everything
+	return true
+}
+
+func (pv *pcView) resetProcessSearch() {
+	pv.setProcRegex(nil)
+	go pv.appView.QueueUpdateDraw(func() {
+		pv.fillTableData()
+	})
+}
+
+func (pv *pcView) searchProcess(search string, isRegex, caseSensitive bool) error {
+	if search == "" {
+		pv.setProcRegex(nil)
+		return nil
+	}
+	searchRegexString := search
+	if !isRegex {
+		searchRegexString = regexp.QuoteMeta(searchRegexString)
+	}
+	if !caseSensitive {
+		searchRegexString = "(?i)" + searchRegexString
+	}
+	searchRegex, err := regexp.Compile(searchRegexString)
+	if err != nil {
+		return err
+	}
+
+	pv.setProcRegex(searchRegex)
+	pv.procTable.Select(1, 1)
+	go pv.appView.QueueUpdateDraw(func() {
+		pv.fillTableData()
+	})
+	return nil
+}
+
 func (pv *pcView) getTableSorter() StateSorter {
 	pv.sortMtx.Lock()
 	defer pv.sortMtx.Unlock()

src/tui/view.go

@@ -10,6 +10,7 @@ import (
 	"github.com/rs/zerolog/log"
 	"os"
 	"os/signal"
+	"regexp"
 	"strconv"
 	"sync"
 	"sync/atomic"
@@ -58,6 +59,8 @@ type pcView struct {
 	project           app.IProject
 	sortMtx           sync.Mutex
 	stateSorter       StateSorter
+	procRegex         *regexp.Regexp
+	procRegexMtx      sync.Mutex
 	procColumns       map[ColumnID]string
 	refreshRate       time.Duration
 	cancelFn          context.CancelFunc
@@ -192,8 +195,15 @@ func (pv *pcView) onMainGridKey(event *tcell.EventKey) *tcell.EventKey {
 		pv.logsText.SetTitle(pv.getLogTitle(pv.getSelectedProcName()))
 	case pv.shortcuts.ShortCutKeys[ActionLogFindExit].key:
 		pv.exitSearch()
+		pv.resetProcessSearch()
 	case pv.shortcuts.ShortCutKeys[ActionNsFilter].key:
 		pv.showNsFilter()
+	case tcell.KeyRune:
+		if event.Rune() == '/' {
+			pv.showProcFilter()
+		} else {
+			return event
+		}
 	default:
 		return event
 	}