package software.amazon.amplify.domain;

import software.amazon.amplify.common.utils.ClientWrapper;
import software.amazon.awssdk.services.amplify.AmplifyClient;
import software.amazon.awssdk.services.amplify.model.GetDomainAssociationResponse;
import software.amazon.cloudformation.proxy.AmazonWebServicesClientProxy;
import software.amazon.cloudformation.proxy.Logger;
import software.amazon.cloudformation.proxy.ProgressEvent;
import software.amazon.cloudformation.proxy.ProxyClient;
import software.amazon.cloudformation.proxy.ResourceHandlerRequest;

public class ReadHandler extends BaseHandlerStd {
    private Logger logger;

    protected ProgressEvent<ResourceModel, CallbackContext> handleRequest(
        final AmazonWebServicesClientProxy proxy,
        final ResourceHandlerRequest<ResourceModel> request,
        final CallbackContext callbackContext,
        final ProxyClient<AmplifyClient> proxyClient,
        final Logger logger) {

        this.logger = logger;
        final ResourceModel model = request.getDesiredResourceState();
        logger.log("INFO: requesting with model: " + model);

        return proxy.initiate("AWS-Amplify-Domain::Read", proxyClient, model, callbackContext)
            .translateToServiceRequest(Translator::translateToReadRequest)
            .makeServiceCall((getDomainAssociationRequest, proxyInvocation) -> (GetDomainAssociationResponse) ClientWrapper.execute(
                    proxy,
                    getDomainAssociationRequest,
                    proxyInvocation.client()::getDomainAssociation,
                    ResourceModel.TYPE_NAME,
                    model.getArn(),
                    logger
            ))
            .done(getDomainAssociationResponse -> {
                ResourceModel modelRet = Translator.translateFromReadResponse(getDomainAssociationResponse);
                logger.log("INFO: returning model: " + modelRet);
                return ProgressEvent.defaultSuccessHandler(modelRet);
            });
    }
}