org.jpedal.examples.viewer.gui.popups.DownloadProgress Maven / Gradle / Ivy

/*
 * ===========================================
 * Java Pdf Extraction Decoding Access Library
 * ===========================================
 *
 * Project Info:  http://www.idrsolutions.com
 * Help section for developers at http://www.idrsolutions.com/support/
 *
 * (C) Copyright 1997-2017 IDRsolutions and Contributors.
 *
 * This file is part of JPedal/JPDF2HTML5
 *
     This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Lesser General Public
    License as published by the Free Software Foundation; either
    version 2.1 of the License, or (at your option) any later version.

    This library is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
    Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public
    License along with this library; if not, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA


 *
 * ---------------
 * DownloadProgress.java
 * ---------------
 */

package org.jpedal.examples.viewer.gui.popups;

import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.net.URL;

import org.jpedal.io.ObjectStore;
import org.jpedal.utils.LogWriter;

public class DownloadProgress {
    //Load file from URL into file then open file
    File tempURLFile;

    boolean isDownloading = true;

    int progress;

    private final String pdfUrl;

    public DownloadProgress(final String pdfUrl) {

        this.pdfUrl = pdfUrl;

    }

    public void startDownload() {

        final URL url;
        final InputStream is;

        try {
            final int fileLength;
            int fileLengthPercent = 0;

            progress = 0;

            String str = "file.pdf";
            if (pdfUrl.startsWith("jar:/")) {
                is = this.getClass().getResourceAsStream(pdfUrl.substring(4));
            } else {
                //Allow for firefox drag and drop link without enough forward slashes
                if (pdfUrl.startsWith("file:/") && pdfUrl.charAt(7) != '/') {
                    url = new URL(pdfUrl.replaceFirst("file:/*?", "file:///"));
                } else {
                    url = new URL(pdfUrl);
                }

                is = url.openStream();

                str = url.getPath().substring(url.getPath().lastIndexOf('/') + 1);
                fileLength = url.openConnection().getContentLength();
                fileLengthPercent = fileLength / 100;
            }
            final String filename = str;

            tempURLFile = ObjectStore.createTempFile(filename);


            final FileOutputStream fos = new FileOutputStream(tempURLFile);

            // Download buffer
            final byte[] buffer = new byte[4096];

            // Download the PDF document
            int read;
            int current = 0;

            while ((read = is.read(buffer)) != -1) {
                current += read;
                progress = current / fileLengthPercent;
                fos.write(buffer, 0, read);
            }
            fos.flush();

            // Close streams
            is.close();
            fos.close();
            progress = 100;

        } catch (final Exception e) {
            LogWriter.writeLog("[PDF] Exception " + e + " opening URL " + pdfUrl);
            e.printStackTrace();
            progress = 100;
        }

        isDownloading = false;

    }

    public File getFile() {
        return tempURLFile;
    }

    public boolean isDownloading() {
        return isDownloading;
    }

    public int getProgress() {
        return progress;
    }
}