From 2bf6a7fc9daee192bc82d559297fa98ccf95844d Mon Sep 17 00:00:00 2001
From: Maas Lalani <maas@lalani.dev>
Date: Thu, 18 Apr 2024 11:17:59 -0400
Subject: [PATCH] fix: enumerator

---
 examples/tree/states/main.go | 17 +++++------------
 1 file changed, 5 insertions(+), 12 deletions(-)

diff --git a/examples/tree/states/main.go b/examples/tree/states/main.go
index 093391ed..4b9d5811 100644
--- a/examples/tree/states/main.go
+++ b/examples/tree/states/main.go
@@ -7,18 +7,11 @@ import (
 	"github.com/charmbracelet/lipgloss/tree"
 )
 
-func openEnumerator(data tree.Data, i int) (indent, prefix string) {
-	if data.Length()-1 == i {
-		return "   ", "▼ "
-	}
-	return "  ", "▼ "
-}
-
-func closedEnumerator(data tree.Data, i int) (indent, prefix string) {
-	if data.Length()-1 == i {
+func enumerator(data tree.Data, i int) (indent, prefix string) {
+	if data.At(i).Hidden() {
 		return "   ", "▶ "
 	}
-	return "  ", "▶ "
+	return "  ", "▼ "
 }
 
 func normalEnumerator(data tree.Data, _ int) (indent, prefix string) {
@@ -46,7 +39,7 @@ func main() {
 					"Cat Food",
 					"Nutella",
 					"Powdered Sugar",
-				).EnumeratorStyle(gray).Enumerator(closedEnumerator).Hide(true),
+				).EnumeratorStyle(gray).Enumerator(normalEnumerator).Hide(true),
 			tree.New().
 				Root("Veggies").
 				Items(
@@ -56,7 +49,7 @@ func main() {
 		).
 		ItemStyle(pink).
 		EnumeratorStyle(gray).
-		Enumerator(openEnumerator)
+		Enumerator(enumerator)
 
 	fmt.Println(t)
 }