Port configuration

1 files changed, 14 insertions, 2 deletions

diff --git a/server.go b/server.go
index aed70d8..aa4a6a5 100644
--- a/server.go
+++ b/server.go
@@ -396,18 +396,23 @@ func handleMethod( w http.ResponseWriter, r *http.Request) {
 func main() {
   var dirRoot string
   var help, version bool
+  var port = -1
 
   helpString := `This is the help`
   versionString := "0.0.0"
 
   const (
     defaultRoot   = "."
+    defaultPort   = 8080
+    portUsage    = "port to listen"
     usageDir     = "the directory to serve"
     usageHelp    = "show some help"
     usageVersion = "show the version"
   )
   flag.StringVar(&dirRoot, "directory", "", usageDir)
   flag.StringVar(&dirRoot, "d", "", usageDir + " (shorthand)")
+  flag.IntVar(&port, "port", 0, portUsage)
+  flag.IntVar(&port, "p", 0, portUsage + " (shorthand)")
   flag.BoolVar(&help, "help", false, usageHelp)
   flag.BoolVar(&help, "h", false, usageHelp+" (shorthand)")
   flag.BoolVar(&version, "version", false, usageVersion)
@@ -432,6 +437,13 @@ func main() {
     }
   }
 
+  if port == 0  {
+    port = config.Config.Port
+    if port == 0 {
+      port = defaultPort
+    }
+  }
+
   if dirRoot == "" {
     dirRoot = config.Config.RootPath
     if dirRoot == "" {
@@ -461,9 +473,9 @@ func main() {
     return
   }
 
-  fmt.Println("Launch server on port:", config.Config.Port)
+  fmt.Println("Launch server on port:", port)
   http.HandleFunc("/", handleMethod)
 
-  log.Fatal(http.ListenAndServe(":" + strconv.Itoa(config.Config.Port), nil))
+  log.Fatal(http.ListenAndServe(":" + strconv.Itoa(port), nil))
   fmt.Println("Bye bye")
 }