Arg management

1 files changed, 16 insertions, 1 deletions

diff --git a/server.go b/server.go
index c7e55a7..b06692e 100644
--- a/server.go
+++ b/server.go
@@ -6,6 +6,7 @@ import (
   "net/http"
   "os"
   "html/template"
+  "flag"
 )
 
 var tmpl *template.Template
@@ -101,8 +102,22 @@ func handleMethod( w http.ResponseWriter, r *http.Request) {
 }
 
 func main() {
+  var dir string
 
-	tmpl = template.Must(template.ParseFiles("dir.html"))
+  const (
+      defaultDir = "."
+      usageDir   = "the directory to serve"
+  )
+  flag.StringVar(&dir, "directory", defaultDir, usageDir)
+  flag.StringVar(&dir, "d", defaultDir, usageDir+" (shorthand)")
+
+ flag.Parse()
+
+ if dir == "" {
+  dir = defaultDir
+ }
+
+  tmpl = template.Must(template.ParseFiles("dir.html"))
 
   fmt.Println("Launch server")
   http.HandleFunc("/", handleMethod)