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

import java.io.ByteArrayInputStream;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileOutputStream;

import java.io.FileReader;

import java.io.IOException;

import java.io.InputStream;

import java.io.OutputStream;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

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

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

import org.gvsig.installer.lib.spi.InstallerProviderLocator;

import org.gvsig.installer.lib.spi.InstallerProviderServices;

import org.gvsig.tools.service.Manager;

import org.slf4j.Logger;

/**

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

 */

public class DefaultInstallerCreationService implements InstallerCreationService {

        public static final String ANT_FILE_NAME = "install.xml";

        public static final String COPIED_FILES_DIRECTORY_NAME = "files";

        private File applicationDirectory;

        private File pluginsDirectory;

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

        private Manager manager = null;

        private List<InstallerInfo> installerInfos = null;        

        private Map<InstallerInfo, String> directories = null;

        private int pluginToInstallIndex = -1;

        protected List<File> selectedFiles = null;

        protected String antScript = null;

        private InstallerProviderServices installerProviderServices = null;

        public DefaultInstallerCreationService(Manager manager) {

                super();

                this.manager = manager;

                installerInfos = new ArrayList<InstallerInfo>();        

                directories = new HashMap<InstallerInfo, String>();

                selectedFiles = new ArrayList<File>();

                installerProviderServices = InstallerProviderLocator.getProviderManager().createInstallerProviderServices();

        }

        public void createInstaller(OutputStream installerStream)

        throws InstallerCreationServiceException {

                if (applicationDirectory == null){

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

                }

                //Write the install.info file

                writeInstallInfo();

                //Create the ant file

                String antScript = getSelectedInstallerInfo().getAntScript();

                if (antScript != null){

                        writeAntFile(antScript);

                }

                //Copy the selected files

                writeSelectedFiles();

                InstallerInfo infInstallerInfo = getSelectedPlugin();

                String pluginFileName = getSelectedPlugin().getCode() + "-" + infInstallerInfo.getVersion() + "-" + infInstallerInfo.getBuild();

                installerProviderServices.compress(getAbsoluteSelectePluginFile(), pluginFileName, installerStream);

        }

        private void writeInstallInfo() throws InstallerCreationServiceException{

                installerProviderServices.writeInstallInfo(new File(directories.get(getSelectedInstallerInfo())), getSelectedInstallerInfo());

        }

        private void writeSelectedFiles() throws InstallerCreationServiceException {

                try{

                        String copiedFilesDirectory = getCopiedFilesDirectoryName();

                        //Deleting the previous files folder

                        File copiedFilesDirectoryfile = new File(copiedFilesDirectory);

                        deleteDirectory(copiedFilesDirectoryfile);

                        InstallerInfo installerInfo = getSelectedInstallerInfo();

                        for (int i=0 ; i<installerInfo.getFileToCopySize() ; i++){

                                String destFile = installerInfo.getFileToCopyAt(i).getAbsolutePath();

                                String sourceFile = applicationDirectory + File.separator + destFile;

                                //Create the final path

                                destFile = pluginsDirectory + File.separator + copiedFilesDirectory + File.separator +  destFile;                                

                                //Copy the files

                                copy(new File(sourceFile), new File(destFile));

                        }

                }catch(IOException e){

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

                }

        }        

        private void writeAntFile(String content) throws InstallerCreationServiceException {

                try{

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

                        OutputStream out = new FileOutputStream(getAbsoluteAntFile());

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

                }

        }

        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 setApplicationDirectory(File applicationDirectory) throws InstallerCreationServiceException {

                if (!applicationDirectory.isDirectory()){

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

                }

                this.applicationDirectory = applicationDirectory;        

                this.pluginsDirectory = new File(applicationDirectory.getAbsoluteFile() + File.separator + "gvSIG" + File.separator + "extensiones") ;

                if ((!pluginsDirectory.exists()) || (!pluginsDirectory.isDirectory())){

                        throw new InstallerCreationServiceException("The plugins directory does't exists or not is a directory");

                }

                //Read all the installed plugins

                String[] plugins = pluginsDirectory.list();

                if (plugins != null) {

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

                        { 

                                String pluginDirectoryName = plugins[i];

                                File pluginDirectory = getAbsolutePluginFile(pluginDirectoryName);

                                DefaultInstallerInfo installerInfo = new DefaultInstallerInfo();

                                installerProviderServices.readInstallInfo(new File(pluginDirectory.getAbsolutePath()), installerInfo);

                                //Checks if the ant file exists

                                File antFile = new File(getAntFileName(pluginDirectory.getAbsolutePath()));

                                if (antFile.exists()){

                                        try {

                                                installerInfo.setAnScript(readFileAsString(antFile.getAbsolutePath()));

                                        } catch (IOException e) {

                                                logger.error("Not possible to read the ant file");

                                        }

                                }

                                installerInfos.add(installerInfo);

                                directories.put(installerInfo, pluginDirectoryName);                                

                        }

                }        

        }

