nl.hsac.fitnesse.fixture.slim.PdfFixture Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of hsac-fitnesse-pdf Show documentation

Fixtures to work with PDFs in FitNesse

There is a newer version: 0.1.6

package nl.hsac.fitnesse.fixture.slim;

import org.apache.commons.io.output.StringBuilderWriter;
import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.text.PDFTextStripper;

import java.io.File;
import java.io.IOException;

/**
 * Fixture to work with PDFs.
 */
public class PdfFixture extends SlimFixture {
    private final PDFTextStripper pdfStripper;

    /**
     * Creates new.
     */
    public PdfFixture() {
        try {
            pdfStripper = createPdfTextStripper();
        } catch (IOException e) {
            throw new SlimFixtureException("Unable to create PDF toolkit", e);
        }
    }

    /**
     * @param pdfFile pdf to get text from.
     * @return text in PDF (HTML preformatted).
     */
    public String pdfTextFormatted(String pdfFile) {
        String text = pdfText(pdfFile);
        return "" + text + "";
    }

    /**
     * @param pdfFile pdf to get text from.
     * @return text in PDF (HTML preformatted).
     */
    public String pdfText(String pdfFile) {
        String file = getFilePathFromWikiUrl(pdfFile);
        return getPdfText(file);
    }

    protected String getPdfText(String file) {
        try (PDDocument doc = PDDocument.load(new File(file))) {
            StringBuilderWriter writer = new StringBuilderWriter(2048);
            pdfStripper.writeText(doc, writer);
            String text = writer.toString();
            text = postProcessText(text);
            return text;
        } catch (IOException e) {
            throw new SlimFixtureException("Unable to read PDF: " + file, e);
        }
    }

    protected String postProcessText(String text) {
        // spaces at end of line are not interesting
        return text.replaceAll(" +\\n", "\n");
    }

    protected PDFTextStripper createPdfTextStripper() throws IOException {
        PDFTextStripper pdfStripper = new PDFTextStripper();
        pdfStripper.setLineSeparator("\n");
        pdfStripper.setPageEnd("\n");
        return pdfStripper;
    }

    public PDFTextStripper getPdfStripper() {
        return pdfStripper;
    }
}