aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorDimitri Sokolyuk <demon@dim13.org>2016-09-25 15:34:08 +0200
committerDimitri Sokolyuk <demon@dim13.org>2016-09-25 15:34:08 +0200
commit82c15dca8ceaa0fd0078cd2ca586006f349840bb (patch)
treea1cb95538c304257b01ec59df7390ec4d457eb4f
parent5f8517c8931ec2de835ce0e318136fb84767ffb3 (diff)
Add flags
-rw-r--r--wiki.go17
1 files 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)
}