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.installer.lib.spi.InstallerProviderLocator;

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

        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 compressPackageIndex(File folder, OutputStream os)

        throws MakePluginPackageServiceException {

        Compress compress = new Compress();

        compress.compressPluginAsPackageIndex(folder, os);

    }

        public void readPackageInfo(File directory, PackageInfo installerInfo)

                        throws InstallerInfoFileException {

                File installInfoFile = new File(directory + File.separator

 + getPackageInfoFileName());

                if (installInfoFile.exists()) {

                        InstallerInfoFileReader reader = new InstallerInfoFileReader();

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

                }

                if (installerInfo.getCode() == null) {

                        installerInfo.setCode(directory.getName());

                }

                if (installerInfo.getName() == null) {

                        installerInfo.setName(directory.getName());

                }

        }

    private String getPackageInfoFileName() {

        return InstallerProviderLocator.getProviderManager()

            .getPackageInfoFileName();

    }

    public void writePackageInfo(File directory, PackageInfo packageInfo)

                        throws InstallerInfoFileException {

        writePackageInfoFile(directory, getPackageInfoFileName(), packageInfo);

        }

    public void writePackageInfoForIndex(File directory, PackageInfo packageInfo)

        throws InstallerInfoFileException {

        writePackageInfoFile(directory, getPackageInfoFileName() + ".index",

            packageInfo);

    }

    private void writePackageInfoFile(File directory, String fileName,

        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

            + fileName);

    }

    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;

        }

}