aboutsummaryrefslogtreecommitdiff
path: root/mardown/ast_header.go
blob: 0ce8a22d99a8064a266ee7d20ca3ed58da63a499 (plain)
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
package mardown

import (
	"errors"
	"fmt"
	"html/template"
)

var ErrInvalidHeader = errors.New("invalid header")

type astHeader struct {
	level   uint
	content *astParagraph
}

func (a *astHeader) Eval() (template.HTML, *ParseError) {
	if a.level > 6 {
		return "", &ParseError{lxs: lexers{}, internal: ErrInvalidCodeFormat}
	}
	var content template.HTML
	content, err := a.content.Eval()
	if err != nil {
		return "", err
	}
	return template.HTML(fmt.Sprintf("<h%d>%s</h%d>", a.level, trimSpace(content), a.level)), nil
}

func header(lxs *lexers) (*astHeader, *ParseError) {
	b := &astHeader{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
}