termwind/render.go at main · codiume/termwind · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
package termwind

import (
	"os"
	"strconv"
	"strings"

	"github.com/charmbracelet/lipgloss"
	"golang.org/x/term"
	"golang.org/x/text/cases"
	"golang.org/x/text/language"
)

// Render a HTML string and returns a styled string
// suitable for terminal output. Supported elements and class names
// are documented in the README.
func Render(input string) (string, error) {
	root, err := Parse(input)
	if err != nil {
		return "", err
	}
	return RenderToString(root), nil
}

func RenderToString(el *Element) string {
	var buf strings.Builder
	termWidth := getTerminalWidth()
	renderElement(el, &buf, 0, termWidth)
	return buf.String()
}

func applyTextTransform(text, transform string) string {
	switch transform {
	case "uppercase":
		return strings.ToUpper(text)
	case "lowercase":
		return strings.ToLower(text)
	case "capitalize":
		return cases.Title(language.English).String(text)
	case "snakecase":
		return strings.ToLower(strings.ReplaceAll(strings.ReplaceAll(text, " ", "_"), "-", "_"))
	default:
		return text
	}
}

func getTerminalWidth() int {
	width, _, err := term.GetSize(int(os.Stdout.Fd()))
	if err != nil || width <= 0 {
		return 80
	}
	return width
}

func renderElement(el *Element, buf *strings.Builder, listIndex int, termWidth int) {
	if el == nil {
		return
	}

	switch el.Tag {
	case TagRaw:
		buf.WriteString(el.Content)
		return

	case TagBreakLine:
		buf.WriteString("\n")
		return

	case TagHr:
		width := termWidth - 2
		buf.WriteString("\n" + strings.Repeat("─", width) + "\n")
		return
	}

	var content strings.Builder
	for i, child := range el.Children {
		childIndex := 0
		if el.Tag == TagOl {
			childIndex = i + 1
		}
		renderElement(child, &content, childIndex, termWidth)
	}

	innerContent := content.String()
	innerContent = applyTextTransform(innerContent, el.Style.TextTransform)

	switch el.Tag {
	case TagDiv:
		style := buildStyle(el, termWidth)
		rendered := style.Render(innerContent)
		buf.WriteString(rendered + "\n")

	case TagParagraph:
		style := buildStyle(el, termWidth)
		buf.WriteString("\n" + style.Render(innerContent) + "\n")

	case TagSpan:
		style := buildStyle(el, termWidth)
		buf.WriteString(style.Render(innerContent))

	case TagUl:
		buf.WriteString(innerContent + "\n")

	case TagOl:
		buf.WriteString(innerContent + "\n")

	case TagLi:
		var prefix string
		if listIndex > 0 {
			prefix = strconv.Itoa(listIndex) + ". "
		} else {
			prefix = "• "
		}
		line := prefix + innerContent
		if el.Style != (Style{}) {
			style := buildStyle(el, termWidth)
			line = style.Render(line)
		}
		buf.WriteString(line + "\n")

	case TagDl:
		buf.WriteString(innerContent + "\n")

	case TagDt:
		dtStyle := lipgloss.NewStyle().Bold(true)
		buf.WriteString(dtStyle.Render(innerContent) + "\n")

	case TagDd:
		ddStyle := lipgloss.NewStyle().MarginLeft(2)
		buf.WriteString(ddStyle.Render(innerContent) + "\n")

	case TagAnchor:
		style := buildStyle(el, termWidth)
		result := style.Render(innerContent)
		if href, ok := el.Attrs["href"]; ok && href != "" {
			linkStyle := lipgloss.NewStyle().Underline(true)
			result += " " + linkStyle.Render("("+href+")")
		}
		buf.WriteString(result)

	default:
		style := buildStyle(el, termWidth)
		buf.WriteString(style.Render(innerContent) + "\n")
	}
}