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

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;

import org.slf4j.Logger;

/**

 * @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 ANT_FILE_NAME = "install.xml";

        public static final String COPIED_FILES_DIRECTORY_NAME = "files";

        private File pluginDirectory;

        private static final Logger logger = org.slf4j.LoggerFactory.getLogger(DefaultInstallerCreationService.class);

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

                //if (isAdvancedMode){

                        //Create the ant file

                        if (antScript != null){

                                writeAntFile();

                        }else{

                                File antFileModel = new File(getClass().getClassLoader().getResource(ANT_FILE_NAME).getFile());

                                try{

                                        copy(antFileModel, new File(getAntFileName()));

                                }catch(IOException e){

                                        throw new InstallerCreationServiceException("Exception copying the ant file");

                                }                        

                        //}

                        //Copy the selected files

                        writeSelectedFiles();

                }

                Compress compress = new Compress();

                compress.compressPlugin(pluginDirectory, installerStream);                        

        }

        private void writeAntFile() throws InstallerCreationServiceException {

                try{

                        ByteArrayInputStream in = new ByteArrayInputStream(antScript.getBytes());

                        OutputStream out = new FileOutputStream(getAntFileName());

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

                }catch(IOException e){

                        throw new InstallerCreationServiceException("Exception writing the ant file");

                }

        }

        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;

        }

        private String getAntFileName(){

                return pluginDirectory + File.separator + ANT_FILE_NAME;

        }

        public InstallerInfo getInstallerInfo() {

                return this;

        }

        public void loadInstallInfoFromDirectory() throws InstallerCreationServiceException {

                File file = new File(getInstallerInfoFileName());

                try{

                        if (file.exists()){

                                //Read the install.info

                                FileInputStream fis = new FileInputStream(file);

                                InstallerInfoFileReader installerInfoFileReader = new InstallerInfoFileReader();

                                installerInfoFileReader.read(this, fis);

                                //Load the ant file                                

                                File antFile = new File(getAntFileName());

                                if (antFile.exists()){

                                        loadAntFile(antFile);

                                }                

                                //Load the files to copy                        

                                File copiedFilesDirectoryfile = new File(getCopiedFilesDirectoryName());

                                if (copiedFilesDirectoryfile.exists()){

                                        loadCopiedFiles(copiedFilesDirectoryfile);

                                }                

                        }

                }catch(IOException e){

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

                }

        }

        /* (non-Javadoc)

         * @see org.gvsig.installer.lib.impl.DefaultInstallerInfo#getAntScript()

         */

        @Override

        public String getAntScript() {

                if (super.getAntScript() != null){

                        return super.getAntScript();

                }

                //if (isAdvancedMode){

                        try {

                                loadAntFile(new File(getClass().getClassLoader().getResource(ANT_FILE_NAME).getFile()));

                                return antScript;

                        } catch (IOException e) {

                                logger.error("reading the ant script model", e);

                        }

                //}

                return null;

        }

        private void loadAntFile(File antFile) throws IOException {

                StringBuffer fileData = new StringBuffer();

                FileInputStream fis = new FileInputStream(antFile);

                byte[] buf = new byte[1024];

                int numRead = 0;

                while ((numRead = fis.read(buf)) != -1) {

                        String readData = new String(buf, 0, numRead);

                        fileData.append(readData);

                        buf = new byte[1024];

                }

                fis.close();

                setAnScript(fileData.toString());

        }

        private void loadCopiedFiles(File file){

                if (file.isDirectory()){

                        File[] files = file.listFiles();

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

                                loadCopiedFiles(files[i]);

                        }

                }else{

                        //Removing the plugin prefix

                        String pluginFile = file.getAbsolutePath().substring(getCopiedFilesDirectoryName().length(), file.getAbsolutePath().length());

                        //Ading the root directory

                        pluginFile = pluginDirectory.getParentFile().getAbsolutePath() + pluginFile;

                        addFileToCopy(new File(pluginFile));

                }

        }

}