Fix APIGW path with stage

APIGW HTTP has started adding the stage to the path in the event. This change checks if the stage is already a prefix in the path, and skips adding it if so.

Signed-off-by: David Calavera <[email protected]>

Author: calavera

lambda-http/src/request.rs

@@ -327,10 +327,17 @@ fn into_websocket_request(ag: ApiGatewayWebsocketProxyRequest) -> http::Request<
 
 #[cfg(any(feature = "apigw_rest", feature = "apigw_http", feature = "apigw_websockets"))]
 fn apigw_path_with_stage(stage: &Option<String>, path: &str) -> String {
-    match stage {
-        None => path.into(),
-        Some(stage) if stage == "$default" => path.into(),
-        Some(stage) => format!("/{stage}{path}"),
+    let stage = match stage {
+        None => return path.into(),
+        Some(stage) if stage == "$default" => return path.into(),
+        Some(stage) => stage,
+    };
+
+    let prefix = format!("/{stage}/");
+    if path.starts_with(&prefix) {
+        path.into()
+    } else {
+        format!("/{stage}{path}")
     }
 }
 
@@ -768,4 +775,25 @@ mod tests {
         let url = build_request_uri("/path with spaces/and multiple segments", &HeaderMap::new(), None, None);
         assert_eq!("/path%20with%20spaces/and%20multiple%20segments", url);
     }
+
+    #[test]
+    fn deserializes_apigw_http_request_with_stage_in_path() {
+        let input = include_str!("../tests/data/apigw_v2_proxy_request_with_stage_in_path.json");
+        let result = from_str(input);
+        assert!(
+            result.is_ok(),
+            "event was not parsed as expected {result:?} given {input}"
+        );
+        let req = result.expect("failed to parse request");
+        assert_eq!("/Prod/my/path", req.uri().path());
+        assert_eq!("/Prod/my/path", req.raw_http_path());
+    }
+
+    #[test]
+    fn test_apigw_path_with_stage() {
+        assert_eq!("/path", apigw_path_with_stage(&None, "/path"));
+        assert_eq!("/path", apigw_path_with_stage(&Some("$default".into()), "/path"));
+        assert_eq!("/Prod/path", apigw_path_with_stage(&Some("Prod".into()), "/Prod/path"));
+        assert_eq!("/Prod/path", apigw_path_with_stage(&Some("Prod".into()), "/path"));
+    }
 }

lambda-http/tests/data/apigw_v2_proxy_request_with_stage_in_path.json

@@ -0,0 +1,57 @@
+{
+  "version": "2.0",
+  "routeKey": "Prod",
+  "rawPath": "/Prod/my/path",
+  "rawQueryString": "parameter1=value1&parameter1=value2&parameter2=value",
+  "cookies": [
+    "cookie1=value1",
+    "cookie2=value2"
+  ],
+  "headers": {
+    "Header1": "value1",
+    "Header2": "value2"
+  },
+  "queryStringParameters": {
+    "parameter1": "value1,value2",
+    "parameter2": "value"
+  },
+  "requestContext": {
+    "accountId": "123456789012",
+    "apiId": "api-id",
+    "authorizer": {
+      "jwt": {
+        "claims": {
+          "claim1": "value1",
+          "claim2": "value2"
+        },
+        "scopes": [
+          "scope1",
+          "scope2"
+        ]
+      }
+    },
+    "domainName": "id.execute-api.us-east-1.amazonaws.com",
+    "domainPrefix": "id",
+    "http": {
+      "method": "POST",
+      "path": "/Prod/my/path",
+      "protocol": "HTTP/1.1",
+      "sourceIp": "IP",
+      "userAgent": "agent"
+    },
+    "requestId": "id",
+    "routeKey": "Prod",
+    "stage": "Prod",
+    "time": "12/Mar/2020:19:03:58 +0000",
+    "timeEpoch": 1583348638390
+  },
+  "body": "Hello from Lambda",
+  "pathParameters": {
+    "parameter1": "value1"
+  },
+  "isBase64Encoded": false,
+  "stageVariables": {
+    "stageVariable1": "value1",
+    "stageVariable2": "value2"
+  }
+}