Application: gvSIG desktop » gvSIG desktop customize

/*

 * To change this license header, choose License Headers in Project Properties.

 * To change this template file, choose Tools | Templates

 * and open the template in the editor.

 */

package org.gvsig.customize;

import java.io.File;

import java.io.FileNotFoundException;

import java.io.FileOutputStream;

import java.io.IOException;

import java.net.URI;

import java.nio.file.FileSystem;

import java.nio.file.FileSystems;

import java.nio.file.Files;

import java.nio.file.Path;

import java.nio.file.Paths;

import java.nio.file.StandardCopyOption;

import java.util.HashMap;

import java.util.Map;

import org.apache.commons.io.FileUtils;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.PluginsLocator;

import org.gvsig.andami.PluginsManager;

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

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

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

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

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 *

 * @author jjdelcerro

 */

public class DistributionBuilder {

    private static final Logger logger = LoggerFactory.getLogger(DistributionBuilder.class);

    private static final String PCKGNAME= "package.gvspks";

    private String distributionId;

    private File onlineInstaller;

    private File workingFolder;

    private File customizePackagePath;

    private MessageBar messagebar;

    private boolean includeCustomizePlugin;

    private File packagesetFolder;

    public DistributionBuilder(MessageBar messagebar) {

        this.messagebar = messagebar;

    }

    private void message(String msg) {

        this.messagebar.message(msg);

    }

    public void setDistributionId(String distributionId) {

        this.distributionId = distributionId;

    }

    public void setOnlineInstaller(File onlineInstaller) {

        this.onlineInstaller = onlineInstaller;

    }

    public void setWorkingFolder(File workingFolder) {

        this.workingFolder = workingFolder;

    }

    void setIncludeCustomizePlugin(boolean includeCustomizePlugin) {

        this.includeCustomizePlugin = includeCustomizePlugin;

    }

    public boolean build() {

        try {

            String target_name = this.onlineInstaller.getName().replace("online", this.distributionId);

            File target = new File(this.workingFolder, target_name);

            message("Creating package set...");

            File packageSet = new File(this.workingFolder, "packages.gvspks");

            createPackageset(packageSet);

            if (this.includeCustomizePlugin) {

                message("Building customize plugin...");

                this.buildCustomizePlugin();

                message("Adding customized plugin to package set...");

                this.addToZip(packageSet, customizePackagePath);

            }

            message("Coping online installer...");

            FileUtils.copyFile(this.onlineInstaller, target);

            message("Adding package set to the installer...");

            addPackagetoZip(target, packageSet);

            message("Installer created.");

        } catch (Exception ex) {

            message("Can't build the distribution.\n" + ex.getMessage());

            logger.warn("Can't build the distribution.");

            logger.warn("working folder:" + this.workingFolder.getAbsolutePath());

            logger.warn("online installer:" + this.onlineInstaller.getAbsolutePath());

            return false;

        }

        return true;

    }

    private void addPackagetoZip(File zipFile, File packageFile) throws IOException{

        FileSystem zipfs = null;

        try {

        Map<String, Object> env = new HashMap<>();

        env.put("create", "true");

        env.put("useTempFile", Boolean.TRUE);

        Path zipPath = Paths.get(zipFile.getAbsolutePath());

        URI uriZip = URI.create("jar:file:" + zipPath.toUri().getPath());

        zipfs = FileSystems.newFileSystem(uriZip,env);

        Path pathInZipfile = zipfs.getPath("/"+PCKGNAME);

        // copy a file into the zip file

        Files.copy( packageFile.toPath(),pathInZipfile,

                StandardCopyOption.REPLACE_EXISTING );

        message("Package set added to the installer.");

        } catch (Exception e) {

            logger.warn("Impossible to add ("+packageFile.getAbsolutePath()+") to ("+zipFile.getAbsolutePath()+")",e);

        } finally{

            if (zipfs != null) {

                try {

                    zipfs.close();

                } catch (IOException ex) {

                }

            }

        }

    }

    private void createPackageset(File zipfile) {

        Map<String, String> env = new HashMap<>();

        env.put("create", "true");

        FileSystem zipfs = null;

        try {

            Path zippath = Paths.get(zipfile.getAbsolutePath());

            final URI uri = URI.create("jar:file:" + zippath.toUri().getPath());

            zipfs = FileSystems.newFileSystem(uri, env);

            File[] packages = packagesetFolder.listFiles();

            for (File newEntry : packages) {

                Path externalFile = Paths.get(newEntry.getAbsoluteFile().toURI());

                Path pathInZipfile = zipfs.getPath("/" + newEntry.getName());

                Files.copy(externalFile, pathInZipfile, StandardCopyOption.REPLACE_EXISTING);

            }

            zipfs.close();

        } catch (Exception ex) {

            logger.warn("Can't create packageset ("+zipfile.getAbsolutePath()+")",ex);

        } finally {

            if (zipfs != null) {

                try {

                    zipfs.close();

                } catch (IOException ex) {

                }

            }

        }

    }

    private void buildCustomizePlugin() throws FileNotFoundException, MakePackageServiceException, IOException {

        FileOutputStream fos = null;

        try {

            PluginsManager pluginsManager = PluginsLocator.getManager();

            PluginServices plugin = pluginsManager.getPlugin(CustomizeExtension.class);

            InstallerManager installManager = InstallerLocator.getInstallerManager();

            FileUtils.copyFileToDirectory(

                    new File(packagesetFolder,"defaultPackages"),

                    plugin.getPluginDirectory()

            );

            PackageInfo pkginfo = pluginsManager.getPackageInfo(CustomizeExtension.class);

            String pkgfilename = installManager.getPackageFileName(pkginfo);

            customizePackagePath = new File(this.workingFolder, pkgfilename);

            MakePackageService maker = installManager.createMakePackage(plugin.getPluginDirectory(), pkginfo);

            fos = new FileOutputStream(customizePackagePath);

            maker.createPackage(fos);

        } finally {

            if (fos != null) {

                fos.close();

            }

        }

    }

    private void addToZip(File zipfile, File newEntry) {

        Map<String, String> env = new HashMap<>();

        env.put("create", "true");

        FileSystem zipfs = null;

        try {

            Path zippath = Paths.get(zipfile.getAbsolutePath());

            final URI uri = URI.create("jar:file:" + zippath.toUri().getPath());

            zipfs = FileSystems.newFileSystem(uri, env);

            Path externalFile = Paths.get(newEntry.getAbsoluteFile().toURI());

            Path pathInZipfile = zipfs.getPath("/" + newEntry.getName());

            Files.copy(externalFile, pathInZipfile, StandardCopyOption.REPLACE_EXISTING);

            zipfs.close();

        } catch (IOException ex) {

        } finally {

            if (zipfs != null) {

                try {

                    zipfs.close();

                } catch (IOException ex) {

                }

            }

        }

    }

    void setPackageSet(File packageSet) {

        this.packagesetFolder = packageSet;

    }

}