populate contexts

bentsku · bentsku · commit 1216a74b292d · 2024-06-28T02:01:53.000+02:00
diff --git a/localstack-core/localstack/services/apigateway/next_gen/execute_api/context.py b/localstack-core/localstack/services/apigateway/next_gen/execute_api/context.py
@@ -65,6 +65,8 @@ class RestApiInvocationContext(RequestContext):
     """The REST API identifier of the invoked API"""
     stage: Optional[str]
     """The REST API stage linked to this invocation"""
+    deployment_id: Optional[str]
+    """The REST API deployment linked to this invocation"""
     region: Optional[str]
     """The region the REST API is living in."""
     account_id: Optional[str]
@@ -87,6 +89,7 @@ def __init__(self, request: Request):
         self.deployment = None
         self.api_id = None
         self.stage = None
+        self.deployment_id = None
         self.account_id = None
         self.region = None
         self.invocation_request = None
diff --git a/localstack-core/localstack/services/apigateway/next_gen/execute_api/handlers/parse.py b/localstack-core/localstack/services/apigateway/next_gen/execute_api/handlers/parse.py
@@ -1,3 +1,4 @@
+import datetime
 import logging
 from collections import defaultdict
 from typing import Optional
@@ -7,6 +8,9 @@
 from werkzeug.datastructures import Headers, MultiDict
 
 from localstack.http import Response
+from localstack.services.apigateway.helpers import REQUEST_TIME_DATE_FORMAT
+from localstack.utils.strings import short_uid
+from localstack.utils.time import timestamp
 
 from ..api import RestApiGatewayHandler, RestApiGatewayHandlerChain
 from ..context import InvocationRequest, RestApiInvocationContext
@@ -32,7 +36,7 @@ def parse_and_enrich(self, context: RestApiInvocationContext):
         context.invocation_request = self.create_invocation_request(context.request)
         # then we can create the ContextVariables, used throughout the invocation as payload and to render authorizer
         # payload, mapping templates and such.
-        context.context_variables = self.create_context_variables(context.invocation_request)
+        context.context_variables = self.create_context_variables(context)
         # then populate the stage variables
         context.stage_variables = self.fetch_stage_variables(context)
 
@@ -108,8 +112,28 @@ def _get_single_and_multi_values_from_headers(
 
         return single_values, multi_values
 
-    def create_context_variables(self, invocation_request: InvocationRequest) -> ContextVariables:
-        context_variables = ContextVariables()
+    @staticmethod
+    def create_context_variables(context: RestApiInvocationContext) -> ContextVariables:
+        invocation_request: InvocationRequest = context.invocation_request
+        domain_name = invocation_request["headers"].get("Host", "")
+        domain_prefix = domain_name.split(".")[0]
+        now = datetime.datetime.now()
+
+        context_variables = ContextVariables(
+            accountId=context.account_id,
+            apiId=context.api_id,
+            deploymentId=context.deployment_id,
+            domainName=domain_name,
+            domainPrefix=domain_prefix,
+            extendedRequestId=short_uid(),
+            httpMethod=invocation_request["http_method"],
+            path=invocation_request["path"],
+            protocol="HTTP/1.1",
+            requestId=short_uid(),
+            requestTime=timestamp(time=now, format=REQUEST_TIME_DATE_FORMAT),
+            requestTimeEpoch=int(now.timestamp() * 1000),
+            stage=context.stage,
+        )
         return context_variables
 
     @staticmethod
diff --git a/localstack-core/localstack/services/apigateway/next_gen/execute_api/handlers/resource_router.py b/localstack-core/localstack/services/apigateway/next_gen/execute_api/handlers/resource_router.py
@@ -18,6 +18,7 @@
 
 from ..api import RestApiGatewayHandler, RestApiGatewayHandlerChain
 from ..context import RestApiInvocationContext
+from ..variables import ContextVariables
 
 LOG = logging.getLogger(__name__)
 
@@ -84,6 +85,7 @@ def route_and_enrich(self, context: RestApiInvocationContext):
         router = self.get_router_for_deployment(context.deployment)
 
         resource, path_parameters = router.match(context)
+        resource: Resource
 
         context.invocation_request["path_parameters"] = path_parameters
         context.resource = resource
@@ -94,6 +96,16 @@ def route_and_enrich(self, context: RestApiInvocationContext):
         )
         context.resource_method = method
 
+        self.update_context_variables_with_resource(context.context_variables, resource)
+
+    @staticmethod
+    def update_context_variables_with_resource(
+        context_variables: ContextVariables, resource: Resource
+    ):
+        # TODO: log updating the context_variables?
+        context_variables["resourcePath"] = resource["path"]
+        context_variables["resourceId"] = resource["id"]
+
     @staticmethod
     @cache
     def get_router_for_deployment(deployment: RestApiDeployment) -> RestAPIResourceRouter:
diff --git a/localstack-core/localstack/services/apigateway/next_gen/execute_api/router.py b/localstack-core/localstack/services/apigateway/next_gen/execute_api/router.py
@@ -64,6 +64,7 @@ def populate_rest_api_invocation_context(
         context.deployment = frozen_deployment
         context.api_id = api_id
         context.stage = stage
+        context.deployment_id = deployment_id
 
 
 class ApiGatewayRouter:
diff --git a/tests/unit/services/apigateway/test_handler_request.py b/tests/unit/services/apigateway/test_handler_request.py
@@ -1,5 +1,5 @@
 import pytest
-from moto.apigateway.models import APIGatewayBackend, apigateway_backends
+from moto.apigateway.models import APIGatewayBackend, Stage, apigateway_backends
 from moto.apigateway.models import RestAPI as MotoRestAPI
 from werkzeug.datastructures import Headers
 
@@ -35,6 +35,7 @@ def dummy_deployment():
     )
 
     moto_backend.apis[TEST_API_ID] = moto_rest_api
+    moto_rest_api.stages[TEST_API_STAGE] = Stage(name=TEST_API_STAGE)
 
     yield freeze_rest_api(
         account_id=TEST_AWS_ACCOUNT_ID,
