fixes #4.

m1k1o · Oct 30, 2022 · edddccb · edddccb
1 parent 872e96a
commit edddccb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/internal/config/config.go b/internal/config/config.go
@@ -98,7 +98,7 @@ func (Server) Init(cmd *cobra.Command) error {
 		return err
 	}
 
-	cmd.PersistentFlags().Bool("proxy", false, "allow reverse proxies")
+	cmd.PersistentFlags().Bool("proxy", false, "allow reverse proxies: X-Forwarded-For headers will be used to determine the client IP")
 	if err := viper.BindPFlag("proxy", cmd.PersistentFlags().Lookup("proxy")); err != nil {
 		return err
 	}

diff --git a/internal/http/http.go b/internal/http/http.go
@@ -29,6 +29,10 @@ func New(config *config.Server) *HttpManagerCtx {
 	router.Use(middleware.RequestLogger(&logformatter{logger}))
 	router.Use(middleware.Recoverer) // Recover from panics without crashing server
 
+	if config.Proxy {
+		router.Use(middleware.RealIP)
+	}
+
 	// serve static files
 	if config.Static != "" {
 		fs := http.FileServer(http.Dir(config.Static))