feat: change styles

list/list.go

@@ -51,6 +51,30 @@ func (l *List) Enumerator(enumerator Enumerator) *List {
 	return l
 }
 
+// Style sets the list style.
+func (l *List) Style(style Style) *List {
+	l.style = style
+	return l
+}
+
+// EnumeratorStyle sets the enumerator style.
+func (l *List) EnumeratorStyle(style lipgloss.Style) *List {
+	l.style.Enumerator = style
+	return l
+}
+
+// ItemStyle sets the item style.
+func (l *List) ItemStyle(style lipgloss.Style) *List {
+	l.style.Item = style
+	return l
+}
+
+// BaseStyle sets the base style.
+func (l *List) BaseStyle(style lipgloss.Style) *List {
+	l.style.Base = style
+	return l
+}
+
 // Indent sets the indent level.
 func (l *List) Indent(indent int) *List {
 	if indent < 0 {

list/list_test.go

@@ -3,6 +3,8 @@ package list
 import (
 	"strings"
 	"testing"
+
+	"github.com/charmbracelet/lipgloss"
 )
 
 func TestList(t *testing.T) {
@@ -148,7 +150,7 @@ func TestDeepNestedList(t *testing.T) {
 	}
 }
 
-func TestEnumeration(t *testing.T) {
+func TestEnumerators(t *testing.T) {
 	tests := []struct {
 		enumeration Enumerator
 		expected    string
@@ -221,6 +223,94 @@ III. Baz
 	}
 }
 
+func TestEnumeratorsTransform(t *testing.T) {
+	tests := []struct {
+		enumeration Enumerator
+		style       lipgloss.Style
+		expected    string
+	}{
+		{
+			enumeration: Alphabet,
+			style:       lipgloss.NewStyle().MarginRight(1).Transform(strings.ToLower),
+			expected: `
+a. Foo
+b. Bar
+c. Baz
+  a. Qux
+  b. Quux
+`,
+		},
+		{
+			enumeration: Arabic,
+			style: lipgloss.NewStyle().MarginRight(1).Transform(func(s string) string {
+				return strings.Replace(s, ".", ")", 1)
+			}),
+			expected: `
+1) Foo
+2) Bar
+3) Baz
+  1) Qux
+  2) Quux
+`,
+		},
+		{
+			enumeration: Roman,
+			style: lipgloss.NewStyle().Transform(func(s string) string {
+				return "(" + strings.Replace(strings.ToLower(s), ".", "", 1) + ") "
+			}),
+			expected: `
+(i) Foo
+(ii) Bar
+(iii) Baz
+  (i) Qux
+  (ii) Quux
+`,
+		},
+		{
+			enumeration: Bullet,
+			style: lipgloss.NewStyle().Transform(func(s string) string {
+				return "- " // this is better done by replacing the enumerator.
+			}),
+			expected: `
+- Foo
+- Bar
+- Baz
+  - Qux
+  - Quux
+`,
+		},
+		{
+			enumeration: Tree,
+			style: lipgloss.NewStyle().MarginRight(1).Transform(func(s string) string {
+				return strings.Replace(s, "─", "───", 1)
+			}),
+			expected: `
+├─── Foo
+├─── Bar
+└─── Baz
+  ├─── Qux
+  └─── Quux
+`,
+		},
+	}
+
+	for _, test := range tests {
+		expected := strings.TrimPrefix(test.expected, "\n")
+
+		l := New().
+			Enumerator(test.enumeration).
+			EnumeratorStyle(test.style).
+			Item("Foo").
+			Item("Bar").
+			Item("Baz").
+			Item(New("Qux", "Quux").Enumerator(test.enumeration).EnumeratorStyle(test.style))
+
+		if l.String() != expected {
+			t.Errorf("expected:\n\n%s\n\ngot:\n\n%s\n", expected, l.String())
+		}
+	}
+}
+
 func TestBullet(t *testing.T) {
 	tests := []struct {
 		enum Enumerator