Add flag to specify custom address

dillera · Aug 27, 2019 · 806a3cf · 806a3cf
1 parent c22f904
commit 806a3cf
Show file tree

Hide file tree

Showing 3 changed files with 65 additions and 63 deletions.
diff --git a/internal/cmd/serve.go b/internal/cmd/serve.go
@@ -17,14 +17,16 @@ func serveCmd() *cobra.Command {
 	}
 
 	cmd.Flags().IntP("port", "p", 8080, "Port that used by server")
+	cmd.Flags().StringP("address", "a", "", "Address the server listens to")
 
 	return cmd
 }
 
 func serveHandler(cmd *cobra.Command, args []string) {
 	port, _ := cmd.Flags().GetInt("port")
+	address, _ := cmd.Flags().GetString("address")
 
-	err := webserver.ServeApp(db, dataDir, port)
+	err := webserver.ServeApp(db, dataDir, address, port)
 	if err != nil {
 		logrus.Fatalf("Server error: %v\n", err)
 	}

diff --git a/internal/webserver/assets-prod.go b/internal/webserver/assets-prod.go
diff --git a/internal/webserver/server.go b/internal/webserver/server.go
@@ -15,7 +15,7 @@ import (
 var httpClient = &http.Client{Timeout: time.Minute}
 
 // ServeApp serves wb interface in specified port
-func ServeApp(DB database.DB, dataDir string, port int) error {
+func ServeApp(DB database.DB, dataDir string, address string, port int) error {
 	// Create handler
 	hdl := handler{
 		DB:           DB,
@@ -68,7 +68,7 @@ func ServeApp(DB database.DB, dataDir string, port int) error {
 	}
 
 	// Create server
-	url := fmt.Sprintf(":%d", port)
+	url := fmt.Sprintf("%s:%d", address, port)
 	svr := &http.Server{
 		Addr:         url,
 		Handler:      router,