package printer
import (
"io"
"strconv"
"strings"
"golang.org/x/net/html"
)
const (
htmlNoList int = iota
htmlOrderedList
htmlUnorderedList
)
type htmlConvertState struct {
htmlListType int
orderedListIndex int
}
func (p Printer) convertHTMLToText(text string, wrapLines bool) string {
var builder strings.Builder
state := htmlConvertState{
htmlListType: htmlNoList,
orderedListIndex: 1,
}
token := html.NewTokenizer(strings.NewReader(text))
for {
tt := token.Next()
switch tt {
case html.ErrorToken:
if wrapLines {
return p.wrapLines(builder.String(), 0)
}
return builder.String()
case html.TextToken:
text := token.Token().Data
builder.WriteString(text)
case html.StartTagToken, html.EndTagToken:
tag := token.Token().String()
processTagToken(&state, &builder, tag)
}
}
}
func processTagToken(state *htmlConvertState, writer io.StringWriter, tag string) {
switch tag {
case "
", "
", "
", "": _, _ = writer.WriteString("\n") case "