feat(tree): hide children

Signed-off-by: Carlos Alexandro Becker <[email protected]>

tree/tree.go

@@ -38,6 +38,7 @@ type Node interface {
 	Value() string
 	Children() Children
 	Hidden() bool
+	SetHidden(bool)
 }
 
 // Leaf is a node without children.
@@ -46,6 +47,9 @@ type Leaf struct {
 	hidden bool
 }
 
+// SetHidden implements Node.
+func (s *Leaf) SetHidden(hidden bool) { s.hidden = hidden }
+
 // Children of a Leaf node are always empty.
 func (Leaf) Children() Children {
 	return NodeChildren(nil)
@@ -77,6 +81,9 @@ type Tree struct { //nolint:revive
 	ronce sync.Once
 }
 
+// SetHidden implements Node.
+func (t *Tree) SetHidden(hidden bool) { t.Hide(hidden) }
+
 // Hidden returns whether this node is hidden.
 func (t *Tree) Hidden() bool {
 	return t.hidden
@@ -147,7 +154,7 @@ func (t *Tree) Child(children ...any) *Tree {
 			t.children = t.children.(NodeChildren).Append(item)
 		case fmt.Stringer:
 			s := Leaf{value: item.String()}
-			t.children = t.children.(NodeChildren).Append(s)
+			t.children = t.children.(NodeChildren).Append(&s)
 		case string:
 			s := Leaf{value: item}
 			t.children = t.children.(NodeChildren).Append(&s)
@@ -180,9 +187,6 @@ func ensureParent(nodes Children, item *Tree) (*Tree, int) {
 			parent.Child(item.children.At(i))
 		}
 		return parent, j
-	case Leaf:
-		item.value = parent.Value()
-		return item, j
 	case *Leaf:
 		item.value = parent.Value()
 		return item, j

tree/tree_test.go

@@ -705,6 +705,33 @@ func TestTypes(t *testing.T) {
 	assertEqual(t, want, tree.String())
 }
 
+func TestLeafHidden(t *testing.T) {
+	tr := tree.New().
+		Child(
+			"Foo",
+			tree.Root("Bar").
+				Child(
+					"Qux",
+					tree.Root("Quux").
+						Child("This should be hidden").
+						Hide(true),
+					"Quuux",
+				),
+			"Baz",
+		)
+
+	// Hide Qux.
+	tr.Children().At(1).Children().At(0).SetHidden(true)
+
+	want := `
+├── Foo
+├── Bar
+│   └── Quuux
+└── Baz
+	`
+	assertEqual(t, want, tr.String())
+}
+
 // assertEqual verifies the strings are equal, assuming its terminal output.
 func assertEqual(tb testing.TB, want, got string) {
 	tb.Helper()