From e4cc4387a6d408521fe6155b36b004bda212ad3e Mon Sep 17 00:00:00 2001
From: ache <ache@ache.one>
Date: Tue, 12 Feb 2019 12:50:54 +0100
Subject: Arg management

---
 server.go | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

(limited to 'server.go')

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)
-- 
cgit v1.2.3