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

import java.io.File;

import java.io.FileInputStream;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.io.OutputStream;

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

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

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

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

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

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

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

/**

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

 */

public class DefaultInstallerCreationService extends DefaultInstallerInfo implements InstallerCreationService {

        public static final String INSTALLINFO_FILE_NAME = "install.info";

        public static final String COPIED_FILES_DIRECTORY_NAME = "files";

        private File pluginDirectory;

        public void createInstaller(OutputStream installerStream)

        throws InstallerCreationServiceException {

                if (pluginDirectory == null){

                        throw new InstallerCreationServiceException("The plugin directory has to be set");

                }

                //Write the install.info file

                writeInstallInfo();

                //Copy the selected files

                writeSelectedFiles();

                Compress compress = new Compress();

                compress.compressPlugin(pluginDirectory, installerStream);                        

        }

        private void writeSelectedFiles() throws InstallerCreationServiceException {

                try{

                        String copiedFilesDirectory = getCopiedFilesDirectoryName();

                        //Deleting the previous files folder

                        File copiedFilesDirectoryfile = new File(copiedFilesDirectory);

                        deleteDirectory(copiedFilesDirectoryfile);

                        String parentDirectory = pluginDirectory.getParent();

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

                                String destFile = selectedFiles.get(i).getAbsolutePath();

                                //Deleting the root where the gvSIG plugins are located

                                destFile = destFile.substring(parentDirectory.length(), destFile.length());

                                //Create the final path

                                destFile = copiedFilesDirectory + destFile;

                                //Copy the files

                                copy(selectedFiles.get(i), new File(destFile));

                        }

                }catch(IOException e){

                        throw new InstallerCreationServiceException("Exception copying the files");

                }

        }

        static public boolean deleteDirectory(File path) {

                if( path.exists() ) {

                        File[] files = path.listFiles();

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

                                if(files[i].isDirectory()) {

                                        deleteDirectory(files[i]);

                                }

                                else {

                                        files[i].delete();

                                }

                        }

                }

                return( path.delete() );

        }

        void copy(File sourceLocation , File targetLocation) throws IOException { 

                if (sourceLocation.isDirectory()) {

                        if (!targetLocation.exists()) {

                                targetLocation.mkdir();

                        }

                        String[] children = sourceLocation.list();

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

                                copy(new File(sourceLocation, children[i]),

                                                new File(targetLocation, children[i]));

                        }

                } else {

                        targetLocation.getParentFile().mkdirs();

                        InputStream in = new FileInputStream(sourceLocation);

                        OutputStream out = new FileOutputStream(targetLocation);

                        // Copy the bits from instream to outstream

                        byte[] buf = new byte[1024];

                        int len;

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

                                out.write(buf, 0, len);

                        }

                        in.close();

                        out.close();

                }                 

        }

        public void setPluginDirectory(File pluginDirectory) throws InstallerCreationServiceException {

                if (!pluginDirectory.isDirectory()){

                        throw new InstallerCreationServiceException("The plugin directory has to be a directory");

                }

                this.pluginDirectory = pluginDirectory;                

                //If the install.info file exists

                File file = new File(getInstallerInfoFileName());

                if (file.exists()){

                        readInstallInfo();

                }else{

                        setCode(pluginDirectory.getName());

                        setName(pluginDirectory.getName());

                }

        }

        private void readInstallInfo() throws InstallerInfoFileException{

                InstallerInfoFileReader installerInfoFileReader = new InstallerInfoFileReader();

                installerInfoFileReader.read(this, getInstallerInfoFileName());

        }        

        private void writeInstallInfo() throws InstallerInfoFileException{

                InstallerInfoFileWriter installerInfoFileWriter = new InstallerInfoFileWriter();

                installerInfoFileWriter.write(this, new File(getInstallerInfoFileName()));

        }

        private String getCopiedFilesDirectoryName(){

                return pluginDirectory + File.separator + COPIED_FILES_DIRECTORY_NAME;

        }

        private String getInstallerInfoFileName(){

                return pluginDirectory + File.separator + INSTALLINFO_FILE_NAME;

        }

        public InstallerInfo getInstallerInfo() {

                return this;

        }

}