-
Notifications
You must be signed in to change notification settings - Fork 475
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
2e94f8b
commit 9941893
Showing
3 changed files
with
266 additions
and
268 deletions.
There are no files selected for viewing
177 changes: 88 additions & 89 deletions
177
Tools/LambdaTestTool-v2/src/Amazon.Lambda.TestTool/HttpContextExtensions.cs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,114 +1,113 @@ | ||
| using Amazon.Lambda.APIGatewayEvents; | ||
| namespace Amazon.Lambda.TestTool; | ||
|
|
||
| using Amazon.Lambda.APIGatewayEvents; | ||
| using System.Text; | ||
| using System.Web; | ||
| using static Amazon.Lambda.APIGatewayEvents.APIGatewayHttpApiV2ProxyRequest; | ||
|
|
||
| namespace Amazon.Lambda.TestTool | ||
| /// <summary> | ||
| /// Provides extension methods to translate an <see cref="HttpContext"/> to different types of API Gateway requests. | ||
| /// </summary> | ||
| public static class HttpContextExtensions | ||
| { | ||
| private static IHttpRequestUtility _httpRequestUtility = new HttpRequestUtility(); | ||
| private static IRouteConfigurationParser _routeConfigurationParser; | ||
|
|
||
| public static void SetHttpRequestUtility(IHttpRequestUtility httpRequestUtility) | ||
| { | ||
| _httpRequestUtility = httpRequestUtility ?? throw new ArgumentNullException(nameof(httpRequestUtility)); | ||
| } | ||
|
|
||
| public static void SetRouteConfigurationParser(IRouteConfigurationParser routeConfigurationParser) | ||
| { | ||
| _routeConfigurationParser = routeConfigurationParser ?? throw new ArgumentNullException(nameof(routeConfigurationParser)); | ||
| } | ||
|
|
||
| /// <summary> | ||
| /// Provides extension methods to translate an <see cref="HttpContext"/> to different types of API Gateway requests. | ||
| /// Translates an <see cref="HttpContext"/> to an <see cref="APIGatewayHttpApiV2ProxyRequest"/>. | ||
| /// </summary> | ||
| public static class HttpContextExtensions | ||
| /// <param name="context">The <see cref="HttpContext"/> to be translated.</param> | ||
| /// <returns>An <see cref="APIGatewayHttpApiV2ProxyRequest"/> object representing the translated request.</returns> | ||
| public static APIGatewayHttpApiV2ProxyRequest ToApiGatewayHttpV2Request( | ||
| this HttpContext context) | ||
| { | ||
| private static IHttpRequestUtility _httpRequestUtility = new HttpRequestUtility(); | ||
| private static IRouteConfigurationParser _routeConfigurationParser; | ||
| var request = context.Request; | ||
|
|
||
| public static void SetHttpRequestUtility(IHttpRequestUtility httpRequestUtility) | ||
| { | ||
| _httpRequestUtility = httpRequestUtility ?? throw new ArgumentNullException(nameof(httpRequestUtility)); | ||
| } | ||
| var matchedConfig = _routeConfigurationParser.GetRouteConfig(request.Method, request.Path); | ||
| var pathParameters = _routeConfigurationParser.ExtractPathParameters(matchedConfig, request.Path); | ||
|
|
||
| public static void SetRouteConfigurationParser(IRouteConfigurationParser routeConfigurationParser) | ||
| { | ||
| _routeConfigurationParser = routeConfigurationParser ?? throw new ArgumentNullException(nameof(routeConfigurationParser)); | ||
| } | ||
| var (headers, _) = _httpRequestUtility.ExtractHeaders(request.Headers); | ||
| var (queryStringParameters, _) = _httpRequestUtility.ExtractQueryStringParameters(request.Query); | ||
|
|
||
| /// <summary> | ||
| /// Translates an <see cref="HttpContext"/> to an <see cref="APIGatewayHttpApiV2ProxyRequest"/>. | ||
| /// </summary> | ||
| /// <param name="context">The <see cref="HttpContext"/> to be translated.</param> | ||
| /// <returns>An <see cref="APIGatewayHttpApiV2ProxyRequest"/> object representing the translated request.</returns> | ||
| public static APIGatewayHttpApiV2ProxyRequest ToApiGatewayHttpV2Request( | ||
| this HttpContext context) | ||
| var httpApiV2ProxyRequest = new APIGatewayHttpApiV2ProxyRequest | ||
| { | ||
| var request = context.Request; | ||
|
|
||
| var matchedConfig = _routeConfigurationParser.GetRouteConfig(request.Method, request.Path); | ||
| var pathParameters = _routeConfigurationParser.ExtractPathParameters(matchedConfig, request.Path); | ||
|
|
||
| var (headers, _) = _httpRequestUtility.ExtractHeaders(request.Headers); | ||
| var (queryStringParameters, _) = _httpRequestUtility.ExtractQueryStringParameters(request.Query); | ||
|
|
||
| var httpApiV2ProxyRequest = new APIGatewayHttpApiV2ProxyRequest | ||
| RouteKey = $"{request.Method} {matchedConfig.Path}", | ||
| RawPath = request.Path, | ||
| RawQueryString = request.QueryString.Value, | ||
| Cookies = request.Cookies.Select(c => $"{c.Key}={c.Value}").ToArray(), | ||
| Headers = headers, | ||
| QueryStringParameters = queryStringParameters, | ||
| PathParameters = pathParameters ?? new Dictionary<string, string>(), | ||
| Body = _httpRequestUtility.ReadRequestBody(request), | ||
| IsBase64Encoded = false, | ||
| RequestContext = new ProxyRequestContext | ||
| { | ||
| RouteKey = $"{request.Method} {matchedConfig.Path}", | ||
| RawPath = request.Path, | ||
| RawQueryString = request.QueryString.Value, | ||
| Cookies = request.Cookies.Select(c => $"{c.Key}={c.Value}").ToArray(), | ||
| Headers = headers, | ||
| QueryStringParameters = queryStringParameters, | ||
| PathParameters = pathParameters ?? new Dictionary<string, string>(), | ||
| Body = _httpRequestUtility.ReadRequestBody(request), | ||
| IsBase64Encoded = false, | ||
| RequestContext = new ProxyRequestContext | ||
| Http = new HttpDescription | ||
| { | ||
| Http = new HttpDescription | ||
| { | ||
| Method = request.Method, | ||
| Path = request.Path, | ||
| Protocol = request.Protocol | ||
| }, | ||
| RouteKey = $"{request.Method} {matchedConfig.Path}" | ||
| Method = request.Method, | ||
| Path = request.Path, | ||
| Protocol = request.Protocol | ||
| }, | ||
| Version = "2.0" | ||
| }; | ||
| RouteKey = $"{request.Method} {matchedConfig.Path}" | ||
| }, | ||
| Version = "2.0" | ||
| }; | ||
|
|
||
| if (_httpRequestUtility.IsBinaryContent(request.ContentType)) | ||
| { | ||
| httpApiV2ProxyRequest.Body = Convert.ToBase64String(System.Text.Encoding.UTF8.GetBytes(httpApiV2ProxyRequest.Body)); | ||
| httpApiV2ProxyRequest.IsBase64Encoded = true; | ||
| } | ||
|
|
||
| return httpApiV2ProxyRequest; | ||
| if (_httpRequestUtility.IsBinaryContent(request.ContentType)) | ||
| { | ||
| httpApiV2ProxyRequest.Body = Convert.ToBase64String(System.Text.Encoding.UTF8.GetBytes(httpApiV2ProxyRequest.Body)); | ||
| httpApiV2ProxyRequest.IsBase64Encoded = true; | ||
| } | ||
|
|
||
| /// <summary> | ||
| /// Translates an <see cref="HttpContext"/> to an <see cref="APIGatewayProxyRequest"/>. | ||
| /// </summary> | ||
| /// <param name="context">The <see cref="HttpContext"/> to be translated.</param> | ||
| /// <returns>An <see cref="APIGatewayProxyRequest"/> object representing the translated request.</returns> | ||
| public static APIGatewayProxyRequest ToApiGatewayRequest( | ||
| this HttpContext context) | ||
| { | ||
| var request = context.Request; | ||
| return httpApiV2ProxyRequest; | ||
| } | ||
|
|
||
| /// <summary> | ||
| /// Translates an <see cref="HttpContext"/> to an <see cref="APIGatewayProxyRequest"/>. | ||
| /// </summary> | ||
| /// <param name="context">The <see cref="HttpContext"/> to be translated.</param> | ||
| /// <returns>An <see cref="APIGatewayProxyRequest"/> object representing the translated request.</returns> | ||
| public static APIGatewayProxyRequest ToApiGatewayRequest( | ||
| this HttpContext context) | ||
| { | ||
| var request = context.Request; | ||
|
|
||
| var matchedConfig = _routeConfigurationParser.GetRouteConfig(request.Method, request.Path); | ||
| var pathParameters = _routeConfigurationParser.ExtractPathParameters(matchedConfig, request.Path); | ||
| var matchedConfig = _routeConfigurationParser.GetRouteConfig(request.Method, request.Path); | ||
| var pathParameters = _routeConfigurationParser.ExtractPathParameters(matchedConfig, request.Path); | ||
|
|
||
| var (headers, multiValueHeaders) = _httpRequestUtility.ExtractHeaders(request.Headers); | ||
| var (queryStringParameters, multiValueQueryStringParameters) = _httpRequestUtility.ExtractQueryStringParameters(request.Query); | ||
| var (headers, multiValueHeaders) = _httpRequestUtility.ExtractHeaders(request.Headers); | ||
| var (queryStringParameters, multiValueQueryStringParameters) = _httpRequestUtility.ExtractQueryStringParameters(request.Query); | ||
|
|
||
| var proxyRequest = new APIGatewayProxyRequest | ||
| { | ||
| Resource = matchedConfig.Path, | ||
| Path = HttpUtility.UrlEncode(request.Path), | ||
| HttpMethod = request.Method, | ||
| Headers = headers, | ||
| MultiValueHeaders = multiValueHeaders, | ||
| QueryStringParameters = queryStringParameters, | ||
| MultiValueQueryStringParameters = multiValueQueryStringParameters, | ||
| PathParameters = pathParameters ?? new Dictionary<string, string>(), | ||
| Body = _httpRequestUtility.ReadRequestBody(request), | ||
| IsBase64Encoded = false | ||
| }; | ||
|
|
||
| if (_httpRequestUtility.IsBinaryContent(request.ContentType)) | ||
| { | ||
| proxyRequest.Body = Convert.ToBase64String(Encoding.UTF8.GetBytes(proxyRequest.Body)); | ||
| proxyRequest.IsBase64Encoded = true; | ||
| } | ||
| var proxyRequest = new APIGatewayProxyRequest | ||
| { | ||
| Resource = matchedConfig.Path, | ||
| Path = HttpUtility.UrlEncode(request.Path), | ||
| HttpMethod = request.Method, | ||
| Headers = headers, | ||
| MultiValueHeaders = multiValueHeaders, | ||
| QueryStringParameters = queryStringParameters, | ||
| MultiValueQueryStringParameters = multiValueQueryStringParameters, | ||
| PathParameters = pathParameters ?? new Dictionary<string, string>(), | ||
| Body = _httpRequestUtility.ReadRequestBody(request), | ||
| IsBase64Encoded = false | ||
| }; | ||
|
|
||
| return proxyRequest; | ||
| if (_httpRequestUtility.IsBinaryContent(request.ContentType)) | ||
| { | ||
| proxyRequest.Body = Convert.ToBase64String(Encoding.UTF8.GetBytes(proxyRequest.Body)); | ||
| proxyRequest.IsBase64Encoded = true; | ||
| } | ||
|
|
||
| return proxyRequest; | ||
| } | ||
| } |
Oops, something went wrong.