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

import java.io.BufferedInputStream;

import java.io.BufferedOutputStream;

import java.io.File;

import java.io.FileInputStream;

import java.io.IOException;

import java.io.OutputStream;

import java.util.ArrayList;

import java.util.List;

import java.util.zip.ZipEntry;

import java.util.zip.ZipOutputStream;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

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

/**

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

 */

public class Compress {

    static final int BUFFER = 2048;

    private static final Logger logger = LoggerFactory

        .getLogger(Compress.class);

    public Compress() {

        super();

    }

    public void compressPluginAsPackageSet(File file, String fileName,

        OutputStream os) throws MakePluginPackageServiceException {

        ArrayList<File> files = new ArrayList<File>();

        ArrayList<String> fileNames = new ArrayList<String>();

        files.add(file);

        fileNames.add(fileName);

        compressPluginsAsPackageSet(files, fileNames, os);

    }

    public void compressPluginsAsPackageSet(File directory, OutputStream os)

        throws MakePluginPackageServiceException {

        File[] files = directory.listFiles();

        List<File> filesArray = new ArrayList<File>();

        List<String> fileNamesArray = new ArrayList<String>();

        for (int i = 0; i < files.length; i++) {

            filesArray.add(files[i]);

            fileNamesArray.add(files[i].getName());

        }

        compressPluginsAsPackageSet(filesArray, fileNamesArray, os);

    }

    public void compressPluginsAsPackageSet(List<File> files,

        List<String> fileNames, OutputStream os)

        throws MakePluginPackageServiceException {

        try {

            ZipOutputStream zos =

                new ZipOutputStream(new BufferedOutputStream(os));

            for (int i = 0; i < files.size(); i++) {

                ZipEntry zipEntry = new ZipEntry(fileNames.get(i));

                zos.putNextEntry(zipEntry);

                compressPluginFiles(files.get(i), zos);

                zos.closeEntry();

            }

            zos.close();

        } catch (Exception e) {

            logger.error("Error compressing the plugin");

            throw new MakePluginPackageServiceException(

                "Error writing the plugin", e);

        }

    }

    public void compressPluginAsPackage(File folder, OutputStream os)

        throws MakePluginPackageServiceException {

        ZipOutputStream zos = new ZipOutputStream(os);

        try {

            ZipEntry zipEntry = new ZipEntry(folder.getName());

            zos.putNextEntry(zipEntry);

            compressPluginFiles(folder, zos);

            zos.closeEntry();

            zos.flush();

            zos.close();

        } catch (IOException e) {

            throw new MakePluginPackageServiceException(

                "Error compressing the plugin folder: " + folder, e);

        }

    }

    private void compressPluginFiles(File fileOrFolder, ZipOutputStream zos)

        throws IOException {

        int parentFileLenght = fileOrFolder.getParentFile().toString().length();

        ZipOutputStream zosPlugin = new ZipOutputStream(zos);

        compressPluginFile(fileOrFolder, parentFileLenght, zosPlugin);

        zosPlugin.finish();

    }

    private void compressPluginFile(File file, int parenFileLength,

        ZipOutputStream zosPlugin) throws IOException {

        String fileName =

            file.toString()

                .substring(parenFileLength, file.toString().length());

        if (file.isDirectory()) {

            // Remove the subversion folders

            if (!file.getName().toUpperCase().equals(".SVN")) {

                // Adding the files

                String[] fileNames = file.list();

                if (fileNames != null) {

                    for (int i = 0; i < fileNames.length; i++) {

                        compressPluginFile(new File(file, fileNames[i]),

                            parenFileLength, zosPlugin);

                    }

                }

            }

        } else {

            byte[] buf = new byte[1024];

            int len;

            ZipEntry zipEntry = new ZipEntry(fileName);

            zosPlugin.putNextEntry(zipEntry);

            FileInputStream fin = new FileInputStream(file);

            BufferedInputStream in = new BufferedInputStream(fin);

            while ((len = in.read(buf)) >= 0) {

                zosPlugin.write(buf, 0, len);

            }

            in.close();

            zosPlugin.closeEntry();

        }

    }

}