79 lines
1.7 KiB
Go
79 lines
1.7 KiB
Go
// SPDX-FileCopyrightText: 2024 Dan Anglin <d.n.i.anglin@gmail.com>
|
|
//
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
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 "<br>", "<p>", "</p>", "</li>":
|
|
_, _ = writer.WriteString("\n")
|
|
case "<ul>":
|
|
state.htmlListType = htmlUnorderedList
|
|
_, _ = writer.WriteString("\n")
|
|
case "<ol>":
|
|
state.htmlListType = htmlOrderedList
|
|
_, _ = writer.WriteString("\n")
|
|
case "</ul>":
|
|
state.htmlListType = htmlNoList
|
|
case "</ol>":
|
|
state.htmlListType = htmlNoList
|
|
state.orderedListIndex = 1
|
|
case "<li>":
|
|
switch state.htmlListType {
|
|
case htmlUnorderedList:
|
|
_, _ = writer.WriteString(symbolBullet + " ")
|
|
case htmlOrderedList:
|
|
_, _ = writer.WriteString(strconv.Itoa(state.orderedListIndex) + ". ")
|
|
state.orderedListIndex++
|
|
}
|
|
}
|
|
}
|