normalize client ip

docusealco · Jun 3, 2024 · 98c04ea · 98c04ea
1 parent c35dfea
commit 98c04ea
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/config/application.rb b/config/application.rb
@@ -11,6 +11,7 @@
 require 'rails/health_controller'
 
 require_relative '../lib/api_path_consider_json_middleware'
+require_relative '../lib/normalize_client_ip_middleware'
 
 Bundler.require(*Rails.groups)
 
@@ -32,6 +33,7 @@ class Application < Rails::Application
     config.action_view.frozen_string_literal = true
 
     config.middleware.insert_before ActionDispatch::Static, Rack::Deflater
+    config.middleware.insert_before ActionDispatch::Static, NormalizeClientIpMiddleware
     config.middleware.insert_before ActionDispatch::Static, ApiPathConsiderJsonMiddleware
 
     config.generators.system_tests = nil

diff --git a/lib/normalize_client_ip_middleware.rb b/lib/normalize_client_ip_middleware.rb
@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+class NormalizeClientIpMiddleware
+  def initialize(app)
+    @app = app
+  end
+
+  def call(env)
+    if env['HTTP_CLIENT_IP'].present? && env['HTTP_X_CLIENT_IP'].present? &&
+       env['HTTP_CLIENT_IP'].starts_with?("#{env['HTTP_X_CLIENT_IP']}:")
+      env['HTTP_CLIENT_IP'] = env['HTTP_X_CLIENT_IP']
+    end
+
+    @app.call(env)
+  end
+end