        private InstallerInfo getSelectedInstallerInfo() throws InstallerCreationServiceException{

                if (pluginToInstallIndex != -1){

                        return installerInfos.get(pluginToInstallIndex);

                }

                throw new InstallerCreationServiceException("There is not a plugin selected");

        }

        private File getAbsoluteSelectePluginFile() throws InstallerCreationServiceException{

                return getAbsolutePluginFile(directories.get(getSelectedInstallerInfo()));

        }

        private File getAbsolutePluginFile(String selectePluginName) throws InstallerCreationServiceException{

                return new File(pluginsDirectory.getAbsolutePath() + File.separator + selectePluginName);

        }

        private String getPluginToInstallDirectory() throws InstallerCreationServiceException{                

                return directories.get(getSelectedInstallerInfo());                

        }

        private File getAbsoluteAntFile() throws InstallerCreationServiceException{                

                return new File(pluginsDirectory.getAbsolutePath() + File.separator + ANT_FILE_NAME);

        }

        private String getCopiedFilesDirectoryName() throws InstallerCreationServiceException{

                return getCopiedFilesDirectoryName(getPluginToInstallDirectory());

        }

        private String getAntFileName() throws InstallerCreationServiceException{

                return getAntFileName(getPluginToInstallDirectory());

        }

        private String getCopiedFilesDirectoryName(String pluginDirectory) throws InstallerCreationServiceException{

                return pluginDirectory + File.separator + COPIED_FILES_DIRECTORY_NAME;

        }        

        private String getAntFileName(String pluginDirectory) throws InstallerCreationServiceException{

                return pluginDirectory + File.separator + ANT_FILE_NAME;

        }

        private void loadAntFile(File antFile) throws IOException, InstallerCreationServiceException {

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

                getSelectedInstallerInfo().setAnScript(fileData.toString());                

        }

        private void loadCopiedFiles(File file) throws InstallerCreationServiceException{

                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 = new File(getPluginToInstallDirectory()).getParentFile().getAbsolutePath() + pluginFile;

                        getSelectedInstallerInfo().addFileToCopy(new File(pluginFile));

                }

        }

        public Manager getManager() {

                return this.manager;

        }

        public InstallerInfo getPluginInfoAt(int index) {

                return installerInfos.get(index);

        }

        public int getPluginsSize() {

                return installerInfos.size();

        }

        public void setSelectedPlugin(int index) throws InstallerCreationServiceException {

                if (index > installerInfos.size()){

                        throw new InstallerCreationServiceException("The plugin doesn't exist");

                }

                pluginToInstallIndex = index;                        

        }

        public void setSelectedPlugin(String code) throws InstallerCreationServiceException {

                int selectedCode = -1;;

                for (int i=0 ; i<getPluginsSize() ; i++){

                        if (installerInfos.get(i).getCode().equals(code)){

                                selectedCode = i;

                                break;

                        }

                }

                if (selectedCode == -1){

                        throw new InstallerCreationServiceException("The plugin doesn't exist");                

                }

                pluginToInstallIndex = selectedCode;

        }

        public InstallerInfo getSelectedPlugin() {

                try {

                        return getSelectedInstallerInfo();

                } catch (InstallerCreationServiceException e) {                        

                        //there is not a selected plugin

                        return null;

                }

        }

        public String getDefaultAntScript() throws InstallerCreationServiceException {

                try {

                        return readFileAsString(getClass().getClassLoader().getResource(ANT_FILE_NAME).getFile());

                } catch (IOException e) {

                        throw new InstallerCreationServiceException("Impossible to read the default ant file", e);

                }

        }

        private String readFileAsString(String filePath)

        throws java.io.IOException{

                StringBuffer fileData = new StringBuffer(1000);

                BufferedReader reader = new BufferedReader(

                                new FileReader(filePath));

                char[] buf = new char[1024];

                int numRead=0;

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

                        String readData = String.valueOf(buf, 0, numRead);

                        fileData.append(readData);

                        buf = new char[1024];

                }

                reader.close();

                return fileData.toString();

        }

}