Support multiple IPs in header

1 files changed, 4 insertions, 1 deletions

diff --git a/app.py b/app.py
index 805ca8a..caa23eb 100644
--- a/app.py
+++ b/app.py
@@ -5,7 +5,10 @@ app = Flask(__name__)
 
 def get_ip(request):
     if 'PROD' in environ:
-        return request.headers.get('X-Forwarded-For')
+        # Heroku routing layers masks the original IP, fetch it from the custom header
+        # And make sure we just take the first IP in case other proxies were in the way
+        ips = request.headers.get('X-Forwarded-For')
+        return ips.split(',')[0]
     else:
         return request.remote_addr