controllers.PdfUserAgent.java Source code

Introduction

Here is the source code for controllers.PdfUserAgent.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.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.MalformedURLException;
import java.net.URL;

import org.xhtmlrenderer.pdf.ITextFSImage;
import org.xhtmlrenderer.pdf.ITextOutputDevice;
import org.xhtmlrenderer.pdf.ITextUserAgent;
import org.xhtmlrenderer.resource.CSSResource;
import org.xhtmlrenderer.resource.ImageResource;
import org.xhtmlrenderer.resource.XMLResource;

import play.Logger;
import play.api.Play;
import scala.Option;

import com.lowagie.text.Image;

public class PdfUserAgent extends ITextUserAgent {

    public PdfUserAgent(ITextOutputDevice outputDevice) {
        super(outputDevice);
    }

    @Override
    public ImageResource getImageResource(String uri) {
        Option<InputStream> option = Play.current().resourceAsStream(uri);
        if (option.isDefined()) {
            InputStream stream = option.get();
            try {
                Image image = Image.getInstance(getData(stream));
                scaleToOutputResolution(image);
                return new ImageResource(uri, new ITextFSImage(image));
            } catch (Exception e) {
                Logger.error("fetching image " + uri, e);
                throw new RuntimeException(e);
            }
        } else {
            return super.getImageResource(uri);
        }
    }

    @Override
    public CSSResource getCSSResource(String uri) {
        Option<InputStream> option = Play.current().resourceAsStream(uri);
        if (option.isDefined())
            return new CSSResource(option.get());
        try {
            // uri is in fact a complete URL
            new URL(uri).getPath();
            return super.getCSSResource(uri);
        } catch (MalformedURLException e) {
            Logger.error("fetching css " + uri, e);
            throw new RuntimeException(e);
        }
    }

    @Override
    public byte[] getBinaryResource(String uri) {
        Option<InputStream> option = Play.current().resourceAsStream(uri);
        if (option.isDefined()) {
            InputStream stream = option.get();
            ByteArrayOutputStream baos = new ByteArrayOutputStream();
            try {
                copy(stream, baos);
            } catch (IOException e) {
                Logger.error("fetching binary " + uri, e);
                throw new RuntimeException(e);
            }
            return baos.toByteArray();
        } else {
            return super.getBinaryResource(uri);
        }
    }

    @Override
    public XMLResource getXMLResource(String uri) {
        Option<InputStream> option = Play.current().resourceAsStream(uri);
        if (option.isDefined()) {
            return XMLResource.load(option.get());
        } else {
            return super.getXMLResource(uri);
        }
    }

    private void scaleToOutputResolution(Image image) {
        float factor = getSharedContext().getDotsPerPixel();
        image.scaleAbsolute(image.getPlainWidth() * factor, image.getPlainHeight() * factor);
    }

    private byte[] getData(InputStream stream) throws IOException {
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        copy(stream, baos);
        return baos.toByteArray();
    }

    private void copy(InputStream stream, OutputStream os) throws IOException {
        byte[] buffer = new byte[1024];
        while (true) {
            int len = stream.read(buffer);
            os.write(buffer, 0, len);
            if (len < buffer.length)
                break;
        }
    }
}