From c52d6f99bdbb4423fe7ad1133c5c1520f097114d Mon Sep 17 00:00:00 2001 From: Emile Date: Fri, 16 Aug 2024 21:42:03 +0200 Subject: host and port flags --- src/main.go | 18 ++++++++++++++++-- 1 file changed, 16 insertions(+), 2 deletions(-) diff --git a/src/main.go b/src/main.go index 27c066c..6442aa8 100644 --- a/src/main.go +++ b/src/main.go @@ -1,6 +1,8 @@ package main import ( + "flag" + "fmt" "log" "net/http" "os" @@ -8,6 +10,9 @@ import ( "github.com/gorilla/mux" ) +var host string +var port int + const database_file string = "main.db" const salt = "oogha3AiH7taimohreeH8Lexoonea5zi" @@ -15,7 +20,16 @@ var ( globalState *State ) +func initFlags() { + flag.StringVar(&host, "host", "127.0.0.1", "the host to listen on") + flag.StringVar(&host, "h", "127.0.0.1", "the host to listen on (shorthand)") + flag.IntVar(&port, "port", 8080, "the port to listen on") + flag.IntVar(&port, "p", 8080, "the port to listen on (shorthand)") +} + func main() { + initFlags() + flag.Parse() // log init log.Println("[i] Setting up logging...") @@ -70,6 +84,6 @@ func main() { auth_needed.HandleFunc("/battle/{id}", battleSingleHandler) auth_needed.HandleFunc("/battle/{id}/submit", battleSubmitHandler) - log.Println("[i] HTTP Server running on port :8080") - log.Fatal(http.ListenAndServe(":8080", r)) + log.Printf("[i] HTTP Server running on %s:%d\n", host, port) + log.Fatal(http.ListenAndServe(fmt.Sprintf("%s:%d", host, port), r)) } -- cgit 1.4.1