Application: gvSIG desktop

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * of the License, or (at your option) any later version.

 *

 * This program 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 General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 */

package org.gvsig.installer.lib.impl.utils;

import java.io.BufferedInputStream;

import java.io.BufferedOutputStream;

import java.io.File;

import java.io.FileOutputStream;

import java.io.IOException;

import java.net.URL;

import java.net.URLConnection;

import java.util.Date;

import java.util.StringTokenizer;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.tools.task.AbstractMonitorableTask;

import org.gvsig.tools.task.SimpleTaskStatus;

/**

 * @author gvSIG Team

 * @version $Id$

 * 

 */

public class Download extends AbstractMonitorableTask {

        private static final Logger LOG = LoggerFactory.getLogger(Download.class);

        private boolean isMyTaskStatus;

        /**

         * @param taskName

         */

        public Download(SimpleTaskStatus taskStatus) {

                this();

                this.taskStatus = taskStatus;

                isMyTaskStatus = false;

        }

        public Download() {

                super("Downloading...");

                isMyTaskStatus = true;

        }

        public File downloadFile(URL url, String defaultFileName)

                        throws IOException {

                URL downloadURL = url;

                // check if the URL ends with '/' and append the file name

                if (defaultFileName != null) {

                        String urlStr = url.toString();

                        if (urlStr.endsWith("/")) {

                                urlStr = urlStr.concat(defaultFileName);

                                downloadURL = new URL(urlStr);

                        }

                }

                URLConnection connection = downloadURL.openConnection();

                String fileName = getFileName(connection);

                if (LOG.isDebugEnabled()) {

                        Date date = new Date(connection.getLastModified());

                        LOG

                                        .debug(

                                                        "Downloading file {} from URL {}, with last modified date: {}",

                                                        new Object[] { fileName, downloadURL, date });

                }

                String fileNamePrefix = fileName;

                String fileNameSuffix = "zip";

                int dotPosition = fileName.lastIndexOf('.');

                if ((dotPosition > -1) && (dotPosition < fileName.length() - 1)) {

                        fileNamePrefix = fileName.substring(0, dotPosition);

                        fileNameSuffix = fileName.substring(dotPosition);

                }

                BufferedInputStream bis = new BufferedInputStream(downloadURL

                                .openStream());

                File localFile = File.createTempFile(fileNamePrefix, fileNameSuffix);

                BufferedOutputStream bos = new BufferedOutputStream(

                                new FileOutputStream(localFile));

                this.taskStatus.setRangeOfValues(0, connection.getContentLength());

                try {

                        byte[] data = new byte[1024];

                        int count = 0;

                        long totalCount = 0; // total bytes readed

                        while ((count = bis.read(data, 0, 1024)) >= 0) {

                                bos.write(data, 0, count);

                                totalCount += count;

                                this.taskStatus.setCurValue(totalCount);

                                if (this.taskStatus.isCancellationRequested()) {

                                        return null;

                                }

                        }

                        if (isMyTaskStatus) {

                                this.taskStatus.terminate();

                                this.taskStatus.remove();

                        }

                        return localFile;

                } finally {

                        bis.close();

                        bos.flush();

                        bos.close();

                }

        }

        /**

         * Returns the file name associated to an url connection.<br />

         * The result is not a path but just a file name.

         * 

         * @param urlConnection

         *            - the url connection

         * @return the file name

         * 

         * @throws IOException

         *             Signals that an I/O exception has occurred.

         */

        private String getFileName(URLConnection urlConnection) throws IOException {

                String fileName = null;

                String contentDisposition = urlConnection

                                .getHeaderField("content-disposition");

                if (contentDisposition != null) {

                        fileName = extractFileNameFromContentDisposition(contentDisposition);

                }

                // if the file name cannot be extracted from the content-disposition

                // header, using the url.getFilename() method

                if (fileName == null) {

                        StringTokenizer st = new StringTokenizer(urlConnection.getURL()

                                        .getFile(), "/");

                        while (st.hasMoreTokens()) {

                                fileName = st.nextToken();

                        }

                }

                return fileName;

        }

        /**

         * Extract the file name from the content disposition header.

         * <p>

         * See <a

         * href="http://www.w3.org/Protocols/rfc2616/rfc2616-sec19.html">http:

         * //www.w3.org/Protocols/rfc2616/rfc2616-sec19.html</a> for detailled

         * information regarding the headers in HTML.

         * 

         * @param contentDisposition

         *            - the content-disposition header. Cannot be <code>null>/code>.

         * @return the file name, or <code>null</code> if the content-disposition

         *         header does not contain the filename attribute.

         */

        private String extractFileNameFromContentDisposition(

                        String contentDisposition) {

                String[] attributes = contentDisposition.split(";");

                for (String a : attributes) {

                        if (a.toLowerCase().contains("filename")) {

                                // The attribute is the file name. The filename is between

                                // quotes.

                                return a.substring(a.indexOf('\"') + 1, a.lastIndexOf('\"'));

                        }

                }

                // not found

                return null;

        }

}