host and port flags

1 files 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))
 }