new: add an option to bind to a specific address

Author: ybizeul

README.md

@@ -51,6 +51,7 @@ the filesystem
 |---------------|-------------|
 | `YBF_DATA_DIR` | points to an alternative direcotry to store data, default is `./data/` in current directory. |
 | `YBF_HTTP_PORT` | TCP port to run the server, default is `8080`. |
+| `YBF_LISTEN_ADDR` | IP address to bind, default is `0.0.0.0`. |
 
 ### Installation
 

cmd/ybfeed/main.go

@@ -9,6 +9,7 @@ import (
 )
 
 var HTTP_PORT int
+var LISTEN_ADDR string
 var DEBUG bool
 var dataDir string
 var maxBodySize int
@@ -30,6 +31,14 @@ func main() {
 				Usage:       "TCP Port to listen",
 				Destination: &HTTP_PORT,
 			},
+			&cli.StringFlag{
+				Name:        "address",
+				Aliases:     []string{"b"},
+				Value:       "0.0.0.0",
+				EnvVars:     []string{"YBF_LISTEN_ADDR"},
+				Usage:       "IP address to bind",
+				Destination: &LISTEN_ADDR,
+			},
 			&cli.BoolFlag{
 				Name:        "debug",
 				Value:       false,
@@ -83,6 +92,7 @@ func run() {
 	api.Version = version
 	api.MaxBodySize = maxBodySize * 1024 * 1024
 	api.HttpPort = HTTP_PORT
+	api.ListenAddr = LISTEN_ADDR
 
 	api.StartServer()
 }

internal/handlers/handlers.go

@@ -85,6 +85,7 @@ type ApiHandler struct {
 	MaxBodySize      int
 	Config           APIConfig
 	HttpPort         int
+	ListenAddr       string
 	WebSocketManager *feed.WebSocketManager
 	FeedManager      *feed.FeedManager
 }
@@ -164,7 +165,7 @@ func (api *ApiHandler) WriteConfig() error {
 }
 func (api *ApiHandler) StartServer() {
 	r := api.GetServer()
-	err := http.ListenAndServe(fmt.Sprintf(":%d", api.HttpPort), r)
+	err := http.ListenAndServe(fmt.Sprintf("%s:%d", api.ListenAddr, api.HttpPort), r)
 	if err != nil {
 		logger.Error("Unable to start HTTP server",
 			slog.String("error", err.Error()))
@@ -209,6 +210,7 @@ func (api *ApiHandler) GetServer() *chi.Mux {
 		slog.String("version", api.Version),
 		slog.String("data_dir", api.BasePath),
 		slog.Int("port", api.HttpPort),
+		slog.String("address", api.ListenAddr),
 		slog.Int("max-upload-size", api.MaxBodySize))
 
 	return r