aboutsummaryrefslogtreecommitdiff
path: root/parse.go
blob: ac187f506eb191c5aa93c3009637438eb85c5e36 (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
package main

import (
	"bytes"
	"regexp"
	"strings"
	"text/template"
)

var (
	doc     = template.Must(template.ParseFiles("tmpl/doc"))
	intlink = regexp.MustCompile(`\[[^\]]+\]`)
	extlink = regexp.MustCompile(`(https?|ftp)://\S+`)
	imglink = regexp.MustCompile(`\.(jpe?g|png|gif)$`)
)

func lines(in string) (ret []string) {
	for _, l := range strings.Split(in, "\r\n") {
		if len(l) > 1 {
			ret = append(ret, l[1:])
		}
	}
	return
}

func (p Page) HTML() string {
	buf := new(bytes.Buffer)
	for _, v := range strings.Split(string(p.Body), "\r\n\r\n") {
		if len(v) < 1 {
			continue
		}
		v = intlink.ReplaceAllStringFunc(v, func(s string) string {
			tmp := new(bytes.Buffer)
			doc.ExecuteTemplate(tmp, "link", s[1:len(s)-1])
			return tmp.String()
		})
		v = extlink.ReplaceAllStringFunc(v, func(s string) string {
			tmp := new(bytes.Buffer)
			if imglink.MatchString(s) {
				doc.ExecuteTemplate(tmp, "img", s)
			} else {
				doc.ExecuteTemplate(tmp, "link", s)
			}
			return tmp.String()
		})

		switch v[0] {
		case ' ':
			doc.ExecuteTemplate(buf, "pre", v)
		case '-', '*':
			doc.ExecuteTemplate(buf, "list", lines(v))
		case '#':
			doc.ExecuteTemplate(buf, "olist", lines(v))
		case '=':
			doc.ExecuteTemplate(buf, "heading", v[1:])
		default:
			doc.ExecuteTemplate(buf, "par", v)
		}
	}
	return buf.String()
}