def to_invocation_context(
request: Request,
url_params: Dict[str, Any] = None) -> ApiInvocationContext:
"""
Converts an HTTP Request object into an ApiInvocationContext.
:param request: the original request
:param url_params: the parameters extracted from the URL matching rules
:return: the ApiInvocationContext
"""
method = request.method
path = request.full_path if request.query_string else request.path
data = restore_payload(request)
headers = Headers(request.headers)
# adjust the X-Forwarded-For header
x_forwarded_for = headers.getlist("X-Forwarded-For")
x_forwarded_for.append(request.remote_addr)
x_forwarded_for.append(request.host)
headers["X-Forwarded-For"] = ", ".join(x_forwarded_for)
# set the x-localstack-edge header, it is used to parse the domain
headers[HEADER_LOCALSTACK_EDGE_URL] = request.host_url.strip("/")
# FIXME: Use the already parsed url params instead of parsing them into the ApiInvocationContext part-by-part.
# We already would have all params at hand to avoid _all_ the parsing, but the parsing
# has side-effects (f.e. setting the region in a thread local)!
# It would be best to use a small (immutable) context for the already parsed params and the Request object
# and use it everywhere.
return ApiInvocationContext(method,
path,
data,
headers,
stage=url_params.get("stage"))
def forward_request(self, method, path, data, headers):
invocation_context = ApiInvocationContext(method, path, data, headers)
forwarded_for = headers.get(HEADER_LOCALSTACK_EDGE_URL, "")
if re.match(PATH_REGEX_USER_REQUEST, path) or "execute-api" in forwarded_for:
result = invoke_rest_api_from_request(invocation_context)
if result is not None:
return result
if helpers.is_test_invoke_method(method, path):
return self._handle_test_invoke_method(invocation_context)
return super().forward_request(method, path, data, headers)
def forward_request(self, method, path, data, headers):
invocation_context = ApiInvocationContext(method, path, data, headers)
forwarded_for = headers.get(HEADER_LOCALSTACK_EDGE_URL, "")
if re.match(PATH_REGEX_USER_REQUEST,
path) or "execute-api" in forwarded_for:
result = invoke_rest_api_from_request(invocation_context)
if result is not None:
return result
data = data and json.loads(to_str(data))
if re.match(PATH_REGEX_AUTHORIZERS, path):
return handle_authorizers(method, path, data, headers)
if re.match(PATH_REGEX_DOC_PARTS, path):
return handle_documentation_parts(method, path, data, headers)
if re.match(PATH_REGEX_VALIDATORS, path):
return handle_validators(method, path, data, headers)
if re.match(PATH_REGEX_RESPONSES, path):
return handle_gateway_responses(method, path, data, headers)
if re.match(PATH_REGEX_PATH_MAPPINGS, path):
return handle_base_path_mappings(method, path, data, headers)
if helpers.is_test_invoke_method(method, path):
# if call is from test_invoke_api then use http_method to find the integration,
# as test_invoke_api makes a POST call to request the test invocation
match = re.match(PATH_REGEX_TEST_INVOKE_API, path)
invocation_context.method = match[3]
if data:
orig_data = data
path_with_query_string = orig_data.get("pathWithQueryString",
None)
if path_with_query_string:
invocation_context.path_with_query_string = path_with_query_string
invocation_context.data = data.get("body")
invocation_context.headers = orig_data.get("headers", {})
result = invoke_rest_api_from_request(invocation_context)
result = {
"status": result.status_code,
"body": to_str(result.content),
"headers": dict(result.headers),
}
return result
return True