Application: gvSIG desktop

/**

 * 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.app.extension;

import java.awt.BorderLayout;

import java.awt.Component;

import java.awt.Font;

import java.awt.GridBagConstraints;

import java.awt.GridBagLayout;

import java.awt.Insets;

import java.util.ArrayList;

import java.util.List;

import javax.swing.JLabel;

import javax.swing.JOptionPane;

import javax.swing.JPanel;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.andami.IconThemeHelper;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.PluginsLocator;

import org.gvsig.andami.actioninfo.ActionInfo;

import org.gvsig.andami.messages.NotificationManager;

import org.gvsig.andami.plugins.IExtension;

import org.gvsig.andami.plugins.status.IExtensionStatus;

import org.gvsig.andami.plugins.status.IUnsavedData;

import org.gvsig.andami.plugins.status.UnsavedData;

import org.gvsig.andami.ui.mdiManager.IWindow;

import org.gvsig.app.ApplicationLocator;

import org.gvsig.app.ApplicationManager;

import org.gvsig.app.project.Project;

import org.gvsig.app.project.ProjectManager;

import org.gvsig.app.project.documents.Document;

import org.gvsig.app.project.documents.table.TableDocument;

import org.gvsig.app.project.documents.table.TableManager;

import org.gvsig.app.project.documents.table.gui.FeatureTableDocumentPanel;

import org.gvsig.fmap.dal.EditingNotification;

import org.gvsig.fmap.dal.EditingNotificationManager;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.exception.ReadException;

import org.gvsig.fmap.dal.exception.WriteException;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.dal.swing.DALSwingLocator;

import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;

import org.gvsig.gui.beans.Messages;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

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

import org.gvsig.tools.util.ArrayUtils;

import org.gvsig.utils.swing.threads.IMonitorableTask;

public class TableEditStopExtension extends AbstractTableEditExtension {

    private static Logger LOGGER

            = LoggerFactory.getLogger(TableEditStopExtension.class);

    @Override

    public void initialize() {

        super.initialize();

        IconThemeHelper.registerIcon("action", "table-stop-editing", this);

    }

    public void execute(String actionCommand) {

        this.execute(actionCommand,null);

    }

    @Override

    public void execute(String actionCommand, Object[] args) {

        if ("table-stop-editing".equals(actionCommand)) {

            TableDocument doc = (TableDocument)ArrayUtils.get(args,0);

            if( doc == null ) {

                doc = (TableDocument) table.getDocument();

            }    

            EditingNotificationManager editingNotification = DALSwingLocator.getEditingNotificationManager();

            EditingNotification notification = editingNotification.notifyObservers(

                    this,

                    EditingNotification.BEFORE_EXIT_EDITING_STORE,

                    doc,

                    doc.getStore());

            if (notification.isCanceled()) {

                return;

            }

            stopEditing(table);

            ApplicationLocator.getManager().refreshMenusAndToolBars();

            editingNotification.notifyObservers(

                    this,

                    EditingNotification.AFTER_EXIT_EDITING_STORE,

                    doc,

                    doc.getStore());

        }

    }

    private void stopEditing(FeatureTableDocumentPanel table) {

        final FeatureStore featureStore = table.getFeatureStore();

        boolean allowWrite = featureStore.allowWrite();

        ApplicationManager application = ApplicationLocator.getManager();

        I18nManager i18n = ToolsLocator.getI18nManager();

        String message;

        Object[] options;

        if(allowWrite) {

            options = new Object[]{

                i18n.getTranslation("_Guardar"),

                "       " + i18n.getTranslation("_Descartar") + "       ",

                i18n.getTranslation("_Continuar")

            };

            message = i18n.getTranslation("stop_editing_message");

        } else {

            options = new Object[]{

                i18n.getTranslation("_Export"),

                "       " + i18n.getTranslation("_Descartar") + "       ",

                i18n.getTranslation("_Continuar")

            };

            message = i18n.getTranslation("export_and_stop_editing_message");

        }

        int resp = JOptionPane

                .showOptionDialog(

                        (Component) PluginServices.getMainFrame(),

                        message,

                        i18n.getTranslation("stop_edition"),

                        JOptionPane.YES_NO_CANCEL_OPTION,

                        JOptionPane.QUESTION_MESSAGE, null, options,

                        options[2]);

        switch( resp ) {

        case JOptionPane.NO_OPTION:

            try {

                // CANCEL EDITING

                featureStore.cancelEditing();

            } catch (DataException e) {

                LOGGER.warn("Problems canceling table editing: "

                    + e.getMessage(), e);

                application.messageDialog(

                    i18n.getTranslation("_Problems_cancel_table_editing")+"\n\n"+

                    i18n.getTranslation("_see_error_log_for_more_information"), 

                    null, 

                    i18n.getTranslation("_Warning"), 

                    JOptionPane.WARNING_MESSAGE, 

                    "Table_cant_cancelEditing"

                );

            }

            break;

        case JOptionPane.YES_OPTION:

            try {

                if(allowWrite){

                    // Save table

                    featureStore.finishEditing();

                } else {

                    exportTable(table.getDocument(), featureStore);

                }

            } catch (DataException e) {

                LOGGER.warn("Problems finish table editing: "

                    + e.getMessage(), e);

                application.messageDialog(

                    i18n.getTranslation("_Problems_finish_table_editing")+"\n\n"+

                    i18n.getTranslation("_see_error_log_for_more_information"), 

                    null, 

                    i18n.getTranslation("_Warning"), 

                    JOptionPane.WARNING_MESSAGE, 

                    "Table_cant_finishEditing"

                );

            }

            break;

        default:

            // This happens when user clicks on [x]

            // to abruptly close previous JOptionPane dialog

            // We do nothing (equivalent to 'Continue editing')

            break;

        }

    }

    private void exportTable(Document document, FeatureStore store) {

        ApplicationManager appManager = ApplicationLocator.getManager();

        I18nManager i18n = ToolsLocator.getI18nManager();

        ActionInfo action = PluginsLocator.getActionInfoManager().getAction("table-exportto");

        action.execute(new Object[]{document, WindowManager.MODE.toInteger(WindowManager.MODE.DIALOG)});

        String msg = "_Do_you_want_to_finish_editing_If_not_exported_the_data_the_changes_will_be_lost";

        if (appManager.confirmDialog(

                i18n.getTranslation(msg),

                i18n.getTranslation("end_editing"),

                JOptionPane.YES_NO_OPTION,

                JOptionPane.QUESTION_MESSAGE) == JOptionPane.YES_OPTION) {

            try {

                store.cancelEditing();

            } catch (Exception e) {

                LOGGER.warn("Can't abort editing of table '" + document.getName() + "'.", e);

                msg = i18n.getTranslation("_Problems_finish_table_editing")+"\n\n"+

                    i18n.getTranslation("_see_error_log_for_more_information");

                appManager.messageDialog(msg, "_Warning", JOptionPane.WARNING_MESSAGE);

            }

        }

    }

    public boolean isEnabled() {

        return true;

    }

    @Override

    public boolean isVisible() {

        IWindow v = PluginServices.getMDIManager().getActiveWindow();

        if (v == null) {

            return false;

        } else if (v instanceof FeatureTableDocumentPanel

                && ((FeatureTableDocumentPanel) v).getModel().getStore()

                .isEditing()) {

            table = (FeatureTableDocumentPanel) v;

            return true;

        } else {

            return false;

        }

    }

    /**

     * <p>

     * This class provides the status of extensions. If this extension has some

     * unsaved editing table (and save them), and methods to check if the

     * extension has some associated background tasks.

     *

     */

    private class StopEditingStatus implements IExtensionStatus {

        /**

         * This method is used to check if this extension has some unsaved

         * editing tables.

         *

         * @return true if the extension has some unsaved editing tables, false

         * otherwise.

         */

        public boolean hasUnsavedData() {

            Project project = ProjectManager.getInstance().getCurrentProject();

            List<Document> tables = project.getDocuments(TableManager.TYPENAME);

            for (int i = 0; i < tables.size(); i++) {

                FeatureStore store = ((TableDocument) tables.get(i)).getStore();

                if (store == null) {

                    continue;

                }

                if (store.isEditing()) {

                    return true;

                }

            }

            return false;

        }

        /**

         * This method is used to check if the extension has some associated

         * background process which is currently running.

         *

         * @return true if the extension has some associated background process,

         * false otherwise.

         */

        public boolean hasRunningProcesses() {

            return false;

        }

        /**

         * <p>

         * Gets an array of the traceable background tasks associated with this

         * extension. These tasks may be tracked, canceled, etc.

         * </p>

         *

         * @return An array of the associated background tasks, or null in case

         * there is no associated background tasks.

         */

        public IMonitorableTask[] getRunningProcesses() {

            return null;

        }

        /**

         * <p>

         * Gets an array of the UnsavedData objects, which contain information

         * about the unsaved editing tables and allows to save it.

         * </p>

         *

         * @return An array of the associated unsaved editing layers, or null in

         * case the extension has not unsaved editing tables.

         */

        public IUnsavedData[] getUnsavedData() {

            Project project = ProjectManager.getInstance().getCurrentProject();

            List<Document> tables = project.getDocuments(TableManager.TYPENAME);

            List<UnsavedTable> unsavedTables = new ArrayList<UnsavedTable>();

            for (int i = 0; i < tables.size(); i++) {

                TableDocument table = (TableDocument) tables.get(i);

                FeatureStore store = table.getStore();

                if (store == null) {

                    continue;

                }

                if (store.isEditing()) {

                    UnsavedTable ul

                            = new UnsavedTable(TableEditStopExtension.this);

                    ul.setTable(table);

                    unsavedTables.add(ul);

                }

            }

            return unsavedTables

                    .toArray(new IUnsavedData[unsavedTables.size()]);

        }

    }

    private class UnsavedTable extends UnsavedData {

        private TableDocument table;

        public UnsavedTable(IExtension extension) {

            super(extension);

        }

        public String getDescription() {

            return PluginServices.getText(this, "editing_table_unsaved");

        }

        public String getResourceName() {

            return table.getName();

        }

        public boolean saveData() {

            return executeSaveTable(table);

        }

        public void setTable(TableDocument table) {

            this.table = table;

        }

        @Override

        public String getIcon() {

            return "document-table-icon-small";

        }

    }

    // TODO Este codigo esta duplicado, tambien esta en la clase Table en el

    // metodo "public void stopEditing()"

    private boolean executeSaveTable(TableDocument table2) {

        FeatureStore fs = table2.getStore();

        if (fs.isEditing()) {

            try {

                fs.finishEditing();

            } catch (WriteException e) {

                NotificationManager.addError(PluginServices.getText(this, "error_saving_table"), e);

                return false;

            } catch (ReadException e) {

                NotificationManager.addError(PluginServices.getText(this, "error_saving_table"), e);

                return false;

            } catch (DataException e) {

                NotificationManager.addError(PluginServices.getText(this, "error_saving_table"), e);

                return false;

            }

        }

        return true;

    }

    @Override

    public IExtensionStatus getStatus() {

        return new StopEditingStatus();

    }

}