/*
 * 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.
 * A copy of the License is located at
 *
 *  http://aws.amazon.com/apache2.0
 *
 * or in the "license" file accompanying this file. This file 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.
 */

package com.amplifyframework.auth.cognito.actions

import com.amplifyframework.auth.cognito.AuthEnvironment
import com.amplifyframework.statemachine.Action
import com.amplifyframework.statemachine.codegen.actions.AuthActions
import com.amplifyframework.statemachine.codegen.data.AmplifyCredential
import com.amplifyframework.statemachine.codegen.data.CredentialType
import com.amplifyframework.statemachine.codegen.events.AuthEvent
import com.amplifyframework.statemachine.codegen.events.AuthenticationEvent
import com.amplifyframework.statemachine.codegen.events.AuthorizationEvent

internal object AuthCognitoActions : AuthActions {
    override fun initializeAuthConfigurationAction(event: AuthEvent.EventType.ConfigureAuth) =
        Action<AuthEnvironment>("InitAuthConfig") { id, dispatcher ->
            logger.verbose("$id Starting execution")
            val storedCredentials = credentialStoreClient.loadCredentials(CredentialType.Amplify)
            val evt = configuration.userPool?.run {
                AuthEvent(AuthEvent.EventType.ConfigureAuthentication(configuration, storedCredentials))
            } ?: AuthEvent(AuthEvent.EventType.ConfigureAuthorization(configuration, storedCredentials))
            logger.verbose("$id Sending event ${evt.type}")
            dispatcher.send(evt)
        }

    override fun initializeAuthenticationConfigurationAction(event: AuthEvent.EventType.ConfigureAuthentication) =
        Action<AuthEnvironment>("InitAuthNConfig") { id, dispatcher ->
            logger.verbose("$id Starting execution")
            val evt = AuthenticationEvent(
                AuthenticationEvent.EventType.Configure(event.configuration, event.storedCredentials)
            )
            logger.verbose("$id Sending event ${evt.type}")
            dispatcher.send(evt)
        }

    override fun initializeAuthorizationConfigurationAction(event: AuthEvent.EventType) =
        Action<AuthEnvironment>("InitAuthZConfig") { id, dispatcher ->
            logger.verbose("$id Starting execution")
            val handleEvent = { credentials: AmplifyCredential ->
                when (credentials) {
                    AmplifyCredential.Empty -> AuthorizationEvent(AuthorizationEvent.EventType.Configure)
                    else -> AuthorizationEvent(AuthorizationEvent.EventType.CachedCredentialsAvailable(credentials))
                }
            }
            val evt = when (event) {
                is AuthEvent.EventType.ConfiguredAuthentication -> handleEvent(event.storedCredentials)
                is AuthEvent.EventType.ConfigureAuthorization -> handleEvent(event.storedCredentials)
                else -> AuthorizationEvent(AuthorizationEvent.EventType.Configure)
            }
            logger.verbose("$id Sending event ${evt.type}")
            dispatcher.send(evt)
        }
}