network.oxalis.as4.util.As4MessageFactory Maven / Gradle / Ivy

Go to download
package network.oxalis.as4.util;

import com.google.inject.Inject;
import com.google.inject.Singleton;
import network.oxalis.as4.api.MessageIdGenerator;
import network.oxalis.as4.lang.AS4Error;
import network.oxalis.as4.lang.OxalisAs4Exception;
import network.oxalis.as4.inbound.ProsessingContext;
import org.apache.cxf.interceptor.Fault;
import org.oasis_open.docs.ebxml_bp.ebbp_signals_2.MessagePartNRInformation;
import org.oasis_open.docs.ebxml_bp.ebbp_signals_2.NonRepudiationInformation;
import org.oasis_open.docs.ebxml_msg.ebms.v3_0.ns.core._200704.Error;
import org.oasis_open.docs.ebxml_msg.ebms.v3_0.ns.core._200704.*;

import jakarta.xml.bind.JAXBContext;
import jakarta.xml.bind.JAXBElement;
import javax.xml.datatype.XMLGregorianCalendar;
import jakarta.xml.soap.*;
import java.util.Date;
import java.util.List;
import java.util.stream.Collectors;

@Singleton
public class As4MessageFactory {

    private final MessageIdGenerator messageIdGenerator;
    private final MessageFactory messageFactory;
    private final JAXBContext jaxbContext;

    @Inject
    public As4MessageFactory(MessageIdGenerator messageIdGenerator) throws SOAPException {
        this(
                messageIdGenerator,
                MessageFactory.newInstance(SOAPConstants.SOAP_1_2_PROTOCOL),
                Marshalling.getInstance()
        );
    }

    public As4MessageFactory(MessageIdGenerator messageIdGenerator, MessageFactory messageFactory, JAXBContext jaxbContext) {
        this.messageFactory = messageFactory;
        this.jaxbContext = jaxbContext;
        this.messageIdGenerator = messageIdGenerator;
    }

    public SOAPMessage createReceiptMessage(UserMessage inUserMessage, ProsessingContext prosessingContext) throws OxalisAs4Exception {

        XMLGregorianCalendar xmlGc = XMLUtil.dateToXMLGeorgianCalendar(
                prosessingContext.getReceiptTimestamp().getDate()
        );

        MessageInfo messageInfo = new MessageInfo();
        messageInfo.setTimestamp(xmlGc);
        messageInfo.setMessageId(messageIdGenerator.generate());
        messageInfo.setRefToMessageId(inUserMessage.getMessageInfo().getMessageId());

        List mpList = prosessingContext.getReferenceList().stream()
                .map(reference -> {
                    MessagePartNRInformation messagePartNRInformation = new MessagePartNRInformation();
                    messagePartNRInformation.setReference(reference);
                    return messagePartNRInformation;
                })
                .collect(Collectors.toList());

        NonRepudiationInformation nri = new NonRepudiationInformation();
        nri.getMessagePartNRInformation().addAll(mpList);

        Receipt receipt = new Receipt();
        receipt.getAny().add(nri);

        SignalMessage signalMessage = new SignalMessage();
        signalMessage.setMessageInfo(messageInfo);
        signalMessage.setReceipt(receipt);

        return marshalSignalMessage(signalMessage);
    }


    public SOAPMessage createErrorMessage(String messageId, AS4Error as4Error) {
        try {

            XMLGregorianCalendar currentDate = XMLUtil.dateToXMLGeorgianCalendar(new Date());


            MessageInfo messageInfo = new MessageInfo();
            messageInfo.setRefToMessageId(messageId);
            messageInfo.setTimestamp(currentDate);
            messageInfo.setMessageId(messageIdGenerator.generate());

            Error error = new Error();
            error.setRefToMessageInError(messageId);
            error.setErrorCode(as4Error.getErrorCode().toString());
            error.setErrorDetail(getErrorDetail(as4Error));
            error.setShortDescription(as4Error.getErrorCode().getShortDescription());
            error.setOrigin(as4Error.getErrorCode().getOrigin().toString());
            error.setCategory(as4Error.getErrorCode().getCatgory().toString());
            error.setSeverity(as4Error.getSeverity().toString());

            SignalMessage signalMessage = new SignalMessage();
            signalMessage.setMessageInfo(messageInfo);
            signalMessage.getError().add(error);

            return marshalSignalMessage(signalMessage);

        } catch (OxalisAs4Exception e) {
            throw new Fault(e.getCause());
        }
    }

    private String getErrorDetail(AS4Error as4Error) {
        StringBuilder sb = new StringBuilder();

        sb.append(as4Error.getMessage());

        Throwable throwable = as4Error.getException();

        while (throwable.getCause() != null) {
            throwable = throwable.getCause();
            sb.append("\ncause: ").append(throwable.getMessage());
        }

        return sb.toString();
    }

    public SOAPMessage marshalSignalMessage(SignalMessage signalMessage) throws OxalisAs4Exception {
        try {
            SOAPMessage message = messageFactory.createMessage();
            SOAPHeader soapHeader = message.getSOAPHeader();

            SOAPHeaderElement messagingHeader = soapHeader.addHeaderElement(Constants.MESSAGING_QNAME);
            messagingHeader.setMustUnderstand(true);

            JAXBElement userMessageJAXBElement = new JAXBElement<>(
                    Constants.SIGNAL_MESSAGE_QNAME,
                    SignalMessage.class,
                    signalMessage
            );

            jaxbContext.createMarshaller().marshal(userMessageJAXBElement, messagingHeader);

            return message;
        } catch (Exception e) {
            throw new OxalisAs4Exception("Unable to marshal SignalMessage", e, AS4ErrorCode.EBMS_0004);
        }
    }
}