Application: gvSIG desktop

import java.util.ArrayList;

import com.hardcode.driverManager.DriverLoadException;

import com.hardcode.gdbms.engine.data.driver.DriverException;

import com.iver.andami.messages.NotificationManager;

import com.iver.andami.plugins.IExtension;

import com.iver.andami.plugins.status.IExtensionStatus;

import com.iver.andami.plugins.status.IUnsavedData;

import com.iver.andami.plugins.status.UnsavedData;

import com.iver.cit.gvsig.fmap.drivers.ITableDefinition;

import com.iver.cit.gvsig.fmap.edition.EditionEvent;

import com.iver.cit.gvsig.fmap.edition.EditionException;

import com.iver.cit.gvsig.fmap.edition.IEditableSource;

import com.iver.cit.gvsig.fmap.edition.IWriteable;

import com.iver.cit.gvsig.fmap.edition.IWriter;

import com.iver.cit.gvsig.project.documents.table.ProjectTable;

import com.iver.cit.gvsig.project.documents.table.ProjectTableFactory;

import com.iver.utiles.swing.threads.IMonitorableTask;

    /**

	 * <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() {

			ProjectExtension pe=(ProjectExtension)PluginServices.getExtension(ProjectExtension.class);

			ProjectTable[] tables=(ProjectTable[])pe.getProject().getDocumentsByType(ProjectTableFactory.registerName).toArray(new ProjectTable[0]);

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

				if (tables[i].getModelo().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() {

			ProjectExtension pe=(ProjectExtension)PluginServices.getExtension(ProjectExtension.class);

			ProjectTable[] tables =(ProjectTable[])pe.getProject().getDocumentsByType(ProjectTableFactory.registerName).toArray(new ProjectTable[0]);

			ArrayList unsavedTables = new ArrayList();

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

				if (tables[i].getModelo().isEditing()) {

					UnsavedTable ul=new UnsavedTable(TableEditStopExtension.this);

					ul.setTable(tables[i]);

					unsavedTables.add(ul);

				}

			}

			return (IUnsavedData[])unsavedTables.toArray(new IUnsavedData[0]);

		}

	}

	private class UnsavedTable extends UnsavedData{

		private ProjectTable 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(ProjectTable 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(ProjectTable table2) {

		IEditableSource ies = table2.getModelo();

		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 (DriverLoadException e) {

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

				return false;

			} catch (EditionException e) {

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

				return false;

			} catch (DriverException e) {

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

				return false;

			}

		}

		return true;

	}

	public IExtensionStatus getStatus() {

		return new StopEditingStatus();

	}