aboutsummaryrefslogtreecommitdiff
path: root/parse.go
blob: a6f2f3b92ce4cf1c892b3a4db122919f2caa3a28 (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
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) []string {
	li := strings.Split(in, "\r\n")
	for i, l := range li {
		li[i] = l[1:]
	}
	return li
}

func (p Page) HTML() string {
	buf := new(bytes.Buffer)
	for _, v := range strings.Split(string(p.Body), "\r\n\r\n") {
		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))
		default:
			doc.ExecuteTemplate(buf, "par", v)
		}
	}
	return buf.String()
}