diff --git a/cmd/web/main.go b/cmd/web/main.go index 07950b6..4f5598b 100644 --- a/cmd/web/main.go +++ b/cmd/web/main.go @@ -1,11 +1,22 @@ package main import ( + "flag" "log" "net/http" ) +type config struct { + addr string + staticDir string +} + func main() { + var cfg config + flag.StringVar(&cfg.addr, "addr", ":4000", "HTTP network address") + flag.StringVar(&cfg.staticDir, "static-dir", "./ui/static", "Path to static assets") + flag.Parse() + mux := http.NewServeMux() fileServer := http.FileServer(http.Dir("./ui/static/")) @@ -14,7 +25,7 @@ func main() { mux.HandleFunc("/snippet/create", snippetCreate) mux.Handle("/static/", http.StripPrefix("/static", fileServer)) - log.Print("Starting server on :4000") - err := http.ListenAndServe(":4000", mux) + log.Printf("Starting server on %s", cfg.addr) + err := http.ListenAndServe(cfg.addr, mux) log.Fatal(err) }