/*******************************************************************************
 * Copyright 2020 Graz University of Technology
 * MOA ZS has been developed in a cooperation between EGIZ
 * and Graz University of Technology.
 * Licensed under the EUPL, Version 1.2 or - as soon they will be approved by
 * the European Commission - subsequent versions of the EUPL (the "Licence");
 * You may not use this work except in compliance with the Licence.
 * You may obtain a copy of the Licence at:
 * https://joinup.ec.europa.eu/news/understanding-eupl-v12
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the Licence is distributed on an "AS IS" basis,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the Licence for the specific language governing permissions and
 * limitations under the Licence.
 *
 * This product combines work with different licenses. See the "NOTICE" text
 * file for details on the various modules and licenses.
 * The "NOTICE" text file is part of the distribution. Any derivative works
 * that you distribute must include a readable copy of the "NOTICE" text file.
 *******************************************************************************/
package at.gv.egiz.moazs.util;

import at.gv.egiz.moazs.repository.DeliveryRepository;
import at.gv.egiz.moazs.scheme.SOAPUtils;
import org.apache.cxf.message.Message;
import org.apache.cxf.phase.AbstractPhaseInterceptor;
import org.apache.cxf.phase.Phase;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.w3c.dom.Element;
import org.xml.sax.SAXException;

import javax.xml.parsers.ParserConfigurationException;
import javax.xml.transform.TransformerException;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.Map;
import java.util.function.Function;

import static at.gv.egiz.moazs.MoaZSException.moaZSException;

/**
 * @author Christof Rabensteiner
 *
 */
@Component
public class StoreSOAPBodyBinaryInRepositoryInterceptor extends AbstractPhaseInterceptor<Message> {

    private static final Logger log = LoggerFactory.getLogger(StoreSOAPBodyBinaryInRepositoryInterceptor.class);

    private final CXFMessageUtils messageUtils;
    private final SOAPUtils soapUtils;
    private final Map<String, Function<String, String>> idGenerators;
    private final DeliveryRepository repository;

    @Autowired
    public StoreSOAPBodyBinaryInRepositoryInterceptor(CXFMessageUtils extractor, SOAPUtils soapUtils,
                                                      Map<String, Function<String, String>> idGenerators,
                                                      DeliveryRepository repository) {
        super(Phase.RECEIVE);
        this.messageUtils = extractor;
        this.soapUtils = soapUtils;
        this.idGenerators = idGenerators;
        this.repository = repository;
    }

    public void handleMessage(Message message) {

        try {
            byte[] content = messageUtils.copyContent(message);

            if (content.length <= 0) {
                return;
            }

            Element document = soapUtils.toDOM(content);
            var rootTagNode = soapUtils.getChildElementOfSoapBody(document);
            var rootTagLocalName = rootTagNode.getLocalName();

            if (!idGenerators.containsKey(rootTagLocalName)) {
                log.trace("Child element {} of <soap:Body> is unknown. Will not store message.", rootTagLocalName);
                return;
            }

            String appDeliveryID = soapUtils.getAppDeliveryIDFrom(document);
            var id = idGenerators.get(rootTagLocalName).apply(appDeliveryID);
            byte[] response = soapUtils.toBytes(rootTagNode);
            repository.store(id, response);

            if(log.isTraceEnabled()) {
                log.trace("Msg Response with AppDeliveryID={} unwrapped and stored: {}. ",
                        appDeliveryID, new String(response, StandardCharsets.UTF_8));
            }

        } catch (ParserConfigurationException | SAXException | IOException | NullPointerException | TransformerException e) {
            throw moaZSException("Could not extract signed data from message.", e);
        }
    }

}