Fix proxy prefix middleware to manage the prefix properly in the admin section (#2727)

* Fix proxy prefix middleware to manage the prefix properly in the admin section

* Add missing argument

Author: falvaradorodriguez

safe_transaction_service/middlewares/proxy_prefix_middleware.py

@@ -1,14 +1,15 @@
 import logging
 
 from django.http import HttpRequest
+from django.urls import set_script_prefix
 
 
 class ProxyPrefixMiddleware:
     """
     Middleware that adds a prefix from the 'HTTP_X_FORWARDED_PREFIX' header
     to the request's full path.
     This helps when the app is behind a proxy that uses URL prefixes.
-    It modifies the request's build_absolute_uri method to include the prefix.
+    It modifies the request's build_absolute_uri and get_full_path methods to include the prefix.
     """
 
     def __init__(self, get_response):
@@ -19,17 +20,27 @@ def __call__(self, request: HttpRequest):
         prefix = request.META.get("HTTP_X_FORWARDED_PREFIX", "")
         self.logger.debug(f"HTTP_X_FORWARDED_PREFIX:{prefix}")
         if prefix:
-            # Set SCRIPT_NAME so Django generates all URLs with the prefix
             request.META["SCRIPT_NAME"] = prefix
-            # Remove prefix from path_info so Django can match URL patterns
-            if request.path_info.startswith(prefix):
-                request.path_info = request.path_info[len(prefix) :]
+            set_script_prefix(prefix)
+
+            original_get_full_path = request.get_full_path
+
+            def patched_get_full_path(force_append_slash=False):
+                path = original_get_full_path(force_append_slash)
+                if path.startswith(prefix):
+                    return path
+                return prefix + path
+
+            request.get_full_path = patched_get_full_path
 
             original_build_absolute_uri = request.build_absolute_uri
 
             def patched_build_absolute_uri(location=None):
                 uri = original_build_absolute_uri(location)
-                return uri.replace(request.get_host(), request.get_host() + prefix, 1)
+                host_with_prefix = request.get_host() + prefix
+                if host_with_prefix in uri:
+                    return uri
+                return uri.replace(request.get_host(), host_with_prefix, 1)
 
             request.build_absolute_uri = patched_build_absolute_uri
 

safe_transaction_service/middlewares/tests/test_middlewares.py

@@ -18,6 +18,7 @@ def test_proxy_prefix_middleware(self):
         get_response.assert_called_once_with(request)
         self.assertEqual(response, "middleware_response")
         self.assertEqual(request.build_absolute_uri(), "http://testserver/test/path")
+        self.assertEqual(request.get_full_path(), "/test/path")
 
         get_response.reset_mock()
         prefix = "/prefix"
@@ -33,3 +34,4 @@ def test_proxy_prefix_middleware(self):
         self.assertEqual(
             request.build_absolute_uri(), "http://testserver/prefix/test/path"
         )
+        self.assertEqual(request.get_full_path(), "/prefix/test/path")