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.

 * 

 */

/*

 * AUTHORS (In addition to CIT):

 * 2010 {Prodevelop}   {Task}

 */

package org.gvsig.installer.lib.impl;

import java.io.BufferedInputStream;

import java.io.BufferedOutputStream;

import java.io.File;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.io.OutputStream;

import java.net.URL;

import java.net.URLConnection;

import java.util.Date;

import java.util.List;

import java.util.Map;

import java.util.StringTokenizer;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.installer.lib.api.PackageInfo;

import org.gvsig.installer.lib.api.creation.MakePluginPackageServiceException;

import org.gvsig.installer.lib.api.execution.InstallPackageServiceException;

import org.gvsig.installer.lib.impl.info.InstallerInfoFileReader;

import org.gvsig.installer.lib.impl.info.InstallerInfoFileWriter;

import org.gvsig.installer.lib.impl.utils.Compress;

import org.gvsig.installer.lib.impl.utils.Decompress;

import org.gvsig.installer.lib.spi.InstallPackageProviderServices;

import org.gvsig.installer.lib.spi.InstallerInfoFileException;

import org.gvsig.tools.service.spi.AbstractProviderServices;

/**

 * @author <a href="mailto:jpiera@gvsig.org">Jorge Piera Llodr&aacute;</a>

 */

public class DefaultInstallerProviderServices extends AbstractProviderServices

    implements InstallPackageProviderServices {

    private static final Logger LOG = LoggerFactory

        .getLogger(DefaultInstallerProviderServices.class);

    private static final String INSTALL_INFO_FILE = "package.info";

    public void decompress(InputStream is, File outputDirectory)

        throws InstallPackageServiceException {

        Decompress decompress = new Decompress();

        decompress.decompressPlugin(is, outputDirectory);

    }

    public PackageInfo readCompressedPackageInfo(InputStream is)

        throws InstallPackageServiceException {

        Decompress decompress = new Decompress();

        try {

            return decompress.readInstallerInfo(is);

        } catch (InstallerInfoFileException e) {

            throw new InstallPackageServiceException(

                "Exception reading a compressed file", e);

        }

    }

    public void compressPackageSet(File folder, String fileName, OutputStream os)

        throws MakePluginPackageServiceException {

        Compress compress = new Compress();

        compress.compressPluginAsPackageSet(folder, fileName, os);

    }

    public void compressPackage(File folder, OutputStream os)

        throws MakePluginPackageServiceException {

        Compress compress = new Compress();

        compress.compressPluginAsPackage(folder, os);

    }

    public void readPackageInfo(File directory, PackageInfo installerInfo)

        throws InstallerInfoFileException {

        File installInfoFile =

            new File(directory + File.separator + INSTALL_INFO_FILE);

        if (installInfoFile.exists()) {

            InstallerInfoFileReader reader = new InstallerInfoFileReader();

            reader.read(installerInfo, installInfoFile.getAbsolutePath());

        } else {

            installerInfo.setCode(directory.getName());

            installerInfo.setName(directory.getName());

        }

    }

    public void writePackageInfo(File directory, PackageInfo installerInfo)

        throws InstallerInfoFileException {

        if (!directory.exists()) {

            throw new InstallerInfoFileException("The directory doesn't exist");

        }

        InstallerInfoFileWriter installerInfoFileWriter =

            new InstallerInfoFileWriter();

        installerInfoFileWriter.write(installerInfo, directory + File.separator

            + INSTALL_INFO_FILE);

    }

    public InputStream searchPackage(InputStream is, String zipEntry)

        throws InstallPackageServiceException {

        Decompress decompress = new Decompress();

        return decompress.searchPlugin(is, zipEntry);

    }

    public void readPackageSetInfo(InputStream is,

        List<PackageInfo> installerInfos, Map<PackageInfo, String> zipEntriesMap)

        throws InstallPackageServiceException {

        Decompress decompress = new Decompress();

        decompress

            .readPackageSetInstallInfos(is, installerInfos, zipEntriesMap);

    }

    public void readPackageInfo(InputStream is,

        List<PackageInfo> installerInfos,

        Map<PackageInfo, String> zipEntriesMap, String name)

        throws InstallPackageServiceException {

        Decompress decompress = new Decompress();

        decompress.readPackageInstallInfo(is, installerInfos, zipEntriesMap,

            name);

    }

    public File downloadFile(URL bundleURL, String defaultFileName)

        throws IOException {

        URL downloadURL = bundleURL;

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

        if (defaultFileName != null) {

            String urlStr = bundleURL.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));

        byte[] data = new byte[1024];

        int count = 0;

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

            bos.write(data, 0, count);

        }

        bis.close();

        bos.flush();

        bos.close();

        return localFile;

    }

    /**

     * 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;

    }

}