gvSIG projects pool

package org.gvsig.pdf.lib.impl;

import java.io.ByteArrayInputStream;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.io.IOException;

import java.io.InputStream;

import java.net.URL;

import org.gvsig.pdf.lib.api.PDFDocument;

import org.slf4j.LoggerFactory;

import org.apache.commons.codec.binary.Hex;

import org.apache.commons.codec.DecoderException;

import org.apache.commons.codec.binary.Base64;

/**

 *

 * @author osc

 */

public class DefaultPDFDocument implements PDFDocument {

//    private byte[] byteDocument;

//    private String strDocument;

//    private File fileDocument;

    private InputStream pdf;

    private static final org.slf4j.Logger LOGGER = LoggerFactory

            .getLogger(DefaultPDFDocument.class);

    public DefaultPDFDocument() {

    }

    @Override

    public void setDocument(Object document) {

        this.pdf = this.load(document);

    }

    @Override

    public InputStream getInputStream(){

        return this.pdf;

    }

    public InputStream load(Object source) {

        InputStream pdfInputSteam = null;

        if (source == null) {

            return null;

        } else if (source instanceof File) {

            File fileSource = (File) source;

            try {

                pdfInputSteam = new FileInputStream(fileSource);

            } catch (FileNotFoundException ex) {

                LOGGER.warn("Not been able to convert file to input stream", ex);

                return null;

            }

        } else if (source instanceof URL) {

            URL urlSource = (URL) source;

            try {

                pdfInputSteam = urlSource.openStream();

            } catch (IOException ex) {

                LOGGER.warn("Not been able to open URL stream", ex);

                return null;

            }

        } else if (source instanceof String) {

                String strSource = (String) source;

                File f = new File(strSource);

                if (f.exists()) {

                    try {

                        pdfInputSteam = new FileInputStream(f);

                    } catch (FileNotFoundException ex) {

                        LOGGER.warn("Not been able to load existing file", ex);

                        return null;

                    }

                } else {

                    try {

                        URL url = new URL(strSource);

                        pdfInputSteam = url.openStream();

                        pdfInputSteam.available(); // Force exception if is null

                    } catch (Exception ex) {

                        try {

                            byte[] data = Hex.decodeHex(strSource.toCharArray());

                            pdfInputSteam = new ByteArrayInputStream(data);

                        } catch (DecoderException ex2) {

                            try {

                                byte[] data = Base64.decodeBase64(strSource);

                                pdfInputSteam = new ByteArrayInputStream(data);

                            } catch (Exception ex3) {

                                return null;

                            }

                        }

                    }

                }

        } else if (source instanceof byte[]) {

            byte[] byteSource = (byte[]) source;

            pdfInputSteam = new ByteArrayInputStream(byteSource);

        }

        return pdfInputSteam;

    }

    @Override

    public byte[] toByteArray() {

    if (this.pdf!=null) {

        try {

            byte[] bytePdf = new byte[pdf.available()];

            pdf.read(bytePdf);

            return bytePdf;

        } catch (IOException ex) {

            LOGGER.warn("Not been able to convert pdf to byte array", ex);

        }

    }

    return null;

    }

//    @Override

//    public File toFile() {

//        if (this.fileDocument != null) {

//            return this.fileDocument;

//        } else if (this.strDocument != null) {

//            return new File(this.strDocument);

//        }

//

//        return null;

//    }

//

//    @Override

//    public String toFileString() {

//        if (this.strDocument != null) {

//            return this.strDocument;

//        } else if (this.fileDocument != null) {

//            return fileDocument.getAbsolutePath();

//        }

//        return null;

//    }

        private String bytearray_hex(byte[] data) {

        StringBuilder builder = new StringBuilder();

        for (byte abyte : data) {

            int v = abyte & 0xff;

            builder.append(String.format("%02x", v));

        }

        return builder.toString();

    }

    @Override

    public String toString() {

        byte[] data = this.toByteArray();

        return bytearray_hex(data);

    }

}