Add cli args for port and address

1 files changed, 20 insertions, 3 deletions

diff --git a/nonsense-time.go b/nonsense-time.go
index ed5faf8..e5a6803 100644
--- a/nonsense-time.go
+++ b/nonsense-time.go
@@ -3,8 +3,11 @@ package main
 import (
 	"context"
 	"encoding/json"
+	"flag"
+	"fmt"
 	"log"
 	"net/http"
+	"os"
 	"time"
 )
 
@@ -79,12 +82,26 @@ func siteStatus(w http.ResponseWriter, req *http.Request) {
 }
 
 func main() {
+
+	port := flag.Int("p", 8080, "the port to listen on")
+	bindAddr := flag.String("b", "", "the adress to bind to (leave empty for all interfaces)")
+
+	flag.Usage = func() {
+		fmt.Fprintf(os.Stderr, "Usage: %s [options]\n", os.Args[0])
+		fmt.Fprintln(os.Stderr, "Options:")
+		flag.PrintDefaults()
+	}
+
+	flag.Parse()
+
+	addr := fmt.Sprintf("%s:%d", *bindAddr, *port)
+
 	mux := http.NewServeMux()
 
 	mux.HandleFunc("GET /vtt/status", vttStatus)
-    mux.HandleFunc("GET /vtt", vttRedirect)
+	mux.HandleFunc("GET /vtt", vttRedirect)
 	mux.HandleFunc("GET /site/status", siteStatus)
 
-	log.Println("Listening on Port 8080")
-	http.ListenAndServe(":8080", mux)
+	log.Println("Listening on ", addr)
+	log.Fatal(http.ListenAndServe(addr, mux))
 }