phoenix/hugo
1
0
Fork 0
mirror of https://github.com/gohugoio/hugo.git synced 2024-05-20 16:22:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove generated title, update trim positions

Browse source
This commit is contained in:
Paul 2021-08-21 16:06:47 +02:00
parent 769439cc5b
commit f1b1e5f41b
No known key found for this signature in database
GPG key ID: 20C692A6C6307F2E
1 changed files with 27 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
markup/pandoc/convert.go
View file

@ -110,6 +110,7 @@ func (a *pandocConverter) extractTOC(src []byte) ([]byte, tableofcontents.Root,
toVisit []*html.Node
)
// find body
f = func(n *html.Node) bool {
if n.Type == html.ElementNode && n.Data == "body" {
body = n
@ -134,6 +135,30 @@ func (a *pandocConverter) extractTOC(src []byte) ([]byte, tableofcontents.Root,
return nil, tableofcontents.Root{}, err
}
// remove by pandoc generated title
f = func(n *html.Node) bool {
if n.Type == html.ElementNode && n.Data == "header" && attr(n, "id") == "title-block-header" {
n.Parent.RemoveChild(n)
return true
}
if n.FirstChild != nil {
toVisit = append(toVisit, n.FirstChild)
}
if n.NextSibling != nil && f(n.NextSibling) {
return true
}
for len(toVisit) > 0 {
nv := toVisit[0]
toVisit = toVisit[1:]
if f(nv) {
return true
}
}
return false
}
f(body)
// find toc
f = func(n *html.Node) bool {
if n.Type == html.ElementNode && n.Data == "nav" && attr(n, "id") == "TOC" {
toc = parseTOC(n)
@ -167,8 +192,8 @@ func (a *pandocConverter) extractTOC(src []byte) ([]byte, tableofcontents.Root,
return nil, tableofcontents.Root{}, err
}
// ltrim <html><head></head><body> and rtrim </body></html> which are added by html.Render
res := buf.Bytes()[25:]
res = res[:len(res)-14]
res := buf.Bytes()[6:]
res = res[:len(res)-7]
return res, toc, nil
}