1 files changed, 44 insertions, 0 deletions
diff --git a/markdown/ast_heading.go b/markdown/ast_heading.go
new file mode 100644
index 0000000..206ff94
--- /dev/null
+++ b/markdown/ast_heading.go
@@ -0,0 +1,44 @@
+package markdown
+
+import (
+	"errors"
+	"html/template"
+	"strings"
+
+	"git.anhgelus.world/anhgelus/small-web/dom"
+)
+
+var ErrInvalidHeader = errors.New("invalid header")
+
+type astHeading struct {
+	level   uint
+	content *astParagraph
+}
+
+func (a *astHeading) Eval(opt *Option) (template.HTML, *ParseError) {
+	if a.level > 6 {
+		return "", &ParseError{lxs: lexers{}, internal: ErrInvalidCodeFormat}
+	}
+	var content template.HTML
+	content, err := a.content.Eval(opt)
+	if err != nil {
+		return "", err
+	}
+	return dom.NewHeading(
+		a.level,
+		template.HTML(strings.TrimSpace(string(content))),
+	).Render(), nil
+}
+
+func heading(lxs *lexers) (*astHeading, *ParseError) {
+	b := &astHeading{level: uint(len(lxs.Current().Value))}
+	if !lxs.Next() {
+		return nil, &ParseError{lxs: *lxs, internal: ErrInvalidHeader}
+	}
+	var err *ParseError
+	b.content, err = paragraph(lxs, true)
+	if err != nil {
+		return nil, err
+	}
+	return b, nil
+}