com.qmetric.document.watermark.PdfWatermarkService.java Source code

Introduction

Here is the source code for com.qmetric.document.watermark.PdfWatermarkService.java

Source

package com.qmetric.document.watermark;

import com.itextpdf.text.pdf.PdfReader;
import com.itextpdf.text.pdf.PdfStamper;
import com.qmetric.document.watermark.strategy.PdfWatermarkStrategy;
import com.qmetric.utilities.file.FileUtils;
import org.apache.commons.vfs.FileObject;

import java.io.IOException;

/**
 * This program is free software; you can redistribute it and/or modify it under the terms of the GNU Affero General Public License version 3 as
 * published by the Free Software Foundation with the addition of the following permission added to Section 15 as permitted in Section 7(a): FOR ANY
 * PART OF THE COVERED WORK IN WHICH THE COPYRIGHT IS OWNED BY 1T3XT, 1T3XT DISCLAIMS THE WARRANTY OF NON INFRINGEMENT OF THIRD PARTY RIGHTS.
 * <br/><br/>This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more details. You should have received a copy of
 * the GNU Affero General Public License along with this program; if not, see http://www.gnu.org/licenses or write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA, 02110-1301 USA, or download the license from the following URL:
 * http://itextpdf.com/terms-of-use/ <br/><br/>The interactive user interfaces in modified source and object code versions of this program must
 * display Appropriate Legal Notices, as required under Section 5 of the GNU Affero General Public License. <br/><br/>In accordance with Section 7(b)
 * of the GNU Affero General Public License, you must retain the producer line in every PDF that is created or manipulated using iText. <br/><br/>You
 * can be released from the requirements of the license by purchasing a commercial license. Buying such a license is mandatory as soon as you develop
 * commercial activities involving the iText software without disclosing the source code of your own applications. These activities include: offering
 * paid services to customers as an ASP, serving PDFs on the fly in a web application, shipping iText with a closed source product.
 */

class PdfWatermarkService {
    private final PdfReaderFactory pdfReaderFactory;

    private final PdfStamperFactory pdfStamperFactory;

    private final FileUtils fileUtils;

    PdfWatermarkService(final PdfReaderFactory pdfReaderFactory, final PdfStamperFactory pdfStamperFactory,
            final FileUtils fileUtils) {
        this.pdfReaderFactory = pdfReaderFactory;
        this.pdfStamperFactory = pdfStamperFactory;
        this.fileUtils = fileUtils;
    }

    void watermark(final String documentSourcePath, final String documentOutputPath,
            final PdfWatermarkStrategy pdfWatermarkStrategy) throws Exception {
        // source reader
        final FileObject sourcePdf = fileUtils.resolveFile(documentSourcePath);

        final PdfReader reader;
        try {
            reader = pdfReaderFactory.newPdfReader(sourcePdf.getContent());
        } catch (final IOException e) {
            throw new RuntimeException(String.format(
                    "Failed to create new pdf reader because pdf password was incorrect. documentSourcePath[%s], "
                            + "documentOutputPath[%s]",
                    documentSourcePath, documentOutputPath), e);
        }

        // create a stamper that will copy the document to a new file
        FileObject outputFile = null;
        PdfStamper outputPdf = null;

        try {
            outputFile = fileUtils.resolveFile(documentOutputPath);

            outputPdf = pdfStamperFactory.newPdfStamper(reader, outputFile.getContent());

            pdfWatermarkStrategy.apply(reader, outputPdf);
        } finally {
            cleanUp(sourcePdf, outputFile, outputPdf);
        }
    }

    private void cleanUp(final FileObject sourcePdf, final FileObject outputFile, final PdfStamper outputPdf)
            throws com.itextpdf.text.DocumentException, IOException {
        if (outputPdf != null) {
            outputPdf.close();
        }

        fileUtils.closeQuietly(outputFile);
        fileUtils.closeQuietly(sourcePdf);
    }
}