controllers.PdfGenerator.java Source code

Introduction

Here is the source code for controllers.PdfGenerator.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */

package controllers;

/**
 *
 * @author Expression EXDER is undefined on line 12, column 14 in Templates/Classes/Class.java.
 */

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.List;

import nu.validator.htmlparser.dom.HtmlDocumentBuilder;

import org.apache.commons.io.FilenameUtils;
import org.apache.commons.io.IOUtils;
import org.w3c.dom.Document;
import org.xhtmlrenderer.pdf.ITextRenderer;

import com.lowagie.text.pdf.BaseFont;

import play.Logger;
import play.Play;
import play.twirl.api.Html;
import play.mvc.Result;
import play.mvc.Results;
import javax.inject.Singleton;

@Singleton
public class PdfGenerator {

    private List<String> defaultFonts = null;

    public PdfGenerator() {

    }

    public void loadTemporaryFonts(List<String> fontsToLoad) {
        defaultFonts = new ArrayList<String>();
        addTemporaryFonts(fontsToLoad);
    }

    public void addTemporaryFonts(List<String> fontsToLoad) {
        if (defaultFonts == null)
            defaultFonts = new ArrayList<String>();
        for (String font : fontsToLoad) {
            try {
                InputStream fin = Play.application().resourceAsStream(font);
                final File tempFile = File.createTempFile("tmp_" + FilenameUtils.getBaseName(font),
                        "." + FilenameUtils.getExtension(font));
                tempFile.deleteOnExit();
                FileOutputStream out = new FileOutputStream(tempFile);
                IOUtils.copy(fin, out);
                defaultFonts.add(tempFile.getAbsolutePath());
            } catch (Exception e) {
                Logger.error("Loading fonts", e);
            }
        }
    }

    public void loadLocalFonts(List<String> fontsToLoad) {
        defaultFonts = new ArrayList<String>();
        addLocalFonts(fontsToLoad);
    }

    public void addLocalFonts(List<String> fontsToLoad) {
        if (defaultFonts == null)
            defaultFonts = new ArrayList<String>();
        for (String font : fontsToLoad)
            defaultFonts.add(font);
    }

    public Result ok(Html html, String documentBaseURL) {
        byte[] pdf = toBytes(html.body(), documentBaseURL);
        return Results.ok(pdf).as("application/pdf");
    }

    public Result ok(Html html, String documentBaseURL, List<String> fonts) {
        byte[] pdf = toBytes(html.body(), documentBaseURL, fonts);
        return Results.ok(pdf).as("application/pdf");
    }

    public byte[] toBytes(Html html, String documentBaseURL) {
        byte[] pdf = toBytes(html.body(), documentBaseURL);
        return pdf;
    }

    public byte[] toBytes(Html html, String documentBaseURL, List<String> fonts) {
        byte[] pdf = toBytes(html.body(), documentBaseURL, fonts);
        return pdf;
    }

    public byte[] toBytes(String string, String documentBaseURL) {
        return toBytes(string, documentBaseURL, defaultFonts);
    }

    public byte[] toBytes(String string, String documentBaseURL, List<String> fonts) {
        ByteArrayOutputStream os = new ByteArrayOutputStream();
        toStream(string, os, documentBaseURL, fonts);
        return os.toByteArray();
    }

    public void toStream(String string, OutputStream os, String documentBaseURL) {
        toStream(string, os, documentBaseURL, defaultFonts);
    }

    public void toStream(String string, OutputStream os, String documentBaseURL, List<String> fonts) {
        try {
            InputStream input = new ByteArrayInputStream(string.getBytes("UTF-8"));
            ITextRenderer renderer = new ITextRenderer();
            if (fonts != null)
                for (String font : fonts) {
                    renderer.getFontResolver().addFont(font, BaseFont.IDENTITY_H, BaseFont.EMBEDDED);
                }
            PdfUserAgent myUserAgent = new PdfUserAgent(renderer.getOutputDevice());
            myUserAgent.setSharedContext(renderer.getSharedContext());
            renderer.getSharedContext().setUserAgentCallback(myUserAgent);
            Document document = new HtmlDocumentBuilder().parse(input);
            renderer.setDocument(document, documentBaseURL);
            renderer.layout();
            renderer.createPDF(os);
        } catch (Exception e) {
            Logger.error("Error creating document from template", e);
        }
    }

}