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.execution;

import java.io.ByteArrayInputStream;

import java.io.ByteArrayOutputStream;

import java.io.File;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.util.ArrayList;

import java.util.List;

import java.util.zip.ZipEntry;

import java.util.zip.ZipException;

import java.util.zip.ZipInputStream;

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

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

import org.gvsig.installer.lib.impl.DefaultInstallerInfo;

import org.gvsig.installer.lib.impl.creation.DefaultInstallerCreationService;

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

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

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

 */

public class Decompress {

        private int BUFFER = 2048;

        private static final Logger logger = LoggerFactory.getLogger(Decompress.class);

        private List<InstallerInfo> installerInfos = null;

        private boolean decompress = false;

        private File outputDirectory = null;

        public void decompressPlugins(InputStream is, File outputDirectory) throws InstallerExecutionServiceException {

                decompress = true;

                this.outputDirectory = outputDirectory;

                decompressFolderOfPlugins(is);

        }        

        public List<InstallerInfo> readInstallInfo(InputStream is) throws InstallerExecutionServiceException{

                decompress = false;

                installerInfos = new ArrayList<InstallerInfo>();

                decompressFolderOfPlugins(is);

                return installerInfos;

        }

        private void decompressFolderOfPlugins(InputStream is) throws InstallerExecutionServiceException {

                ZipInputStream zipInputStream = new ZipInputStream(is);        

                ZipEntry entry = null;

                try {

                        while((entry = zipInputStream.getNextEntry()) != null) {                                

                                logger.debug("Extracting Plugin: " + entry);

                                decompressPlugin(new ZipInputStream(zipInputStream));

                                zipInputStream.closeEntry();

                        }

                        zipInputStream.close();                        

                } catch (Exception e) {

                        throw new InstallerExecutionServiceException("Error reading the plugin", e);

                }                         

        }        

        private void decompressPlugin(ZipInputStream zipInputStream) throws ZipException, IOException, InstallerInfoFileException{

                ZipEntry entry = null;

                byte data[] = new byte[BUFFER];                

                int count;

                while((entry = zipInputStream.getNextEntry()) != null) {                                

                        logger.debug("Extracting: " + entry.getName());        

                        if (decompress){

                                FileOutputStream fos = new FileOutputStream(outputDirectory.getAbsolutePath() + File.separator + entry.getName());

                                while ((count = zipInputStream.read(data, 0, BUFFER)) != -1) {

                                        fos.write(data, 0, count);

                                }

                                fos.flush();                

                                fos.close();        

                        }else{

                                if (entry.getName().endsWith(File.separator + DefaultInstallerCreationService.INSTALLINFO_FILE_NAME)){

                                        installerInfos.add(readInstallInfo(zipInputStream));

                                }

                        }

                        zipInputStream.closeEntry();

                }                

        }        

        private InstallerInfo readInstallInfo(ZipInputStream zipInputStream) throws IOException, InstallerInfoFileException{

                int count;

                byte data[] = new byte[BUFFER];

                ByteArrayOutputStream out = new ByteArrayOutputStream();

                while ((count = zipInputStream.read(data, 0, BUFFER)) != -1) {

                        out.write(data, 0, count);

                }

                out.flush();

                InstallerInfo installerInfo = new DefaultInstallerInfo();

                InstallerInfoFileReader installerInfoFileReader = new InstallerInfoFileReader();

                installerInfoFileReader.read(installerInfo, new ByteArrayInputStream(out.toByteArray()));

                return installerInfo;

        }        

}