/
server.go
95 lines (86 loc) · 2.08 KB
/
server.go
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
package main
import (
"bytes"
"fmt"
"html/template"
"io/ioutil"
"log"
"net/http"
"os"
"strings"
"time"
"github.com/yuin/goldmark"
)
// MarkdownHandler contains code & configuration
type MarkdownHandler struct {
Filename string
Markdown string
HTML template.HTML // as converted from Markdown
ModTime time.Time
Interval int
Template *template.Template
Verbose bool
}
// NewMarkdownHandler returns obj
func NewMarkdownHandler(filename string, interval int, verbose bool) *MarkdownHandler {
t, err := template.New("md").Parse(string(Client))
if err != nil {
log.Fatal(err)
}
return &MarkdownHandler{
Filename: filename,
Interval: interval,
Template: t,
Verbose: verbose,
}
}
// ServeHTTP implements the Handler interface to respond to request by converting markdown & rendering html
func (s *MarkdownHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
path := r.URL.Path[1:]
s.Log("GET /" + path)
if strings.Compare(path, "md") == 0 {
if s.IsModified() {
s.Refresh()
}
fmt.Fprint(w, s.HTML)
} else {
s.Refresh()
s.Template.Execute(w, s) // Execute against MarkdownHandler struct members
}
}
// IsModified checks whether ModTime has changed
func (s *MarkdownHandler) IsModified() bool {
stat, err := os.Stat(s.Filename)
if err != nil {
log.Fatal(err)
}
prev := s.ModTime
curr := stat.ModTime()
if prev.Equal(curr) {
return false
} else {
s.ModTime = curr
return true
}
}
// Refresh updates the input Markdown and converts to output HTML stored in the struct
func (s *MarkdownHandler) Refresh() {
s.Log("Refresh Markdown!")
markdown, err := ioutil.ReadFile(s.Filename)
if err != nil {
log.Fatal(err)
}
s.Markdown = string(markdown)
s.Markdown = strings.ReplaceAll(s.Markdown, "\r\n", "\n") // Windows
s.Markdown = strings.ReplaceAll(s.Markdown, "\r", "\n") // Mac
var html bytes.Buffer
if err := goldmark.Convert([]byte(s.Markdown), &html); err != nil {
s.Log("ERROR: Unable to parse Markdown")
}
s.HTML = template.HTML(html.String())
}
func (s *MarkdownHandler) Log(message string) {
if s.Verbose {
log.Print(message)
}
}