no.digipost.api.client.internal.delivery.DocumentsPreparer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of digipost-api-client-java Show documentation
${project.artifactId}
There is a newer version: 16.4.1
Show newest version
/*
 * Copyright (C) Posten Norge AS
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package no.digipost.api.client.internal.delivery;

import no.digipost.api.client.DigipostClientConfig;
import no.digipost.api.client.errorhandling.DigipostClientException;
import no.digipost.api.client.errorhandling.ErrorCode;
import no.digipost.api.client.representations.Channel;
import no.digipost.api.client.representations.Document;
import no.digipost.api.client.representations.Message;
import no.digipost.api.client.security.Encrypter;
import no.digipost.print.validate.PdfValidationResult;
import no.digipost.print.validate.PdfValidationSettings;
import no.digipost.print.validate.PdfValidator;
import no.digipost.sanitizing.HtmlValidationResult;
import no.digipost.sanitizing.HtmlValidator;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;
import java.io.InputStream;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Optional;
import java.util.function.Supplier;

import static java.util.Optional.empty;
import static no.digipost.api.client.representations.Channel.PRINT;
import static no.digipost.api.client.representations.FileType.HTM;
import static no.digipost.api.client.representations.FileType.HTML;
import static no.digipost.api.client.representations.FileType.PDF;
import static no.digipost.print.validate.PdfValidationResult.EVERYTHING_OK;
import static no.digipost.sanitizing.HtmlValidationResult.HTML_EVERYTHING_OK;
import static org.apache.commons.io.IOUtils.toByteArray;

class DocumentsPreparer {

    private static final Logger LOG = LoggerFactory.getLogger(DocumentsPreparer.class);
    static final String SANITATION_DIFF_ADVICE = "Kjør DigipostValidatingHtmlSanitizer for å se sanitert html.";

    private final PdfValidator pdfValidator;
    private final HtmlValidator htmlValidator;

    DocumentsPreparer(PdfValidator pdfValidator, HtmlValidator htmlValidator) {
        this.pdfValidator = pdfValidator;
        this.htmlValidator = htmlValidator;
    }

    Map prepare(
            Map documentsAndContent, Message message,
            Encrypter encrypter, Supplier pdfValidationSettings, DigipostClientConfig config) throws IOException {

        final Map prepared = new LinkedHashMap<>();

        for (Document document : (Iterable) documentsAndContent.keySet().stream().sorted(message.documentOrder())::iterator) {
            if (document.willBeEncrypted()) {
                byte[] byteContent = toByteArray(documentsAndContent.get(document));
                LOG.debug("Validerer dokument med uuid '{}' før kryptering", document.uuid);
                validateAndSetNrOfPages(message.getChannel(), document, byteContent, pdfValidationSettings);
                validateHtml(document, byteContent, config);
                LOG.debug("Krypterer innhold for dokument med uuid '{}'", document.uuid);
                prepared.put(document, encrypter.encrypt(byteContent));
            } else {
                prepared.put(document, documentsAndContent.get(document));
            }
        }
        return prepared;
    }

    void validateHtml(Document document, byte[] content, DigipostClientConfig config) {
        HtmlValidationResult htmlValidation = HTML_EVERYTHING_OK;
        if (document.is(HTML) || document.is(HTM)) {
            htmlValidation = htmlValidator.valider(content);
        }

        if (!htmlValidation.okForWeb) {
            throw new DigipostClientException(ErrorCode.INVALID_HTML_CONTENT, htmlValidation.toString());
        }
        if (htmlValidation.hasDiffAfterSanitizing) {
            if (config.failOnHtmlDiff) {
                throw new DigipostClientException(ErrorCode.HTML_CONTENT_SANITIZED, SANITATION_DIFF_ADVICE);
            } else {
                LOG.warn("Din html vil forandre seg i Digipost fordi du har elementer som blir lagt til eller fjernet.\n{}", SANITATION_DIFF_ADVICE);
            }
        }
    }

    Optional validateAndSetNrOfPages(Channel channel, Document document, byte[] content, Supplier pdfValidationSettings) {
        if (channel == PRINT && !document.is(PDF)) {
            throw new DigipostClientException(ErrorCode.INVALID_PDF_CONTENT,
                    "PDF is required for direct-to-print messages. Document with uuid " + document.uuid + " had filetype " + document.getDigipostFileType());
        }

        PdfValidationResult pdfValidation;
        Optional pdfInfo;
        if (document.is(PDF)) {
            LOG.debug("Validerer PDF-dokument med uuid '{}'", document.uuid);
            pdfValidation = pdfValidator.validate(content, pdfValidationSettings.get());
            if (document.willBeEncrypted()) {
                document.setNumberOfEncryptedPages(pdfValidation.pages);
            }
            pdfInfo = Optional.of(new PdfInfo(pdfValidation.pages));
        } else {
            pdfValidation = EVERYTHING_OK;
            pdfInfo = empty();
        }

        if ((channel == PRINT && !pdfValidation.okForPrint) || !pdfValidation.okForWeb) {
            throw new DigipostClientException(ErrorCode.INVALID_PDF_CONTENT, pdfValidation.toString());
        }
        return pdfInfo;
    }

    static class PdfInfo {
        final int pages;
        final boolean hasOddNumberOfPages;

        public PdfInfo(int numberOfPages) {
            this.pages = numberOfPages;
            this.hasOddNumberOfPages = pages % 2 == 1;
        }
    }

}