gvSIG projects pool

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 gvSIG Association.

 *

 * 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 3

 * 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.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.topology.app.mainplugin;

import java.awt.event.ActionEvent;

import java.io.File;

import java.io.IOException;

import java.nio.charset.StandardCharsets;

import javax.swing.AbstractAction;

import static javax.swing.Action.ACTION_COMMAND_KEY;

import static javax.swing.Action.NAME;

import javax.swing.JFileChooser;

import javax.swing.JOptionPane;

import javax.swing.filechooser.FileFilter;

import org.apache.commons.io.FileUtils;

import org.apache.commons.io.FilenameUtils;

import org.apache.commons.lang.StringUtils;

import org.gvsig.andami.IconThemeHelper;

import org.gvsig.andami.plugins.Extension;

import org.gvsig.app.ApplicationLocator;

import org.gvsig.app.ApplicationManager;

import org.gvsig.app.project.Project;

import org.gvsig.configurableactions.ConfigurableActionsMamager;

import org.gvsig.filedialogchooser.FileDialogChooser;

import org.gvsig.filedialogchooser.FileDialogChooserManager;

import org.gvsig.json.Json;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.swing.api.ToolsSwingLocator;

import org.gvsig.tools.swing.api.threadsafedialogs.ThreadSafeDialogsManager;

import org.gvsig.tools.swing.api.windowmanager.Dialog;

import org.gvsig.tools.swing.api.windowmanager.WindowManager;

import org.gvsig.tools.swing.api.windowmanager.WindowManager_v2;

import org.gvsig.tools.util.ToolsUtilLocator;

import org.gvsig.topology.lib.api.TopologyLocator;

import org.gvsig.topology.lib.api.TopologyManager;

import org.gvsig.topology.lib.api.TopologyPlan;

import org.gvsig.topology.swing.api.JTopologyPlanProperties;

import org.gvsig.topology.swing.api.JTopologyReport;

import org.gvsig.topology.swing.api.TopologySwingLocator;

import org.gvsig.topology.swing.api.TopologySwingManager;

/**

 *

 * @author gvSIG Team

 */

@SuppressWarnings("UseSpecificCatch")

public class TopologyExtension extends Extension {

    public static final String TOPOLOGYPLAN_FILE_EXTENSION = "gvtoplan";

    @Override

    public void initialize() {

    }

    @Override

    public void postInitialize() {

        AppTopologyServices services = new AppTopologyServices();

        TopologyManager manager = TopologyLocator.getTopologyManager();

        manager.setDefaultServices(services);

        TopologySwingManager swingManager = TopologySwingLocator.getTopologySwingManager();

        swingManager.setDefaultServices(services);

        IconThemeHelper.registerIcon("document", "document-view-tree-icon", this);

        IconThemeHelper.registerIcon("topology", "topology-tree-folder", this);

        IconThemeHelper.registerIcon("topology", "topology-tree-project", this);

        ConfigurableActionsMamager cfgActionsManager = ToolsUtilLocator.getConfigurableActionsMamager();

        cfgActionsManager.addConfigurableAction(JTopologyPlanProperties.CONFIGURABLE_PANEL_ID, new SaveTopologyPlan());

        cfgActionsManager.addConfigurableAction(JTopologyPlanProperties.CONFIGURABLE_PANEL_ID, new LoadTopologyPlan());

    }

    @Override

