network.oxalis.vefa.peppol.sbdh.SbdWriter Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2015-2017 Direktoratet for forvaltning og IKT
 *
 * This source code is subject to dual licensing:
 *
 *
 * Licensed under the EUPL, Version 1.1 or – as soon they
 * will be approved by the European Commission - subsequent
 * versions of the EUPL (the "Licence");
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 *
 *
 * See the Licence for the specific language governing
 * permissions and limitations under the Licence.
 */

package network.oxalis.vefa.peppol.sbdh;

import network.oxalis.vefa.peppol.common.model.Header;
import network.oxalis.vefa.peppol.common.util.ExceptionUtil;
import network.oxalis.vefa.peppol.sbdh.util.XMLBinaryOutputStream;
import network.oxalis.vefa.peppol.sbdh.util.XMLStreamWriterWrapper;
import network.oxalis.vefa.peppol.sbdh.util.XMLTextOutputStream;
import network.oxalis.vefa.peppol.sbdh.lang.SbdhException;

import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamWriter;
import java.io.Closeable;
import java.io.IOException;
import java.io.OutputStream;

public class SbdWriter implements Closeable {

    private XMLStreamWriter writer;

    public static SbdWriter newInstance(OutputStream outputStream, Header header) throws SbdhException {
        return new SbdWriter(outputStream, header);
    }

    private SbdWriter(final OutputStream outputStream, Header header) throws SbdhException {
        writer = ExceptionUtil.perform(SbdhException.class, "Unable to initiate SBD.", () ->
                SbdhHelper.XML_OUTPUT_FACTORY.createXMLStreamWriter(outputStream, "UTF-8"));

        initiateDocument(header);
    }

    private void initiateDocument(final Header header) throws SbdhException {
        ExceptionUtil.perform(SbdhException.class, () -> {
                    writer.writeStartDocument("UTF-8", "1.0");
                    writer.writeStartElement("", Ns.QNAME_SBD.getLocalPart(), Ns.SBDH);
                    writer.writeDefaultNamespace(Ns.SBDH);
                    SbdhWriter.write(writer, header);
                }
        );
    }

    public XMLStreamWriter xmlWriter() {
        return new XMLStreamWriterWrapper(writer);
    }

    public OutputStream binaryWriter(String mimeType) throws XMLStreamException {
        return binaryWriter(mimeType, null);
    }

    public OutputStream binaryWriter(String mimeType, String encoding) throws XMLStreamException {
        return new XMLBinaryOutputStream(xmlWriter(), mimeType, encoding);
    }

    public OutputStream textWriter(String mimeType) throws XMLStreamException {
        return new XMLTextOutputStream(xmlWriter(), mimeType);
    }

    private void finalizeDocument() throws SbdhException {
        ExceptionUtil.perform(SbdhException.class, () -> {
            writer.writeEndElement();
            writer.writeEndDocument();
        });
    }

    @Override
    public void close() throws IOException {
        ExceptionUtil.perform(IOException.class, () -> {
            finalizeDocument();
            writer.close();
        });
    }
}