aboutsummaryrefslogtreecommitdiff
path: root/mardown/ast_header.go
blob: 7c997d5a6d55cb0719b044d7ffef8bd2dfa76a06 (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
40
41
42
43
44
45
46
47
48
49
50
package mardown

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

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

type astHeader struct {
	level   uint
	content []block
}

func (a *astHeader) Eval() (template.HTML, error) {
	if a.level > 6 {
		return "", ErrInvalidHeader
	}
	var content template.HTML
	for _, b := range a.content {
		c, err := b.Eval()
		if err != nil {
			return "", err
		}
		content += c
	}
	return template.HTML(fmt.Sprintf("<h%d>%s<h%d>", a.level, content, a.level)), nil
}

func header(lxs lexers) (block, error) {
	b := &astHeader{level: uint(len(lxs.Current().Value))}
	for lxs.Next() && lxs.Current().Type != lexerBreak {
		bl, err := getBlock(lxs)
		if err != nil {
			return nil, err
		}
		// if this is a header, just consider it as literal #
		if h, ok := bl.(*astHeader); ok {
			var s string
			for range h.level {
				s += "#"
			}
			b.content = append(b.content, astLiteral(s))
			b.content = append(b.content, h.content...)
		}
		b.content = append(b.content, bl)
	}
	return b, nil
}