    public void execute(String action) {

        if( StringUtils.equalsIgnoreCase("tools-topology-create-or-modify", action) ) {

            I18nManager i18n = ToolsLocator.getI18nManager();

            ApplicationManager application = ApplicationLocator.getManager();

            TopologyManager manager = TopologyLocator.getTopologyManager();

            TopologySwingManager swingManager = TopologySwingLocator.getTopologySwingManager();

            WindowManager_v2 winManager = (WindowManager_v2) ToolsSwingLocator.getWindowManager();

            final JTopologyPlanProperties panel = swingManager.createJTopologyPlan();

            TopologyPlan plan = getCurrentTopologyPlan();

            if( plan == null ) {

                plan = manager.createTopologyPlan();

            }

            panel.put(plan);

            final Dialog dlg = winManager.createDialog(

                    panel.asJComponent(),

                    i18n.getTranslation("_Topology_plan"),

                    null, 

                    WindowManager_v2.BUTTONS_APPLY_OK_CANCEL

            );

            dlg.setButtonLabel(WindowManager_v2.BUTTON_APPLY, i18n.getTranslation("_Execute"));

            dlg.addActionListener((ActionEvent e) -> {

                TopologyPlan plan1;

                switch(dlg.getAction()) {

                    case WindowManager_v2.BUTTON_OK:

                        plan1 = panel.fetch(null);

                        setCurrentTopologyPlan(plan1);

                        break;

                    case WindowManager_v2.BUTTON_APPLY:

                        plan1 = panel.fetch(null);

                        setCurrentTopologyPlan(plan1);

                        JTopologyReport jreport = swingManager.createJTopologyReport(plan1);

                        jreport.put(plan1);

                        winManager.showWindow(

                                jreport.asJComponent(), 

                                i18n.getTranslation("_Topology_plan_error_inspector") + " ("+plan1.getName()+")",

                                WindowManager.MODE.TOOL

                        );

                        Thread th = new Thread(() -> {

                            plan1.execute();

                        }, "TopologyPlan-"+ plan1.getName());

                        th.start();

                        break;

                }

            });

            dlg.show(WindowManager.MODE.WINDOW);

        } else if( StringUtils.equalsIgnoreCase("tools-topology-execute", action) ) {

            I18nManager i18n = ToolsLocator.getI18nManager();

            final TopologySwingManager swingManager = TopologySwingLocator.getTopologySwingManager();

            final WindowManager_v2 winManager = (WindowManager_v2) ToolsSwingLocator.getWindowManager();

            final TopologyPlan plan = getCurrentTopologyPlan();

            if( plan == null ) {

                // TODO: Mensaje de crear plan

                return;

            }

            JTopologyReport panel = swingManager.createJTopologyReport(plan);

            panel.put(plan);

            winManager.showWindow(

                    panel.asJComponent(), 

                    i18n.getTranslation("_Topology_plan_error_inspector") + " ("+plan.getName()+")",

                    WindowManager.MODE.TOOL

            );

            Thread th = new Thread(() -> {

                plan.execute();

            }, "TopologyPlan-"+ plan.getName());

            th.start();

        }

    }

    private static String getCurrentTopologyPlanAsJson() {

        ApplicationManager application = ApplicationLocator.getManager();

//        final ViewDocument view = (ViewDocument) application.getActiveDocument(ViewManager.TYPENAME);

//        if( view==null ) {

//            return null;

//        }

//        String jsonPlan = (String) view.getProperty("TopologyPlan");

        Project project = application.getCurrentProject();

        String jsonPlan = (String) project.getProperty("TopologyPlan");

        return jsonPlan;

    }

    private static boolean setCurrentTopologyPlan(String jsonPlan) {

        ApplicationManager application = ApplicationLocator.getManager();

//        final ViewDocument view = (ViewDocument) application.getActiveDocument(ViewManager.TYPENAME);

//        if( view==null ) {

//            return false;

//        }

//        view.setProperty("TopologyPlan", jsonPlan);

        Project project = application.getCurrentProject();

        project.setProperty("TopologyPlan", jsonPlan);

        return true;

    }

    public static TopologyPlan getCurrentTopologyPlan() {

        String jsonPlan_s = getCurrentTopologyPlanAsJson();

        if( StringUtils.isBlank(jsonPlan_s) ) {

            return null;

        }

        try {

            TopologyManager manager = TopologyLocator.getTopologyManager();

            TopologyPlan plan = manager.createTopologyPlan();

            plan.fromJson(Json.createObject(jsonPlan_s));

            return plan;

        } catch (Exception ex) {

            throw new RuntimeException("Can't retrieve topology plan from json.",ex);

        }

    }

    public static boolean setCurrentTopologyPlan(TopologyPlan plan) {

        return setCurrentTopologyPlan(plan.toJson().toString());

    }

    @Override

    public boolean isEnabled() {

//        ApplicationManager application = ApplicationLocator.getManager();

//        Document view = application.getActiveDocument(ViewManager.TYPENAME);

//        return view!=null;

        return true;

    }

    @Override

    public boolean isVisible() {

        return true;

    }

    private static class SaveTopologyPlan extends AbstractAction {

        @SuppressWarnings("OverridableMethodCallInConstructor")

        public SaveTopologyPlan() {

            I18nManager i18n = ToolsLocator.getI18nManager();

            this.putValue(NAME, i18n.getTranslation("_Save_topology_plan"));

            this.putValue(ACTION_COMMAND_KEY, "SaveTopologyPLanToFile");

        }

        @Override

