Application: gvSIG desktop » gvSIG desktop customize

package org.gvsig.customize;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.io.File;

import java.io.FileOutputStream;

import java.io.IOException;

import java.nio.file.DirectoryIteratorException;

import java.nio.file.DirectoryStream;

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.util.ArrayList;

import java.util.Arrays;

import java.util.Collections;

import java.util.HashMap;

import java.util.HashSet;

import java.util.List;

import java.util.Map;

import java.util.Set;

import javax.swing.DefaultListModel;

import javax.swing.JTextField;

import javax.swing.ListModel;

import javax.swing.ListSelectionModel;

import javax.swing.SwingUtilities;

import javax.swing.event.ListSelectionEvent;

import javax.swing.event.ListSelectionListener;

import org.apache.commons.io.FileUtils;

import org.apache.commons.io.IOUtils;

import org.apache.commons.lang3.BooleanUtils;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.PluginsLocator;

import org.gvsig.andami.PluginsManager;

import org.gvsig.app.ApplicationLocator;

import org.gvsig.app.ApplicationManager;

import org.gvsig.tools.dynobject.DynObject;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class DistributionParametersPanel extends DistributionParametersPanelLayout implements MessageBar {

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

    private List<ActionListener> listeners = new ArrayList<ActionListener>();

    private final DynObject pluginProperties;

    private final PluginServices plugin;

    private JListWithCheckbox lchkPackages;

    private Set<String> defaultPackages = new HashSet<String>();

    public DistributionParametersPanel() {

        super();

        PluginsManager pluginManager = PluginsLocator.getManager();

        this.plugin = pluginManager.getPlugin(CustomizeExtension.class);

        this.pluginProperties = plugin.getPluginProperties();

        this.initComponents();

    }

    private String getPropertyPath(String name) {

        File f = (File) pluginProperties.getDynValue(name);

        if (f == null) {

            return "";

        }

        return f.getAbsolutePath();

    }

    private void initComponents() {

        this.lchkPackages = new JListWithCheckbox(this.lstPackages);

        this.setCloseAndCreateVisible(false);

        this.txtOnlineInstaller.setEditable(false);

        this.txtPackageSet.setEditable(false);

        this.txtWorkingFolder.setEditable(false);

        this.chkIncludeCustomizePlugin.setSelected(false);

        loadFormValues();

        this.butBrowseOnlineInstaller.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent ae) {

                doBrowseFile(getOnlineInstaller(), "Select the online installer", txtOnlineInstaller);

            }

        });

        this.butBrowsePackageSet.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent ae) {

                doBrowseFile(getPackageSetFolder(), "Select the package set", txtPackageSet);

                unzipPackagset();

            }

        });

        this.butBrowseWorkingFolder.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent ae) {

                doBrowseFolder(getWorkingFolder(), "Select the working folder", txtWorkingFolder);

            }

        });

        this.butClose.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent ae) {

                doClose();

            }

        });

        this.butCreate.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent ae) {

                doCreate();

            }

        });

        this.butAddPackage.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent ae) {

                doAddPackage();

            }

        });

        this.butRemovePackage.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent ae) {

                doRemovePackage();

            }

        });

        this.lchkPackages.addChecksListener(new ListSelectionListener() {

            @Override

            public void valueChanged(ListSelectionEvent lse) {

                doChecksValueChanged(lse);

            }

        });

        this.butCheckAll.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent ae) {

                doCheckAll();

            }

        });

        this.butUncheckAll.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent ae) {

                doUncheckAll();

            }

        });

    }

    private void doCheckAll() {

        ListModel model = this.lchkPackages.getModel();

        this.lchkPackages.getCheckedModel().addSelectionInterval(0, model.getSize()-1);

        for( int i=0; i<model.getSize(); i++) {

            String filename = (String) model.getElementAt(i);

            String code = getPackageCode(filename);

            this.defaultPackages.add(code);

        }        

    }

    private void doUncheckAll() {

        this.lchkPackages.getCheckedModel().clearSelection();

        this.defaultPackages.clear();

    }

    private void doChecksValueChanged(ListSelectionEvent lse) {

        ListSelectionModel checkedsModel = this.lchkPackages.getCheckedModel();

        ListModel model = this.lchkPackages.getModel();

        for (int i = lse.getFirstIndex(); i <= lse.getLastIndex(); i++) {

            String packageCode = getPackageCode((String) model.getElementAt(i));

            if (checkedsModel.isSelectedIndex(i)) {

                this.defaultPackages.add(packageCode);

            } else {

                this.defaultPackages.remove(packageCode);

            }

        }

    }

    private void loadFormValues() {

        this.txtDistributionId.setText((String) pluginProperties.getDynValue("distributionID"));

        this.txtOnlineInstaller.setText(getPropertyPath("onlineInstaller"));

        this.txtPackageSet.setText(getPropertyPath("packageSet"));

        this.txtWorkingFolder.setText(getPropertyPath("WorkingFolder"));

        this.chkIncludeCustomizePlugin.setSelected(BooleanUtils.isTrue((Boolean) pluginProperties.getDynValue("includeCustomizePlugin")));

        unzipPackagset();

        fillPackageList();

    }

    private void saveFormValues() {

        pluginProperties.setDynValue("onlineInstaller", getOnlineInstaller());

        pluginProperties.setDynValue("packageSet", getPackageSet());

        pluginProperties.setDynValue("WorkingFolder", getWorkingFolder());

        pluginProperties.setDynValue("distributionID", getDistributionId());

        pluginProperties.setDynValue("includeCustomizePlugin", getIncludeCustomizePlugin());

        plugin.savePluginProperties();

    }

    private void updateDefaultPackagesFile() {

        ArrayList lines = new ArrayList();

        lines.addAll(this.defaultPackages);

        if( this.getIncludeCustomizePlugin() ) {

            lines.add(this.plugin.getPluginName());

        }

        Collections.sort(lines);

        try {

            FileUtils.writeLines(

                    new File(this.getPackageSetFolder(), "defaultPackages"), 

                    lines

            );

        } catch (IOException ex) {

            this.message("Can't write defaultPackages. " + ex.getMessage());

            logger.warn("Can't write defaultPackages.", ex);

        }

    }

    private void doCreate() {

        saveFormValues();

        updateDefaultPackagesFile();

        this.fireActionPerformed();

    }

    private void doClose() {

        saveFormValues();

        this.setVisible(false);

    }

    private void doBrowseFile(File currentValue, String label, JTextField field) {

        ApplicationManager application = ApplicationLocator.getManager();

        File folder = currentValue;

        if (folder != null) {

            folder = folder.getParentFile();

        }

        File[] folders = application.showOpenFileDialog(label, folder);

        if (folders != null && folders.length == 1) {

            field.setText(folders[0].getAbsolutePath());

        }

    }

    private void doBrowseFolder(File currentValue, String label, JTextField field) {

        ApplicationManager application = ApplicationLocator.getManager();

        File folder = currentValue;

        if (folder != null) {

            folder = folder.getParentFile();

        }

        File[] folders = application.showOpenDirectoryDialog(label, folder);

        if (folders != null && folders.length == 1) {

            field.setText(folders[0].getAbsolutePath());

        }

    }

    public void message(final String msg) {

        if (!SwingUtilities.isEventDispatchThread()) {

            SwingUtilities.invokeLater(new Runnable() {

                @Override

                public void run() {

                    message(msg);

                }

            });

            return;

        }

        this.lblMessages.setText("<html>" + msg.replace("\n", "<br>\n") + "</html>");

    }

    public File getOnlineInstaller() {

        String path = this.txtOnlineInstaller.getText();

        if (StringUtils.isBlank(path)) {

            return null;

        }

        return new File(path);

    }

    private File getPackageSet() {

        String path = this.txtPackageSet.getText();

        if (StringUtils.isBlank(path)) {

            return null;

        }

        return new File(path);        

    }

    public File getPackageSetFolder() {

        File packagesetFolder = new File(getWorkingFolder(), "packages");

        return packagesetFolder;   

    }

    public File getWorkingFolder() {

        String path = this.txtWorkingFolder.getText();

        if (StringUtils.isBlank(path)) {

            return null;

        }

        return new File(path);

    }

    public String getDistributionId() {

        String id = this.txtDistributionId.getText();

        if (StringUtils.isBlank(id)) {

            return null;

        }

        return id.trim();

    }

    public boolean getIncludeCustomizePlugin() {

        return this.chkIncludeCustomizePlugin.isSelected();

    }

    public void setCloseAndCreateVisible(boolean visible) {

        this.butClose.setVisible(visible);

        this.butCreate.setVisible(visible);

    }

    public void setCloseAndCreateEnabled(final boolean enabled) {

        if (!SwingUtilities.isEventDispatchThread()) {

            SwingUtilities.invokeLater(new Runnable() {

                @Override

                public void run() {

                    setCloseAndCreateEnabled(enabled);

                }

            });

            return;

        }

        this.butClose.setEnabled(enabled);

        this.butCreate.setEnabled(enabled);

    }

    public void addActionListener(ActionListener l) {

        if (this.listeners.contains(l)) {

            return;

        }

        this.listeners.add(l);

    }

    protected void fireActionPerformed() {

        ActionEvent event = new ActionEvent(this, 0, "create_distribution");

        for (ActionListener listener : listeners) {

            try {

                listener.actionPerformed(event);

            } catch (Exception ex) {

                logger.warn("Error triggering event", ex);

            }

        }

    }

    private void loadDefaultPackages(File packages) {

        try {

            this.defaultPackages = new HashSet<String>();

            List<String> lines = FileUtils.readLines(packages);

            this.defaultPackages.clear();

            for (String line : lines) {

                line = line.trim();

                if (line.startsWith("#") || line.startsWith(";")) {

                    continue;

                }

                this.defaultPackages.add(line.toLowerCase());

            }

        } catch (IOException ex) {

            logger.warn("Can't load defaultPackages from '"+packages.getAbsolutePath()+"'.",ex);

        }

    }

    private String getPackageCode(String filename) {

        if( StringUtils.isEmpty(filename) ) {

            return filename;

        }

        String[] ss = filename.split("-");

        if (ss.length > 3) {

            return ss[3].toLowerCase();

        }

        return "";

    }

    private void fillPackageList() {

        File f = this.getWorkingFolder();

        if (f == null) {

            return;

        }

        DefaultListModel model = new DefaultListModel();

        int first = this.lchkPackages.getFirstVisibleIndex();

        File packagesetFolder = new File(f, "packages");

        if (packagesetFolder.exists()) {

            File[] packagesArray = packagesetFolder.listFiles();

            List<File>packages = Arrays.asList(packagesArray);

            Collections.sort(packages);

            for (File package1 : packages) {

                String filename = package1.getName();

                if ( ! "defaultPackages".equalsIgnoreCase(filename)) {

                    model.addElement(filename);

                }

            }

        }

        this.lchkPackages.setModel(model);

        updateCheckeds();

        this.lchkPackages.ensureIndexIsVisible(first);

    }

    private void updateCheckeds() {

        this.lchkPackages.getCheckedModel().clearSelection();

        ListModel model = this.lchkPackages.getModel();

        for( int i=0; i<model.getSize(); i++) {

            String filename = (String) model.getElementAt(i);

            String code = getPackageCode(filename);

            if (this.defaultPackages.contains(code)) {

                this.lchkPackages.getCheckedModel().addSelectionInterval(i, i);

            }

        }

    }

    private void unzipPackagset() {

        final Runnable updategui = new Runnable() {

            @Override

            public void run() {

                fillPackageList();

                updateCheckeds();

                lchkPackages.setEnabled(true);

                butBrowsePackageSet.setEnabled(true);

                butAddPackage.setEnabled(true);

                butRemovePackage.setEnabled(true);

            }

        };

        final Runnable task = new Runnable() {

            @Override

            public void run() {

                message("Updating package set");

                File packageSet = getPackageSet();

                if (packageSet == null) {

                    return;

                }

                if( getWorkingFolder() == null ) {

                    return;

                }

                File packagesetFolder = getPackageSetFolder();

                if (packagesetFolder.exists()) {

                    try {

                        FileUtils.deleteDirectory(packagesetFolder);

                    } catch (IOException ex) {

                        message("Can't delete 'packages' directory. " + ex.getMessage());

                    }

                }

                packagesetFolder.mkdir();

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

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

                FileSystem zipfs = null;

                try {

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

                    zipfs = FileSystems.newFileSystem(zippath, null);

                    Iterable<Path> dirs = zipfs.getRootDirectories();

                    for (Path name : dirs) {

                        DirectoryStream<Path> stream = Files.newDirectoryStream(name);

                        for (Path file : stream) {

                            FileOutputStream fos = null;

                            try {

                                fos = new FileOutputStream(new File(packagesetFolder, file.getFileName().toString()));

                                Files.copy(file, fos);

                            } catch (IOException | DirectoryIteratorException x) {

                                // Do nothing

                            } finally {

                                if (fos != null) {

                                    IOUtils.closeQuietly(fos);

                                }

                            }

                        }

                    }

                    loadDefaultPackages(new File(packagesetFolder, "defaultPackages"));

                } catch (Exception ex) {

                } finally {

                    message("");

                    if (zipfs != null) {

                        try {

                            zipfs.close();

                        } catch (IOException ex) {

                        }

                    }

                    SwingUtilities.invokeLater(updategui);

                }

            }

        };

        lchkPackages.setEnabled(false);

        butBrowsePackageSet.setEnabled(false);

        butAddPackage.setEnabled(false);

        butRemovePackage.setEnabled(false);

        Thread thread = new Thread(task);

        thread.start();

    }

    private void doRemovePackage() {

        List<String> filenames = this.lchkPackages.getSelectedValuesList();

        if (filenames == null || filenames.isEmpty()) {

            return;

        }

        File packagesetFolder = new File(getWorkingFolder(), "packages");

        for( String filename : filenames ) {

            File filetoremove = new File(packagesetFolder, filename);

            filetoremove.delete();

        }

        fillPackageList();

     }

    private void doAddPackage() {

        ApplicationManager application = ApplicationLocator.getManager();

        File[] files = application.showOpenFileDialog("Select package to add", this.getWorkingFolder());

        if (files == null) {

            return;

        }

        File packagesetFolder = new File(getWorkingFolder(), "packages");

        for (File file : files) {

            try {

                FileUtils.copyFileToDirectory(file, packagesetFolder);

            } catch (IOException ex) {

                this.message("Can't copy package to working folder. " + ex.getMessage());

            }

        }

        fillPackageList();

    }

}