-
Notifications
You must be signed in to change notification settings - Fork 249
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(core,api): IAM auth mode for HTTP requests (REST and GQL) (#1893)
- Loading branch information
Travis Sheppard
authored
Jul 21, 2022
1 parent
7a82e9e
commit 2fb4c0c
Showing
12 changed files
with
538 additions
and
52 deletions.
There are no files selected for viewing
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
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
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
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
110 changes: 110 additions & 0 deletions
110
packages/api/amplify_api/lib/src/decorators/authorize_http_request.dart
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 |
|---|---|---|
| @@ -0,0 +1,110 @@ | ||
| // Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. | ||
| // | ||
| // Licensed under the Apache License, Version 2.0 (the "License"); | ||
| // you may not use this file except in compliance with the License. | ||
| // You may obtain a copy of the License at | ||
| // | ||
| // http://www.apache.org/licenses/LICENSE-2.0 | ||
| // | ||
| // Unless required by applicable law or agreed to in writing, software | ||
| // distributed under the License is distributed on an "AS IS" BASIS, | ||
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| // See the License for the specific language governing permissions and | ||
| // limitations under the License. | ||
|
|
||
| import 'dart:async'; | ||
|
|
||
| import 'package:amplify_core/amplify_core.dart'; | ||
| import 'package:http/http.dart' as http; | ||
| import 'package:meta/meta.dart'; | ||
|
|
||
| /// Transforms an HTTP request according to auth providers that match the endpoint | ||
| /// configuration. | ||
| @internal | ||
| Future<http.BaseRequest> authorizeHttpRequest(http.BaseRequest request, | ||
| {required AWSApiConfig endpointConfig, | ||
| required AmplifyAuthProviderRepository authProviderRepo}) async { | ||
| if (request.headers.containsKey(AWSHeaders.authorization)) { | ||
| return request; | ||
| } | ||
| final authType = endpointConfig.authorizationType; | ||
|
|
||
| switch (authType) { | ||
| case APIAuthorizationType.apiKey: | ||
| final authProvider = _validateAuthProvider( | ||
| authProviderRepo | ||
| .getAuthProvider(APIAuthorizationType.apiKey.authProviderToken), | ||
| authType); | ||
| final apiKey = endpointConfig.apiKey; | ||
| if (apiKey == null) { | ||
| throw const ApiException( | ||
| 'Auth mode is API Key, but no API Key was found in config.'); | ||
| } | ||
|
|
||
| final authorizedRequest = await authProvider.authorizeRequest( | ||
| _httpToAWSRequest(request), | ||
| options: ApiKeyAuthProviderOptions(apiKey)); | ||
| return authorizedRequest.httpRequest; | ||
| case APIAuthorizationType.iam: | ||
| final authProvider = _validateAuthProvider( | ||
| authProviderRepo | ||
| .getAuthProvider(APIAuthorizationType.iam.authProviderToken), | ||
| authType); | ||
| final service = endpointConfig.endpointType == EndpointType.graphQL | ||
| ? AWSService.appSync | ||
| : AWSService.apiGatewayManagementApi; // resolves to "execute-api" | ||
|
|
||
| final authorizedRequest = await authProvider.authorizeRequest( | ||
| _httpToAWSRequest(request), | ||
| options: IamAuthProviderOptions( | ||
| region: endpointConfig.region, | ||
| service: service, | ||
| ), | ||
| ); | ||
| return authorizedRequest.httpRequest; | ||
| case APIAuthorizationType.function: | ||
| case APIAuthorizationType.oidc: | ||
| case APIAuthorizationType.userPools: | ||
| throw UnimplementedError('${authType.name} not implemented.'); | ||
| case APIAuthorizationType.none: | ||
| return request; | ||
| } | ||
| } | ||
|
|
||
| T _validateAuthProvider<T extends AmplifyAuthProvider>( | ||
| T? authProvider, APIAuthorizationType authType) { | ||
| if (authProvider == null) { | ||
| throw ApiException('No auth provider found for auth mode ${authType.name}.', | ||
| recoverySuggestion: 'Ensure auth plugin correctly configured.'); | ||
| } | ||
| return authProvider; | ||
| } | ||
|
|
||
| AWSBaseHttpRequest _httpToAWSRequest(http.BaseRequest request) { | ||
| final method = AWSHttpMethod.fromString(request.method); | ||
| if (request is http.Request) { | ||
| return AWSHttpRequest( | ||
| method: method, | ||
| uri: request.url, | ||
| headers: { | ||
| AWSHeaders.contentType: 'application/x-amz-json-1.1', | ||
| ...request.headers, | ||
| }, | ||
| body: request.bodyBytes, | ||
| ); | ||
| } else if (request is http.StreamedRequest) { | ||
| return AWSStreamedHttpRequest( | ||
| method: method, | ||
| uri: request.url, | ||
| headers: { | ||
| AWSHeaders.contentType: 'application/x-amz-json-1.1', | ||
| ...request.headers, | ||
| }, | ||
| body: request.finalize(), | ||
| ); | ||
| } else { | ||
| throw UnimplementedError( | ||
| 'Multipart HTTP requests are not supported.', | ||
| ); | ||
| } | ||
| } |
Oops, something went wrong.