| gvSIG desktop 1

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

 * 

 */

package org.gvsig.app.extension;

import java.util.ArrayList;

import java.util.List;

import javax.swing.JOptionPane;

import org.gvsig.andami.IconThemeHelper;

import org.gvsig.andami.PluginServices;

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.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.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.utils.swing.threads.IMonitorableTask;

public class TableEditStopExtension extends AbstractTableEditExtension {

        public void initialize() {

                super.initialize();

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

        }

    public void execute(String actionCommand) {

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

            stopEditing(table);

        }

    }

    public void stopEditing(FeatureTableDocumentPanel table) {

        // FIXME

        int resp =

            JOptionPane.showConfirmDialog(null,

                PluginServices.getText(this, "realmente_desea_guardar") + " : "

                    + table.getModel().getName(), "Guardar",

                JOptionPane.YES_NO_OPTION);

        try {

            if (resp == JOptionPane.NO_OPTION) { // CANCEL EDITING

                table.getModel().getStore().cancelEditing();

            } else { // GUARDAMOS LA TABLA

                table.getModel().getStore().finishEditing();

            }

        } catch (DataException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

    }

    /**

     * @see org.gvsig.andami.plugins.IExtension#isEnabled()

     */

    public boolean isEnabled() {

        return true;

    }

    /**

     * @see org.gvsig.andami.plugins.IExtension#isVisible()

     */

    public boolean isVisible() {

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

        if (v == null) {

            return false;

        } else

            if (v instanceof FeatureTableDocumentPanel

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

                    .isEditing()

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

                    .getAssociatedLayer() == null) {

                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.

     * 

     * @author Vicente Caballero Navarro

     * 

     */

    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;

        }

    }

    // TODO Este c?digo est? duplicado, tambi?n est? en la clase Table en el

    // m?todo "public void stopEditing()"

    private boolean executeSaveTable(TableDocument table2) {

        FeatureStore fs = table2.getStore();

        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;

        }

        // if (ies instanceof IWriteable) {

        // IWriteable w = (IWriteable) ies;

        // IWriter writer = w.getWriter();

        // if (writer == null) {

        // return false;

        // }

        // try {

        // ITableDefinition tableDef = ies.getTableDefinition();

        // writer.initialize(tableDef);

        // ies.stopEdition(writer, EditionEvent.ALPHANUMERIC);

        // ies.getSelection().clear();

        // } catch (InitializeWriterException e) {

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

        // return false;

        // } catch (StopWriterVisitorException e) {

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

        // return false;

        // } catch (ReadDriverException e) {

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

        // return false;

        // }

        // }

        return true;

    }

    public IExtensionStatus getStatus() {

        return new StopEditingStatus();

    }

}