feat: make port configurable via command-line flag

1 files changed, 6 insertions, 2 deletions

diff --git a/main.go b/main.go
index 3512c24..ec799df 100644
--- a/main.go
+++ b/main.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"flag"
 	"io"
 	"log"
 	"net/http"
@@ -14,15 +15,18 @@ func (h *handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 }
 
 func main() {
+	port := flag.String("port", "8080", "Default localhost port")
+	flag.Parse()
+
 	mux := http.NewServeMux()
 
 	mux.Handle("/{$}", &handler{})
 
 	srv := http.Server{
-		Addr:    ":8080",
+		Addr:    ":" + *port,
 		Handler: mux,
 	}
 
-	log.Println("Serving on port 8080...")
+	log.Printf("Serving on port %s...\n", *port)
 	log.Fatal(srv.ListenAndServe())
 }