feat(processtree): Introduce WithHideOnSuccess option (#934)

Reviewed-by: Cezar Craciunoiu <cezar.craciunoiu@unikraft.io> Approved-by: Cezar Craciunoiu <cezar.craciunoiu@unikraft.io>
unikraft · Oct 26, 2023 · 190f42b · 190f42b
2 parents 4066d0e + 55cfb30
commit 190f42b
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/tui/processtree/options.go b/tui/processtree/options.go
@@ -33,3 +33,10 @@ func WithFailFast(failFast bool) ProcessTreeOption {
 		return nil
 	}
 }
+
+func WithHideOnSuccess(hide bool) ProcessTreeOption {
+	return func(pt *ProcessTree) error {
+		pt.hide = hide
+		return nil
+	}
+}
diff --git a/tui/processtree/processtree.go b/tui/processtree/processtree.go
@@ -76,6 +76,7 @@ type ProcessTree struct {
 	errChan  chan error
 	failFast bool
 	oldOut   iostreams.FileWriter
+	hide     bool
 }
 
 func NewProcessTree(ctx context.Context, opts []ProcessTreeOption, tree ...*ProcessTreeItem) (*ProcessTree, error) {

diff --git a/tui/processtree/view.go b/tui/processtree/view.go
@@ -57,6 +57,10 @@ func (pt ProcessTree) View() string {
 }
 
 func (stm ProcessTree) printItem(pti *ProcessTreeItem, offset uint) string {
+	if pti.status == StatusSuccess && stm.hide {
+		return ""
+	}
+
 	failed := 0
 	completed := 0
 	running := 0