added ip_selector

lib/ipinfo-rails.rb

@@ -2,18 +2,25 @@
 
 require 'rack'
 require 'ipinfo'
+require 'ipinfo-rails/ipselector/default_ipselector'
 
 class IPinfoMiddleware
     def initialize(app, cache_options = {})
         @app = app
         @token = cache_options.fetch(:token, nil)
         @ipinfo = IPinfo.create(@token, cache_options)
         @filter = cache_options.fetch(:filter, nil)
+        @ip_selector = cache_options.fetch(:ip_selector, nil)
     end
 
     def call(env)
         env['called'] = 'yes'
         request = Rack::Request.new(env)
+        ip_selected = if @ip_selector.nil? 
+                        DefaultIPSelector.new(request)
+                      else
+                        @ip_selector.new(request)
+                      end
 
         filtered = if @filter.nil?
                        is_bot(request)
@@ -24,7 +31,7 @@ def call(env)
         if filtered
             env['ipinfo'] = nil
         else
-            ip = request.ip
+            ip = ip_selected.get_ip()
             env['ipinfo'] = @ipinfo.details(ip)
         end
 
@@ -41,4 +48,4 @@ def is_bot(request)
             false
         end
     end
-end
+end

lib/ipinfo-rails/ipselector/default_ipselector.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+require 'ipinfo-rails/ipselector/ipselector_interface'
+
+class DefaultIPSelector
+    include IPSelectorInterface
+    def initialize(request)
+        @request = request
+    end
+
+    def get_ip()
+        return @request.ip
+    end
+
+end

lib/ipinfo-rails/ipselector/ipselector_interface.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+module IPSelectorInterface
+    class InterfaceNotImplemented < StandardError; end
+    def get_ip()
+        raise InterfaceNotImplemented
+    end
+end

lib/ipinfo-rails/ipselector/xforwarded_ipselector.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+require 'ipinfo-rails/ipselector/ipselector_interface'
+
+class XForwardedIPSelector
+    include IPSelectorInterface
+    def initialize(request)
+        @request = request
+    end
+
+    def get_ip()
+        return @request.env['HTTP_X_FORWARDED_FOR']
+    end
+
+end