aboutsummaryrefslogtreecommitdiff
path: root/mardown/ast.go
blob: ed4d1f8e30ba9656f027e1bd8c532024877412da (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
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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
package mardown

import (
	"encoding/json"
	"errors"
	"fmt"
	"html/template"
	"strings"
)

var ErrUnkownLexType = errors.New("unkown lex type")

type block interface {
	Eval() (template.HTML, error)
}

type tree struct {
	blocks []block
}

func (t *tree) Eval() (template.HTML, error) {
	var content template.HTML
	for _, c := range t.blocks {
		ct, err := c.Eval()
		if err != nil {
			return "", err
		}
		content += ct
	}
	return content, nil
}

func (t *tree) String() string {
	b, _ := json.MarshalIndent(t, "", "  ")
	return string(b)
}

func ast(lxs *lexers) (*tree, error) {
	tr := new(tree)
	newLine := true
	for lxs.Next() {
		b, err := getBlock(lxs, newLine)
		if err != nil {
			return nil, err
		}
		if b != nil {
			tr.blocks = append(tr.blocks, b)
		}
		if !lxs.Finished() {
			newLine = lxs.Current().Type == lexerBreak
		}
	}
	return tr, nil
}

func getBlock(lxs *lexers, newLine bool) (block, error) {
	var b block
	var err error
	switch lxs.Current().Type {
	case lexerHeader:
		if !newLine {
			b, err = paragraph(lxs, false)
		} else {
			b, err = header(lxs)
		}
	case lexerExternal:
		if newLine && lxs.Current().Value == "![" {
			b, err = external(lxs)
		} else {
			b, err = paragraph(lxs, false)
		}
	case lexerQuote:
		if newLine {
			b, err = quote(lxs)
		} else {
			b, err = paragraph(lxs, false)
		}
	case lexerList:
		if newLine {
			b, err = list(lxs)
		} else {
			b, err = paragraph(lxs, false)
		}
	case lexerCode:
		if !newLine && len(lxs.Current().Value) == 3 {
			return nil, ErrInvalidCodeBlockPosition
		}
		if len(lxs.Current().Value) == 1 {
			b, err = paragraph(lxs, false)
		} else {
			b, err = code(lxs)
		}
	case lexerLiteral, lexerModifier:
		b, err = paragraph(lxs, false)
	case lexerBreak: // do nothing
	default:
		err = errors.Join(ErrUnkownLexType, fmt.Errorf("type received: %s", lxs.Current().Type))
	}
	return b, err
}

func trimSpace(s template.HTML) template.HTML {
	return template.HTML(strings.TrimSpace(string(s)))
}