com.codeborne.pdftest.PDF Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pdf-test Show documentation
PDF Test: PDF testing library
There is a newer version: 1.9.1
package com.codeborne.pdftest;

import com.codeborne.pdftest.matchers.ContainsExactText;
import com.codeborne.pdftest.matchers.ContainsText;
import com.codeborne.pdftest.matchers.ContainsTextCaseInsensitive;
import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.pdmodel.interactive.digitalsignature.PDSignature;
import org.apache.pdfbox.text.PDFTextStripper;
import org.hamcrest.Matcher;

import java.io.*;
import java.net.URI;
import java.net.URL;
import java.util.Calendar;

import static com.codeborne.pdftest.IO.readBytes;
import static com.codeborne.pdftest.IO.readFile;
import static com.codeborne.pdftest.IO.toURL;

public class PDF {
  public final byte[] content;
  
  public final String text;
  public final int numberOfPages;
  public final String author;
  public final Calendar creationDate;
  public final String creator;
  public final String keywords;
  public final String producer;
  public final String subject;
  public final String title;
  public final boolean encrypted;
  public final boolean signed;
  public final String signerName;
  public final Calendar signatureTime;

  private PDF(String name, byte[] content) {
    this.content = content;

    try (InputStream inputStream = new ByteArrayInputStream(content)) {
      PDDocument pdf = PDDocument.load(inputStream);
      try {
        this.text = new PDFTextStripper().getText(pdf);
        this.numberOfPages = pdf.getNumberOfPages();
        this.author = pdf.getDocumentInformation().getAuthor();
        this.creationDate = pdf.getDocumentInformation().getCreationDate();
        this.creator = pdf.getDocumentInformation().getCreator();
        this.keywords = pdf.getDocumentInformation().getKeywords();
        this.producer = pdf.getDocumentInformation().getProducer();
        this.subject = pdf.getDocumentInformation().getSubject();
        this.title = pdf.getDocumentInformation().getTitle();
        this.encrypted = pdf.isEncrypted();
        
        PDSignature signature = pdf.getLastSignatureDictionary();
        this.signed = signature != null;
        this.signerName = signature == null ? null : signature.getName();
        this.signatureTime = signature == null ? null : signature.getSignDate();
      }
      finally {
        try {pdf.close();}
        catch (IOException ignore) {}
      }
    }
    catch (Exception e) {
      throw new IllegalArgumentException("Invalid PDF " + name, e);
    }
  }

  public PDF(File pdfFile) {
    this(pdfFile.getAbsolutePath(), readFile(pdfFile));
  }
  
  public PDF(URL url) {
    this(url.toString(), readBytes(url));
  }

  public PDF(URI uri) {
    this(toURL(uri));
  }
  
  public PDF(byte[] content) {
    this("", content);
  }

  public PDF(InputStream inputStream) {
    this(readBytes(inputStream));
  }
  
  public static Matcher containsText(String text) {
    return new ContainsText(text);
  }
  public static Matcher containsExactText(String text) {
    return new ContainsExactText(text);
  }
  public static Matcher containsTextCaseInsensitive(String text) {
    return new ContainsTextCaseInsensitive(text);
  }
}