aboutsummaryrefslogtreecommitdiff
path: root/wiki.go
blob: 8afd5fda18682efbf02b2f72c2a1a034a8c61571 (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 (
	"flag"
	"io/ioutil"
	"net/http"
	"os"
	"path"
	"text/template"

	"github.com/russross/blackfriday"
)

var (
	listen    = flag.String("listen", ":8080", "Listen at")
	listentls = flag.String("listentls", ":8443", "Listen TLS at")
	cert      = flag.String("cert", "cert/cert.pem", "TLS Certificate")
	key       = flag.String("key", "cert/key.pem", "TLS Key")
	data      = flag.String("data", "data", "Data dir")
)

type Page struct {
	Title string
	Body  []byte
	Pages []os.FileInfo
}

func (p *Page) fileName() string {
	return path.Join(*data, p.Title)
}

func (p *Page) save() error {
	return ioutil.WriteFile(p.fileName(), p.Body, 0600)
}

func loadPage(title string) (p *Page, err error) {
	p = &Page{Title: title}
	p.Body, err = ioutil.ReadFile(p.fileName())
	return
}

func (p *Page) del() error {
	return os.Remove(p.fileName())
}

func (p *Page) render(w http.ResponseWriter, tmpl *template.Template) {
	err := tmpl.ExecuteTemplate(w, "root", p)
	if err != nil {
		http.Error(w, err.Error(), http.StatusInternalServerError)
	}
}

func (p *Page) HTML() string {
	return string(blackfriday.MarkdownCommon([]byte(p.Body)))
}

func main() {
	flag.Parse()
	go http.ListenAndServeTLS(*listentls, *cert, *key, nil)
	http.ListenAndServe(*listen, nil)
}