package software.amazon.iotwireless.fuotatask;

import software.amazon.awssdk.core.exception.SdkServiceException;
import software.amazon.awssdk.services.iotwireless.IotWirelessClient;
import software.amazon.awssdk.services.iotwireless.model.AccessDeniedException;
import software.amazon.awssdk.services.iotwireless.model.ConflictException;
import software.amazon.awssdk.services.iotwireless.model.DeleteFuotaTaskRequest;
import software.amazon.awssdk.services.iotwireless.model.GetFuotaTaskRequest;
import software.amazon.awssdk.services.iotwireless.model.InternalServerException;
import software.amazon.awssdk.services.iotwireless.model.IotWirelessRequest;
import software.amazon.awssdk.services.iotwireless.model.ThrottlingException;
import software.amazon.awssdk.services.iotwireless.model.UpdateFuotaTaskRequest;
import software.amazon.awssdk.services.iotwireless.model.ValidationException;
import software.amazon.cloudformation.exceptions.CfnAccessDeniedException;
import software.amazon.cloudformation.exceptions.CfnGeneralServiceException;
import software.amazon.cloudformation.exceptions.CfnInternalFailureException;
import software.amazon.cloudformation.exceptions.CfnInvalidRequestException;
import software.amazon.cloudformation.exceptions.CfnNotFoundException;
import software.amazon.cloudformation.exceptions.CfnThrottlingException;
import software.amazon.cloudformation.exceptions.ResourceAlreadyExistsException;
import software.amazon.cloudformation.exceptions.ResourceNotFoundException;
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;

import java.util.function.Function;

public abstract class BaseHandlerStd extends BaseHandler<CallbackContext> {
    @Override
    public final ProgressEvent<ResourceModel, CallbackContext> handleRequest(
            final AmazonWebServicesClientProxy proxy,
            final ResourceHandlerRequest<ResourceModel> request,
            final CallbackContext callbackContext,
            final Logger logger) {
        return handleRequest(
                proxy,
                request,
                callbackContext != null ? callbackContext : new CallbackContext(),
                proxy.newProxy(ClientBuilder::getClient),
                logger
        );
    }

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

    private String extractResourceIdFromFuotaRequests(final IotWirelessRequest request) {
        if (request instanceof DeleteFuotaTaskRequest) {
            final DeleteFuotaTaskRequest deleteFuotaTaskRequest = (DeleteFuotaTaskRequest) request;
            return deleteFuotaTaskRequest.id();
        }
        if (request instanceof GetFuotaTaskRequest) {
            final GetFuotaTaskRequest getFuotaTaskRequest = (GetFuotaTaskRequest) request;
            return getFuotaTaskRequest.id();
        }
        if (request instanceof UpdateFuotaTaskRequest) {
            final UpdateFuotaTaskRequest updateFuotaTaskRequest = (UpdateFuotaTaskRequest) request;
            return updateFuotaTaskRequest.id();
        } else {
            return "";
        }
    }

    public RuntimeException handleException(final Exception error, final IotWirelessRequest request, final String typeName, Function<IotWirelessRequest, String> extractResourceIdFromRequests) {
        if (error instanceof ResourceNotFoundException) {
            return new CfnNotFoundException(typeName, extractResourceIdFromRequests.apply(request));
        }
        if (error instanceof ConflictException) {
            return new ResourceAlreadyExistsException(error);
        }
        if (error instanceof InternalServerException) {
            return new CfnInternalFailureException(error);
        }
        if (error instanceof ThrottlingException) {
            return new CfnThrottlingException(typeName, error);
        }
        if (error instanceof ValidationException) {
            return new CfnInvalidRequestException(typeName, error);
        }
        if (error instanceof AccessDeniedException) {
            return new CfnAccessDeniedException(typeName, error);
        }
        // Handle other kinds of error with status code
        if (error instanceof SdkServiceException) {
            switch (((SdkServiceException) error).statusCode()) {
                case 400:
                    return new CfnInvalidRequestException(typeName, error);
                case 403:
                    return new CfnAccessDeniedException(typeName, error);
                case 404:
                    return new CfnNotFoundException(typeName, extractResourceIdFromRequests.apply(request));
                case 409:
                    return new ResourceAlreadyExistsException(error);
                case 429:
                    return new CfnThrottlingException(typeName, error);
                case 500:
                    return new CfnInternalFailureException(error);
            }
        }
        return new CfnGeneralServiceException(typeName, error);
    }

    public RuntimeException handleException(final Exception error, final IotWirelessRequest request) {
        return handleException(error, request, ResourceModel.TYPE_NAME, this::extractResourceIdFromFuotaRequests);
    }
}