fix(Auth): Adding credentialsProvider to Cognito clients

aws-amplify · Jun 10, 2024 · 4c9cec5 · 4c9cec5
1 parent 53d0fff
commit 4c9cec5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/AmplifyPlugins/Auth/Sources/AWSCognitoAuthPlugin/AWSCognitoAuthPlugin+Configure.swift b/AmplifyPlugins/Auth/Sources/AWSCognitoAuthPlugin/AWSCognitoAuthPlugin+Configure.swift
@@ -7,7 +7,6 @@
 
 import Foundation
 @_spi(InternalAmplifyConfiguration) import Amplify
-
 import AWSCognitoIdentity
 import AWSCognitoIdentityProvider
 import AWSPluginsCore
@@ -112,6 +111,8 @@ extension AWSCognitoAuthPlugin {
                 configuration.retryStrategyOptions = RetryStrategyOptions(maxRetriesBase: Int(maxRetryUnwrapped))
             }
 
+            configuration.credentialsProvider = AmplifyAWSCredentialsProvider()
+
             return CognitoIdentityProviderClient(config: configuration)
         default:
             fatalError()
@@ -134,6 +135,8 @@ extension AWSCognitoAuthPlugin {
                 configuration.retryStrategyOptions = RetryStrategyOptions(maxRetriesBase: Int(maxRetryUnwrapped))
             }
 
+            configuration.credentialsProvider = AmplifyAWSCredentialsProvider()
+
             return CognitoIdentityClient(config: configuration)
         default:
             fatalError()

diff --git a/AmplifyPlugins/Core/AmplifyCredentials/AmplifyAWSCredentialsProvider.swift b/AmplifyPlugins/Core/AmplifyCredentials/AmplifyAWSCredentialsProvider.swift
@@ -13,6 +13,8 @@ import Foundation
 
 public class AmplifyAWSCredentialsProvider: AWSClientRuntime.CredentialsProviding {
 
+    public init() { }
+
     public func getCredentials() async throws -> AWSClientRuntime.AWSCredentials {
         let authSession = try await Amplify.Auth.fetchAuthSession()
         if let awsCredentialsProvider = authSession as? AuthAWSCredentialsProvider {