package at.gv.egiz.moazs.backend;


import at.gv.egiz.moazs.MoaZSException;
import at.gv.egiz.moazs.client.ClientFactory;
import at.gv.egiz.moazs.client.TnvzHelper;
import at.gv.egiz.moazs.repository.DeliveryRepository;
import at.gv.egiz.moazs.scheme.Mzs2MsgConverter;
import at.gv.egiz.moazs.scheme.RequestStatusResponse;
import at.gv.zustellung.app2mzs.xsd.DeliveryRequestType;
import at.gv.zustellung.msg.xsd.App2ZusePort;
import at.gv.zustellung.msg.xsd.DeliveryRequestStatusType;
import at.gv.zustellung.tnvz.xsd.TNVZServicePort;
import org.apache.log4j.Logger;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Profile;
import org.springframework.stereotype.Component;

import java.util.function.Consumer;

import static at.gv.egiz.moazs.scheme.RequestStatusResponse.*;
import static at.gv.egiz.moazs.scheme.RequestStatusResponse.generateError;
import static at.gv.zustellung.msg.xsd.DeliveryRequestStatusType.Error.errorBuilder;
import static at.gv.zustellung.msg.xsd.DeliveryRequestStatusType.deliveryRequestStatusTypeBuilder;
import static java.lang.String.format;

@Component
@Profile("!cluster")
public class DeliveryRequestBackend implements Consumer<String> {

    private static final Logger log = Logger.getLogger(DeliveryRequestBackend.class);
    private static final String MZS_PIPELINE_ERROR_MSG =
            "An error occured while processing the DeliveryRequest with AppDeliveryID=%s. ";
    private static final String DELIVERY_REQUEST_MISSING_ERROR_MSG =
            "DeliveryRequest for AppDeliveryID=%s is not in repository. " ;
    private static final String BINARY_RESPONSE_MISSING_ERROR_MSG =
            "Binary DeliveryRequestStatus for AppDeliveryID=%s is not in repository. " ;

    private final DeliveryRepository repository;
    private final TnvzHelper tnvzHelper;
    private final Mzs2MsgConverter converter;
    private final ClientFactory clientFactory;
    private final Consumer<byte[]> signatureVerifier;

    @Autowired
    public DeliveryRequestBackend(DeliveryRepository repository,
                                  TnvzHelper tnvzHelper,
                                  Mzs2MsgConverter converter,
                                  ClientFactory clientFactory,
                                  Consumer<byte[]> signatureVerifier) {
        this.repository = repository;
        this.tnvzHelper = tnvzHelper;
        this.converter = converter;
        this.clientFactory = clientFactory;
        this.signatureVerifier = signatureVerifier;
    }

    /**
     * Performs all {@code DeliveryRequest}'s Back-End Tasks.
     *
     * Fetches {@code DeliveryRequest} referred by appDeliveryID from
     * {@code DeliveryRepository}, makes sure that all necessary
     * tasks (query tnvz, query msg, accept status) are executed and
     * stores the response back to {@code DeliveryRepository}.
     * @param appDeliveryID
     */
    @Override
    public void accept(String appDeliveryID) {

        DeliveryRequestStatusType status = null;
        var fallbackAnswerBuilder = errorBuilder().withAppDeliveryID(appDeliveryID);

        try {
            var mzsRequest = repository.retrieveDeliveryRequest(appDeliveryID).orElseThrow(
                    () -> MoaZSException.moaZSException(format(DELIVERY_REQUEST_MISSING_ERROR_MSG, appDeliveryID)));
            fallbackAnswerBuilder.withDeliverySystem(mzsRequest.getConfig().getMSGClient().getURL());

            var msgRequest = buildMsgRequest(mzsRequest);
            var msgClientParams = mzsRequest.getConfig().getMSGClient();
            App2ZusePort client = clientFactory.create(msgClientParams, App2ZusePort.class);
            status = client.delivery(msgRequest);

            var response = new RequestStatusResponse(status);

            verifySignedStatus(response.getResponseID(), appDeliveryID);
            repository.store(response);

        } catch (MoaZSException exception) {
            log.error(format(MZS_PIPELINE_ERROR_MSG, appDeliveryID), exception);

            var templateAnswer = (status == null)
                    ? fallbackAnswerBuilder.build()
                    : getAnswer(status);

            var errorStatus = generateError(exception, templateAnswer);
            repository.store(errorStatus);
        }
    }

    private void verifySignedStatus(String responseID, String appDeliveryID) throws MoaZSException {
        try {
            var signedStatus = repository.retrieveBinaryResponse(responseID).orElseThrow(
                    () -> MoaZSException.moaZSException(format(BINARY_RESPONSE_MISSING_ERROR_MSG, responseID)));
            signatureVerifier.accept(signedStatus);
        } catch (MoaZSException ex) {
            var message = format(MsgResponseBackend.MOASP_SIGNATURE_INVALID_ERROR_MSG, appDeliveryID);
            var code = MoaZSException.ERROR_MOASP_SIGNATURE_INVALID;
            throw MoaZSException.moaZSException(message, code, ex);
        }
    }

    private at.gv.zustellung.msg.xsd.DeliveryRequestType buildMsgRequest(DeliveryRequestType mzsRequest) throws MoaZSException {
        if (mzsRequest.getConfig().isPerformQueryPersonRequest()) {
            var tnvzClientParams = mzsRequest.getConfig().getTNVZClient();
            TNVZServicePort tvnzPort = clientFactory.create(tnvzClientParams, TNVZServicePort.class);
            var identification = tnvzHelper.performQueryPersonRequest(mzsRequest, tvnzPort);
            return converter.convert(mzsRequest, identification);
        } else {
            return converter.convert(mzsRequest);
        }
    }

}