aboutsummaryrefslogtreecommitdiff
path: root/parse.go
blob: ccf2a8790e9e92f78f98dd7a9611d4eaf0cb6c8d (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
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 parse(in []byte) string {
	buf := new(bytes.Buffer)
	for _, v := range strings.Split(string(in), "\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 '-':
			lines := strings.Split(v, "\r\n")
			for i, l := range lines {
				lines[i] = l[1:]
			}
			doc.ExecuteTemplate(buf, "list", lines)
		default:
			doc.ExecuteTemplate(buf, "par", v)
		}
	}
	return buf.String()
}