        public void actionPerformed(ActionEvent e) {

            final I18nManager i18n = ToolsLocator.getI18nManager();

            JTopologyPlanProperties panel = (JTopologyPlanProperties) e.getSource();

            TopologyPlan plan = panel.fetch(null);

            // _Select_topology_plan_file

            // Select topology plan file

            // Seleccionar archivo de plan de topología 

            FileDialogChooserManager fileDislogs = ToolsUtilLocator.getFileDialogChooserManager();

            FileDialogChooser dialog = fileDislogs.create("TopologyPlan");

            dialog.setCurrentDirectory(null);

            dialog.setDialogTitle(i18n.getTranslation("_Select_topology_plan_file"));

            dialog.setMultiSelectionEnabled(false);

            dialog.addChoosableFileFilter(new FileFilter() {

                @Override

                public boolean accept(File f) {

                    return f.getName().endsWith(","+TOPOLOGYPLAN_FILE_EXTENSION);

                }

                @Override

                public String getDescription() {

                    final I18nManager i18n = ToolsLocator.getI18nManager();

                    return i18n.getTranslation("_Topology_plan");

                }

            });

            int n = dialog.showSaveDialog(panel.asJComponent());

            if( n == JFileChooser.APPROVE_OPTION ) {

                File file = dialog.getSelectedFile();

                if( FilenameUtils.indexOfExtension(file.getName())<0 ) {

                    file = new File(file.getAbsolutePath()+"."+TOPOLOGYPLAN_FILE_EXTENSION);

                }

                try {

                    String fcontents = plan.toJson().toString();

                    FileUtils.write(file, fcontents, StandardCharsets.UTF_8);

                } catch (IOException ex) {

                    logger.warn("Can't save topology plan",ex);

                    ThreadSafeDialogsManager dialogs = ToolsSwingLocator.getThreadSafeDialogsManager();

                    dialogs.messageDialog(

                            i18n.getTranslation("_Cant_save_topology_plan"), 

                            i18n.getTranslation("_Topology_plan"), 

                            JOptionPane.WARNING_MESSAGE

                    );

                }

            }

        }

    }

    private static class LoadTopologyPlan extends AbstractAction {

        @SuppressWarnings("OverridableMethodCallInConstructor")

        public LoadTopologyPlan() {

            I18nManager i18n = ToolsLocator.getI18nManager();

            this.putValue(NAME, i18n.getTranslation("_Load_topology_plan"));

            this.putValue(ACTION_COMMAND_KEY, "LoadTopologyPLanFromFile");

        }

        @Override

        public void actionPerformed(ActionEvent e) {

            final I18nManager i18n = ToolsLocator.getI18nManager();

            JTopologyPlanProperties panel = (JTopologyPlanProperties) e.getSource();

            FileDialogChooserManager fileDislogs = ToolsUtilLocator.getFileDialogChooserManager();

            FileDialogChooser dialog = fileDislogs.create("TopologyPlan");

            dialog.setCurrentDirectory(null);

            dialog.setDialogTitle(i18n.getTranslation("_Select_topology_plan_file"));

            dialog.setMultiSelectionEnabled(false);

            dialog.addChoosableFileFilter(new FileFilter() {

                @Override

                public boolean accept(File f) {

                    if( f.isDirectory() ) {

                        return true;

                    }

                    return FilenameUtils.isExtension(f.getName(), TOPOLOGYPLAN_FILE_EXTENSION);

                }

                @Override

                public String getDescription() {

                    return "Topology plan ("+TOPOLOGYPLAN_FILE_EXTENSION+")";

                }

            });

            int n = dialog.showOpenDialog(panel.asJComponent());

            if( n == JFileChooser.APPROVE_OPTION ) {

                File file = dialog.getSelectedFile();

                try {

                    String fcontents = FileUtils.readFileToString(file, StandardCharsets.UTF_8);

                    TopologyManager manager = TopologyLocator.getTopologyManager();

                    TopologyPlan plan = manager.createTopologyPlan();

                    plan.fromJson(Json.createObject(fcontents));

                    panel.put(plan);

                } catch (IOException ex) {

                    logger.warn("Can't load topology plan",ex);

                    ThreadSafeDialogsManager dialogs = ToolsSwingLocator.getThreadSafeDialogsManager();

                    dialogs.messageDialog(

                            i18n.getTranslation("_Cant_load_topology_plan"), 

                            i18n.getTranslation("_Topology_plan"), 

                            JOptionPane.WARNING_MESSAGE

                    );

                }

            }

        }

    }

}