gochan/gochan
1
0
Fork 0
mirror of https://github.com/Eggbertx/gochan.git synced 2025-08-29 08:56:23 -07:00
Code Releases Activity

render all inner nodes to buffer instead of using regular expression to remove body tag

Browse source
This commit is contained in:
Eggbertx 2024-03-27 22:57:35 -07:00
parent 36a5b2f5a3
commit a9de4c4abb
1 changed files with 10 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

23
pkg/gctemplates/misc.go
View file

@ -3,35 +3,32 @@ package gctemplates
import (
"bytes"
"html/template"
"regexp"
"strings"
"github.com/gochan-org/gochan/pkg/gcutil"
x_html "golang.org/x/net/html"
)
var (
bodyElementRE = regexp.MustCompile("</?body>")
)
// truncateHTML truncates a template.HTML string to a certain visible character limit and line limit
func truncateHTML(htmlText template.HTML, characterLimit, maxLines int) template.HTML {
if htmlText == "" {
return ""
}
dom, err := x_html.Parse(strings.NewReader(string(htmlText)))
node, err := x_html.Parse(strings.NewReader(string(htmlText)))
if err != nil {
gcutil.LogError(err).Send()
return template.HTML("Server error truncating HTML, failed to parse html")
}
if dom.Type == x_html.DocumentNode {
dom = dom.FirstChild.LastChild
}
truncateHTMLNodes(dom, characterLimit, maxLines)
buf := new(bytes.Buffer)
x_html.Render(buf, dom)
innerBody := bodyElementRE.ReplaceAllString(buf.String(), "") // strip out <body> and </body>
return template.HTML(innerBody) // skipcq: GSC-G203
if node.Type == x_html.DocumentNode {
node = node.FirstChild.LastChild // FirstChild is <html>, has <head> and <body> children
}
truncateHTMLNodes(node, characterLimit, maxLines)
for node = node.FirstChild; node != nil; node = node.NextSibling {
// render all nodes inside body node
x_html.Render(buf, node)
}
return template.HTML(buf.String()) // skipcq: GSC-G203
}
func removeNextSiblings(node *x_html.Node) {