77
88import Foundation
99@_spi ( InternalAmplifyConfiguration) import Amplify
10- import AWSClientRuntime
1110import AWSCognitoIdentity
1211import AWSCognitoIdentityProvider
1312import AWSPluginsCore
1413import ClientRuntime
14+ import AWSClientRuntime
1515@_spi ( PluginHTTPClientEngine) import InternalAmplifyCredentials
1616@_spi ( InternalHttpEngineProxy) import AWSPluginsCore
17- import SmithyRetries
1817import SmithyRetriesAPI
18+ import SmithyRetries
1919
2020extension AWSCognitoAuthPlugin {
2121
@@ -35,16 +35,13 @@ extension AWSCognitoAuthPlugin {
3535 } else {
3636 throw PluginError . pluginConfigurationError (
3737 AuthPluginErrorConstants . decodeConfigurationError. errorDescription,
38- AuthPluginErrorConstants . decodeConfigurationError. recoverySuggestion
39- )
38+ AuthPluginErrorConstants . decodeConfigurationError. recoverySuggestion)
4039 }
4140
4241 let credentialStoreResolver = CredentialStoreState . Resolver ( ) . eraseToAnyResolver ( )
4342 let credentialEnvironment = credentialStoreEnvironment ( authConfiguration: authConfiguration)
44- let credentialStoreMachine = StateMachine (
45- resolver: credentialStoreResolver,
46- environment: credentialEnvironment
47- )
43+ let credentialStoreMachine = StateMachine ( resolver: credentialStoreResolver,
44+ environment: credentialEnvironment)
4845 let credentialsClient = CredentialStoreOperationClient (
4946 credentialStoreStateMachine: credentialStoreMachine)
5047
@@ -59,40 +56,35 @@ extension AWSCognitoAuthPlugin {
5956 let hubEventHandler = AuthHubEventHandler ( )
6057 let analyticsHandler = try UserPoolAnalytics (
6158 authConfiguration. getUserPoolConfiguration ( ) ,
62- credentialStoreEnvironment: credentialEnvironment. credentialStoreEnvironment
63- )
64-
65- configure (
66- authConfiguration: authConfiguration,
67- authEnvironment: authEnvironment,
68- authStateMachine: authStateMachine,
69- credentialStoreStateMachine: credentialStoreMachine,
70- hubEventHandler: hubEventHandler,
71- analyticsHandler: analyticsHandler
72- )
59+ credentialStoreEnvironment: credentialEnvironment. credentialStoreEnvironment)
60+
61+ configure ( authConfiguration: authConfiguration,
62+ authEnvironment: authEnvironment,
63+ authStateMachine: authStateMachine,
64+ credentialStoreStateMachine: credentialStoreMachine,
65+ hubEventHandler: hubEventHandler,
66+ analyticsHandler: analyticsHandler)
7367 }
7468
75- func configure(
76- authConfiguration: AuthConfiguration ,
77- authEnvironment: AuthEnvironment ,
78- authStateMachine: AuthStateMachine ,
79- credentialStoreStateMachine: CredentialStoreStateMachine ,
80- hubEventHandler: AuthHubEventBehavior ,
81- analyticsHandler: UserPoolAnalyticsBehavior ,
82- queue: OperationQueue = OperationQueue ( )
83- ) {
69+ func configure( authConfiguration: AuthConfiguration ,
70+ authEnvironment: AuthEnvironment ,
71+ authStateMachine: AuthStateMachine ,
72+ credentialStoreStateMachine: CredentialStoreStateMachine ,
73+ hubEventHandler: AuthHubEventBehavior ,
74+ analyticsHandler: UserPoolAnalyticsBehavior ,
75+ queue: OperationQueue = OperationQueue ( ) ) {
8476
8577 self . authConfiguration = authConfiguration
8678 self . queue = queue
8779 self . queue. maxConcurrentOperationCount = 1
8880 self . authEnvironment = authEnvironment
8981 self . authStateMachine = authStateMachine
9082 self . credentialStoreStateMachine = credentialStoreStateMachine
91- internalConfigure ( )
92- listenToStateMachineChanges ( )
83+ self . internalConfigure ( )
84+ self . listenToStateMachineChanges ( )
9385 self . hubEventHandler = hubEventHandler
9486 self . analyticsHandler = analyticsHandler
95- taskQueue = TaskQueue ( )
87+ self . taskQueue = TaskQueue ( )
9688 }
9789
9890 // MARK: - Configure Helpers
@@ -105,7 +97,7 @@ extension AWSCognitoAuthPlugin {
10597 endpointResolver: userPoolConfig. endpoint? . resolver
10698 )
10799
108- if var httpClientEngineProxy {
100+ if var httpClientEngineProxy = httpClientEngineProxy {
109101 httpClientEngineProxy. target = baseClientEngine ( for: configuration)
110102 configuration. httpClientEngine = UserAgentSettingClientEngine (
111103 target: httpClientEngineProxy
@@ -222,8 +214,7 @@ extension AWSCognitoAuthPlugin {
222214 authenticationEnvironment: authenticationEnvironment,
223215 authorizationEnvironment: nil ,
224216 credentialsClient: credentialsClient,
225- logger: log
226- )
217+ logger: log)
227218
228219 case . identityPools( let identityPoolConfigurationData) :
229220 let authorizationEnvironment = authorizationEnvironment (
@@ -235,13 +226,10 @@ extension AWSCognitoAuthPlugin {
235226 authenticationEnvironment: nil ,
236227 authorizationEnvironment: authorizationEnvironment,
237228 credentialsClient: credentialsClient,
238- logger: log
239- )
229+ logger: log)
240230
241- case . userPoolsAndIdentityPools(
242- let userPoolConfigurationData,
243- let identityPoolConfigurationData
244- ) :
231+ case . userPoolsAndIdentityPools( let userPoolConfigurationData,
232+ let identityPoolConfigurationData) :
245233 let authenticationEnvironment = authenticationEnvironment (
246234 userPoolConfigData: userPoolConfigurationData)
247235 let authorizationEnvironment = authorizationEnvironment (
@@ -253,49 +241,39 @@ extension AWSCognitoAuthPlugin {
253241 authenticationEnvironment: authenticationEnvironment,
254242 authorizationEnvironment: authorizationEnvironment,
255243 credentialsClient: credentialsClient,
256- logger: log
257- )
244+ logger: log)
258245 }
259246 }
260247
261248 private func authenticationEnvironment( userPoolConfigData: UserPoolConfigurationData ) -> AuthenticationEnvironment {
262249
263- let srpAuthEnvironment = BasicSRPAuthEnvironment (
264- userPoolConfiguration: userPoolConfigData,
265- cognitoUserPoolFactory: makeUserPool
266- )
250+ let srpAuthEnvironment = BasicSRPAuthEnvironment ( userPoolConfiguration: userPoolConfigData,
251+ cognitoUserPoolFactory: makeUserPool)
267252 let srpSignInEnvironment = BasicSRPSignInEnvironment ( srpAuthEnvironment: srpAuthEnvironment)
268253 let userPoolEnvironment = BasicUserPoolEnvironment (
269254 userPoolConfiguration: userPoolConfigData,
270255 cognitoUserPoolFactory: makeUserPool,
271256 cognitoUserPoolASFFactory: makeCognitoASF,
272- cognitoUserPoolAnalyticsHandlerFactory: makeUserPoolAnalytics
273- )
257+ cognitoUserPoolAnalyticsHandlerFactory: makeUserPoolAnalytics)
274258 let hostedUIEnvironment = hostedUIEnvironment ( userPoolConfigData)
275- return BasicAuthenticationEnvironment (
276- srpSignInEnvironment: srpSignInEnvironment,
277- userPoolEnvironment: userPoolEnvironment,
278- hostedUIEnvironment: hostedUIEnvironment
279- )
259+ return BasicAuthenticationEnvironment ( srpSignInEnvironment: srpSignInEnvironment,
260+ userPoolEnvironment: userPoolEnvironment,
261+ hostedUIEnvironment: hostedUIEnvironment)
280262 }
281263
282264 private func hostedUIEnvironment( _ configuration: UserPoolConfigurationData ) -> HostedUIEnvironment ? {
283265 guard let hostedUIConfig = configuration. hostedUIConfig else {
284266 return nil
285267 }
286- return BasicHostedUIEnvironment (
287- configuration: hostedUIConfig,
288- hostedUISessionFactory: makeHostedUISession,
289- urlSessionFactory: makeURLSession,
290- randomStringFactory: makeRandomString
291- )
268+ return BasicHostedUIEnvironment ( configuration: hostedUIConfig,
269+ hostedUISessionFactory: makeHostedUISession,
270+ urlSessionFactory: makeURLSession,
271+ randomStringFactory: makeRandomString)
292272 }
293273
294274 private func authorizationEnvironment( identityPoolConfigData: IdentityPoolConfigurationData ) -> AuthorizationEnvironment {
295- BasicAuthorizationEnvironment (
296- identityPoolConfiguration: identityPoolConfigData,
297- cognitoIdentityFactory: makeIdentityClient
298- )
275+ BasicAuthorizationEnvironment ( identityPoolConfiguration: identityPoolConfigData,
276+ cognitoIdentityFactory: makeIdentityClient)
299277 }
300278
301279 private func credentialStoreEnvironment( authConfiguration: AuthConfiguration ) -> CredentialEnvironment {
@@ -313,9 +291,8 @@ extension AWSCognitoAuthPlugin {
313291 let operation = AuthConfigureOperation (
314292 request: request,
315293 authStateMachine: authStateMachine,
316- credentialStoreStateMachine: credentialStoreStateMachine
317- )
318- queue. addOperation ( operation)
294+ credentialStoreStateMachine: credentialStoreStateMachine)
295+ self . queue. addOperation ( operation)
319296 }
320297}
321298
0 commit comments