package at.gv.egiz.moazs.pipeline;


import at.gv.egiz.moazs.MoaZSException;
import at.gv.egiz.moazs.msg.MsgClientFactory;
import at.gv.egiz.moazs.msg.SignatureVerifier;
import at.gv.egiz.moazs.msg.StoreSOAPBodyBinaryInRepositoryInterceptor;
import at.gv.egiz.moazs.repository.DeliveryRepository;
import at.gv.egiz.moazs.scheme.Mzs2MsgConverter;
import at.gv.egiz.moazs.scheme.NameSpace;
import at.gv.egiz.moazs.tnvz.TnvzClient;
import at.gv.egiz.moazs.tnvz.TnvzResultVerifier;
import at.gv.zustellung.app2mzs.xsd.DeliveryRequestType;
import at.gv.zustellung.msg.xsd.DeliveryAnswerType;
import at.gv.zustellung.msg.xsd.DeliveryRequestStatusType;
import at.gv.zustellung.tnvz.xsd.PersonResultType;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Profile;
import org.springframework.stereotype.Component;

import static at.gv.egiz.moazs.MoaZSException.moaZSExceptionBuilder;
import static at.gv.egiz.moazs.util.NullCoalesce.coalesce;
import static at.gv.zustellung.msg.xsd.DeliveryRequestStatusType.Error.errorBuilder;
import static at.gv.zustellung.msg.xsd.DeliveryRequestStatusType.deliveryRequestStatusTypeBuilder;
import static at.gv.zustellung.msg.xsd.ErrorInfoType.errorInfoTypeBuilder;

@Component
@Profile("!cluster")
public class SameThreadDeliveryPipeline implements DeliveryPipeline {

    private final DeliveryRepository repository;
    private final TnvzClient tnvzClient;
    private final TnvzResultVerifier tnvzVerifier;
    private final Mzs2MsgConverter converter;
    private final MsgClientFactory msgClientFactory;
    private final SignatureVerifier verifier;
    private final StoreSOAPBodyBinaryInRepositoryInterceptor interceptor;

    @Autowired
    public SameThreadDeliveryPipeline(DeliveryRepository repository,
                                      TnvzClient tnvzClient,
                                      TnvzResultVerifier tnvzVerifier,
                                      Mzs2MsgConverter converter,
                                      StoreSOAPBodyBinaryInRepositoryInterceptor interceptor,
                                      MsgClientFactory msgClientFactory,
                                      SignatureVerifier verifier
    ) {
        this.repository = repository;
        this.tnvzClient = tnvzClient;
        this.tnvzVerifier = tnvzVerifier;
        this.converter = converter;
        this.msgClientFactory = msgClientFactory;
        this.verifier = verifier;
        this.interceptor = interceptor;
    }

    @Override
    public void processRequest(String appDeliveryId) {

        DeliveryRequestType mzsRequest;
        PersonResultType tnvzResult = null;
        at.gv.zustellung.msg.xsd.DeliveryRequestType msgRequest;
        DeliveryRequestStatusType status;

        try {
            mzsRequest = repository.getDeliveryRequest(appDeliveryId).orElseThrow();

            if (mzsRequest.getConfig().isPerformQueryPersonRequest()) {
                tnvzResult = performTnvzQuery(mzsRequest);
                msgRequest = converter.convert(mzsRequest, tnvzResult.getSuccess().getIdentification());
            } else {
                msgRequest = converter.convert(mzsRequest);
            }

            status = msgClientFactory.create(msgRequest, mzsRequest.getConfig(), interceptor).send();

            var signedStatus = repository.getSignedDeliveryRequestStatus(appDeliveryId).get();
            if (verifier.verify(signedStatus)) {
                repository.add(status);
            } else {
                throw moaZSExceptionBuilder("Signature of DeliveryRequestStatus with AppDeliveryId={} " +
                        "is invalid.", appDeliveryId)
                        .withErrorCode(MoaZSException.ERROR_MOASP_SIGNATURE_INVALID)
                        .withMzsRequest(mzsRequest)
                        .withTnvzResult(tnvzResult)
                        .withMsgRequest(msgRequest)
                        .withMsgResult(status)
                        .build();
            }
        } catch (MoaZSException exception) {
            var errorStatus = generateErrorStatus(exception, appDeliveryId);
            repository.add(errorStatus);
        }
    }

    private PersonResultType performTnvzQuery(DeliveryRequestType request) {
        var result =  tnvzClient.query(request.getSender(), request.getReceiver());
        tnvzVerifier.verify(request, result);
        return result;
    }

    private DeliveryRequestStatusType generateErrorStatus(MoaZSException exception, String appDeliveryId) {

        var infoBuilder = errorInfoTypeBuilder()
                .withText(exception.getMessage())
                .withCode(exception.getErrorCode());

        var errorBuilder = errorBuilder()
                .withErrorInfo(infoBuilder.build())
                .withAppDeliveryID(appDeliveryId);

        if (exception.getMzsRequest() != null) {
            errorBuilder.withDeliverySystem(exception.getMzsRequest().getConfig().getServer().getZUSEUrlID());
        }

        if (exception.getTnvzResult() != null && exception.getTnvzResult().getError() != null) {
            errorBuilder.withPreAdviceNoteSent(exception.getTnvzResult().getError().getPreAdviceNoteSent());
        }

        if (exception.getMsgResult() != null) {
            var answer = getAnswerFromResult(exception.getMsgResult());
            errorBuilder.withGZ(answer.getGZ());
            errorBuilder.withZSDeliveryID(answer.getZSDeliveryID());
        }

        return deliveryRequestStatusTypeBuilder()
                .withError(errorBuilder.build())
                .withVersion(NameSpace.MSG_VERSION)
                .build();

    }

    private DeliveryAnswerType getAnswerFromResult(DeliveryRequestStatusType msgResult) {
        return coalesce(msgResult.getSuccess(), msgResult.getPartialSuccess(), msgResult.getError()).get();
    }


}