From 82c15dca8ceaa0fd0078cd2ca586006f349840bb Mon Sep 17 00:00:00 2001 From: Dimitri Sokolyuk Date: Sun, 25 Sep 2016 15:34:08 +0200 Subject: Add flags --- wiki.go | 17 ++++++++++++++--- 1 file changed, 14 insertions(+), 3 deletions(-) diff --git a/wiki.go b/wiki.go index 6d5f0f0..8afd5fd 100644 --- a/wiki.go +++ b/wiki.go @@ -1,14 +1,24 @@ 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 @@ -16,7 +26,7 @@ type Page struct { } func (p *Page) fileName() string { - return "data/" + p.Title + return path.Join(*data, p.Title) } func (p *Page) save() error { @@ -45,6 +55,7 @@ func (p *Page) HTML() string { } func main() { - go http.ListenAndServeTLS(":8443", "cert/cert.pem", "cert/key.pem", nil) - http.ListenAndServe(":8080", nil) + flag.Parse() + go http.ListenAndServeTLS(*listentls, *cert, *key, nil) + http.ListenAndServe(*listen, nil) } -- cgit v1.2.3