diff options
Diffstat (limited to 'server.go')
-rw-r--r-- | server.go | 17 |
1 files changed, 16 insertions, 1 deletions
@@ -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) |