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

import java.io.File;

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.io.IOException;

import java.io.InputStream;

import java.net.URL;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

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

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

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

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

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

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

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

import org.gvsig.installer.lib.spi.execution.InstallPackageProvider;

import org.gvsig.tools.service.Manager;

import org.gvsig.tools.service.ServiceException;

/**

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

 */

public class DefaultInstallPackageService implements InstallPackageService {

    private Map<PackageInfo, File> packageInfoFileMap = null;

    private Map<PackageInfo, String> zipEntriesMap = null;

    private List<PackageInfo> packageInfos = null;

    private Manager manager;

    private static final Logger logger = LoggerFactory

        .getLogger(DefaultInstallPackageService.class);

    private InstallPackageProviderServices installerProviderServices = null;

    public DefaultInstallPackageService(DefaultInstallerManager manager) {

        super();

        this.manager = manager;

        packageInfoFileMap = new HashMap<PackageInfo, File>();

        packageInfos = new ArrayList<PackageInfo>();

        zipEntriesMap = new HashMap<PackageInfo, String>();

        installerProviderServices =

            InstallerProviderLocator.getProviderManager()

                .createInstallerProviderServices();

    }

    public void installPackage(File applicationDirectory,

        PackageInfo packageInfo) throws InstallPackageServiceException {

        if (!applicationDirectory.exists()) {

            throw new InstallPackageServiceException(

                "application_directory_not_found");

        }

        if (!packageInfoFileMap.containsKey(packageInfo)) {

            throw new InstallPackageServiceException("package_not_found");

        }

        InputStream bundleStream = null;

        try {

            File file = packageInfoFileMap.get(packageInfo);

            bundleStream = new FileInputStream(file);

        } catch (FileNotFoundException e) {

            throw new InstallPackageServiceException("not_uri_selected");

        }

        InstallPackageProvider installerExecutionProvider =

            createProvider(packageInfo);

        if (packageInfo.getDownloadURL() == null) {

            InputStream packageStream =

                installerProviderServices.searchPackage(bundleStream,

                    zipEntriesMap.get(packageInfo));

            installerExecutionProvider.install(applicationDirectory,

                packageStream);

        } else {

            File downloadedBundle = downloadFile(packageInfo.getDownloadURL());

            try {

                FileInputStream fis = new FileInputStream(downloadedBundle);

                BufferedInputStream bis = new BufferedInputStream(fis);

                installerExecutionProvider.install(applicationDirectory, bis);

                bis.close();

                fis.close();

            } catch (FileNotFoundException e) {

                throw new InstallPackageServiceException(

                    "Downloaded bundle file not found: " + downloadedBundle, e);

            } catch (IOException e) {

                throw new InstallPackageServiceException(

                    "IO error installing the downloaded bundle: "

                        + downloadedBundle, e);

            }

        }

    }

    public void installPackage(File applicationDirectory, String packageCode)

        throws InstallPackageServiceException {

        PackageInfo packageInfo = getPackageInfo(packageCode);

        if (packageInfo == null) {

            throw new InstallPackageServiceException(

                "The package doesn't exist");

        }

        installPackage(applicationDirectory, packageInfo);

    }

    private InstallPackageProvider createProvider(PackageInfo packageInfo)

        throws InstallPackageServiceException {

        InstallerProviderManager installerProviderManager =

            (InstallerProviderManager) ((DefaultInstallerManager) manager)

                .getProviderManager();

        try {

            return installerProviderManager.createExecutionProvider(packageInfo

                .getType());

        } catch (ServiceException e) {

            throw new InstallPackageServiceException(

                "Error creating the provider", e);

        }

    }

    public PackageInfo getPackageInfo(int index) {

        if (index >= packageInfos.size()) {

            return null;

        }

        return packageInfos.get(index);

    }

    public PackageInfo getPackageInfo(String packageCode) {

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

            if (packageInfos.get(i).getCode().equals(packageCode)) {

                return packageInfos.get(i);

            }

        }

        return null;

    }

    public void addBundle(File bundle) throws InstallPackageServiceException {

        if (!bundle.exists()) {

            throw new InstallPackageServiceException(

                "Only an existing file is supported");

        }

        try {

            int packageInfoCount = packageInfos.size();

            installerProviderServices.readPackageInfo(new FileInputStream(

                bundle), packageInfos, zipEntriesMap);

            for (int i = packageInfoCount; i < packageInfos.size(); i++) {

                packageInfoFileMap.put(packageInfos.get(i), bundle);

            }

        } catch (FileNotFoundException e) {

            logger.error("File not found", e);

        }

    }

    public void addBundle(URL bundleURL) throws InstallPackageServiceException {

        File bundle = downloadFile(bundleURL);

        addBundle(bundle);

    }

    private File downloadFile(URL bundleURL)

        throws InstallPackageServiceException {

        try {

            return InstallerProviderLocator.getProviderManager()

                .createInstallerProviderServices().downloadFile(bundleURL);

        } catch (IOException e) {

            throw new InstallPackageServiceException(

                "Bundle file download error from URL: " + bundleURL, e);

        }

    }

    public void addBundlesFromDirectory(File directory)

        throws InstallPackageServiceException {

        if (!directory.isDirectory()) {

            throw new InstallPackageServiceException(

                "The application directory has to be a directory");

        }

        File[] files = directory.listFiles();

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

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

                addBundle(files[i]);

            }

        }

    }

    public int getPackageCount() {

        if (packageInfos == null) {

            return 0;

        }

        return packageInfos.size();

    }

    public Manager getManager() {

        return this.manager;

    }

}