feat: make usage of X-Real-IP and X-Forwarded-For header configurable

Author: d-Rickyy-b

config.sample.yaml

@@ -13,5 +13,6 @@ prometheus:
   listen_port: 8080
   metrics_url: "/metrics"
   expose_system_metrics: false
+  real_ip: false
   whitelist:
     - "127.0.0.1/8"

internal/config/config.go

@@ -19,6 +19,7 @@ var (
 type ServerConfig struct {
 	ListenAddr string   `yaml:"listen_addr"`
 	ListenPort int      `yaml:"listen_port"`
+	RealIP     bool     `yaml:"real_ip"`
 	Whitelist  []string `yaml:"whitelist"`
 }
 

internal/web/server.go

@@ -242,6 +242,10 @@ func NewMetricsServer(networkIf string, port int, certPath, keyPath string) *Web
 	server.initServer()
 	server.routes.Use(middleware.Recoverer)
 
+	if config.AppConfig.Prometheus.RealIP {
+		server.routes.Use(middleware.RealIP)
+	}
+
 	// Enable IP whitelist if configured
 	if len(config.AppConfig.Prometheus.Whitelist) > 0 {
 		server.routes.Use(IPWhitelist)
@@ -262,6 +266,10 @@ func NewWebsocketServer(networkIf string, port int, certPath, keyPath string) *W
 	}
 	server.initServer()
 
+	if config.AppConfig.Webserver.RealIP {
+		server.routes.Use(middleware.RealIP)
+	}
+
 	// Enable IP whitelist if configured
 	if len(config.AppConfig.Webserver.Whitelist) > 0 {
 		server.routes.Use(IPWhitelist)