1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
package build
import (
"fmt"
"os"
"strings"
"github.com/adrg/frontmatter"
"github.com/gomarkdown/markdown"
"github.com/gomarkdown/markdown/ast"
"github.com/gomarkdown/markdown/html"
"github.com/gomarkdown/markdown/parser"
"github.com/HimanshuSardana/kite/pkg/content"
)
type TOCItem struct {
Level int
Text string
ID string
}
type ParsedPage struct {
Frontmatter content.Frontmatter
Content []byte
TOC []TOCItem
}
func ParseMarkdown(path string) (*ParsedPage, error) {
md, err := os.ReadFile(path)
if err != nil {
return nil, fmt.Errorf("reading %s: %w", path, err)
}
var matter content.Frontmatter
rest, err := frontmatter.Parse(strings.NewReader(string(md)), &matter)
if err != nil {
return nil, fmt.Errorf("parsing frontmatter: %w", err)
}
extensions := parser.CommonExtensions | parser.AutoHeadingIDs
p := parser.NewWithExtensions(extensions)
doc := p.Parse(rest)
var toc []TOCItem
ast.WalkFunc(doc, func(node ast.Node, entering bool) ast.WalkStatus {
if heading, ok := node.(*ast.Heading); ok && entering {
text := extractText(heading)
id := string(heading.HeadingID)
toc = append(toc, TOCItem{
Level: heading.Level,
Text: text,
ID: id,
})
}
return ast.GoToNext
})
renderer := html.NewRenderer(html.RendererOptions{
Flags: html.CommonFlags,
})
output := markdown.Render(doc, renderer)
return &ParsedPage{
Frontmatter: matter,
Content: output,
TOC: toc,
}, nil
}
func extractText(h *ast.Heading) string {
var text string
ast.WalkFunc(h, func(node ast.Node, entering bool) ast.WalkStatus {
if leaf, ok := node.(*ast.Text); ok && entering {
text += string(leaf.Literal)
}
return ast.GoToNext
})
return text
}
|
