apollo/enbas
1
0
Fork 0
Code Issues 4 Pull requests Projects Activity Actions

checkpoint: process unordered lists

Browse source
This commit is contained in:
Dan Anglin 2024-06-01 01:08:43 +01:00
parent 5cf6116fe8
commit ffc40d5165
Signed by: dananglin
GPG key ID: 0C1D44CFBEE68638
4 changed files with 23 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
internal/model/account.go
View file

@ -88,7 +88,7 @@ func (a Account) Display(noColor bool) string {
metadata += fmt.Sprintf(
"\n %s: %s",
utilities.FieldFormat(noColor, field.Name),
utilities.StripHTMLTags(field.Value),
utilities.ConvertHTMLToText(field.Value),
)
}
@ -105,7 +105,7 @@ func (a Account) Display(noColor bool) string {
utilities.FieldFormat(noColor, "Following:"), a.FollowingCount,
utilities.FieldFormat(noColor, "Statuses:"), a.StatusCount,
utilities.HeaderFormat(noColor, "BIOGRAPHY:"),
utilities.WrapLines(utilities.StripHTMLTags(a.Note), "\n ", 80),
utilities.WrapLines(utilities.ConvertHTMLToText(a.Note), "\n ", 80),
utilities.HeaderFormat(noColor, "METADATA:"),
metadata,
utilities.HeaderFormat(noColor, "ACCOUNT URL:"),

2
internal/model/status.go
View file

@ -180,7 +180,7 @@ func (s Status) Display(noColor bool) string {
format,
utilities.DisplayNameFormat(noColor, s.Account.DisplayName), s.Account.Username,
utilities.HeaderFormat(noColor, "CONTENT:"),
utilities.WrapLines(utilities.StripHTMLTags(s.Content), "\n ", 80),
utilities.WrapLines(utilities.ConvertHTMLToText(s.Content), "\n ", 80),
utilities.HeaderFormat(noColor, "STATUS ID:"),
s.ID,
utilities.HeaderFormat(noColor, "CREATED AT:"),

2
internal/model/timeline.go
View file

@ -30,7 +30,7 @@ func (t Timeline) Display(noColor bool) string {
createdAt = status.Reblog.CreatedAt
}
builder.WriteString(utilities.WrapLines(utilities.StripHTMLTags(status.Content), "\n", 80) + "\n\n")
builder.WriteString(utilities.WrapLines(utilities.ConvertHTMLToText(status.Content), "\n", 80) + "\n\n")
builder.WriteString(utilities.FieldFormat(noColor, "ID:") + " " + statusID + "\t" + utilities.FieldFormat(noColor, "Created at:") + " " + utilities.FormatTime(createdAt) + "\n")
builder.WriteString(separator + "\n")
}

29
internal/utilities/html.go
View file

@ -1,16 +1,19 @@
package utilities
import (
"io"
"strings"
"golang.org/x/net/html"
)
func StripHTMLTags(text string) string {
token := html.NewTokenizer(strings.NewReader(text))
func ConvertHTMLToText(text string) string {
var builder strings.Builder
var unorderedList bool
token := html.NewTokenizer(strings.NewReader(text))
for {
tt := token.Next()
switch tt {
@ -21,18 +24,24 @@ func StripHTMLTags(text string) string {
builder.WriteString(text)
case html.StartTagToken, html.EndTagToken:
tag := token.Token().String()
builder.WriteString(transformTag(tag))
unorderedList = processTag(&builder, tag, unorderedList)
}
}
}
func transformTag(tag string) string {
func processTag(writer io.StringWriter, tag string, unorderedList bool) bool {
switch tag {
case "<br>":
return "\n"
case "<p>", "</p>":
return "\n"
case "<br>", "<p>", "</p>", "</li>":
_, _ = writer.WriteString("\n")
case "<ul>":
unorderedList = true
case "</ul>":
unorderedList = false
case "<li>":
if unorderedList {
_, _ = writer.WriteString("• ")
}
}
return ""
return unorderedList
}