Application: gvSIG desktop

/* 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.awt.Component;

import java.io.File;

import java.text.MessageFormat;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import java.util.prefs.Preferences;

import javax.swing.JOptionPane;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.andami.IconThemeHelper;

import org.gvsig.andami.Launcher;

import org.gvsig.andami.Launcher.TerminationProcess;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.messages.NotificationManager;

import org.gvsig.andami.plugins.Extension;

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.andami.ui.mdiManager.WindowInfo;

import org.gvsig.andami.ui.wizard.UnsavedDataPanel;

import org.gvsig.app.project.Project;

import org.gvsig.app.project.ProjectManager;

import org.gvsig.app.project.documents.gui.ProjectWindow;

import org.gvsig.app.project.documents.view.ViewManager;

import org.gvsig.gui.beans.swing.JFileChooser;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.extensionpoint.ExtensionPointManager;

import org.gvsig.tools.persistence.exception.PersistenceException;

import org.gvsig.utils.GenericFileFilter;

import org.gvsig.utils.save.AfterSavingListener;

import org.gvsig.utils.save.BeforeSavingListener;

import org.gvsig.utils.save.SaveEvent;

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

/**

 * Extension que proporciona controles para crear proyectos nuevos, abrirlos y

 * guardarlos. Adem?s los tipos de tabla que soporta el proyecto son a?adidos

 * en esta clase.

 *

 * @author Fernando Gonz?lez Cort?s

 */

public class ProjectExtension extends Extension implements IExtensionStatus {

    private static final Logger LOG =

        LoggerFactory.getLogger(ProjectExtension.class);

    private static String projectPath = null;

    private ProjectWindow projectFrame;

    private Project p;

    private String lastSavePath;

    private WindowInfo seedProjectWindow;

    public static final String PROJECT_FILE_CHOOSER_ID =

        "PROJECT_FILECHOOSER_ID";

    /**

     * Use UTF-8 for encoding, as it can represent characters from

     * any language.

     *

     * Another sensible option would be

     * encoding = System.getProperty("file.encoding");

     * but this would need some extra testing.

     *

     * @deprecated see PersistentManager

     */

    public static String PROJECTENCODING = "UTF-8";

    private List<BeforeSavingListener> beforeSavingListeners =

        new ArrayList<BeforeSavingListener>();

    private List<AfterSavingListener> afterSavingListeners =

        new ArrayList<AfterSavingListener>();

    /**

     * @see com.iver.mdiApp.plugins.IExtension#initialize()

     */

    public void initialize() {

        try {

            Class.forName("javax.media.jai.EnumeratedParameter");

        } catch (ClassNotFoundException e) {

            NotificationManager

                .addError(

                    "La m?quina virtual con la que se ejecuta gvSIG no tiene JAI instalado",

                    e);

        }

        initializeDocumentActionsExtensionPoint();

        registerIcons();

    }

    private void registerIcons() {

            IconThemeHelper.registerIcon("action", "application-project-new", this);

            IconThemeHelper.registerIcon("action", "application-project-open", this);

            IconThemeHelper.registerIcon("action", "application-project-save", this);

            IconThemeHelper.registerIcon("action", "application-project-save-as", this);

            IconThemeHelper.registerIcon("action", "application-exit", this);

        IconThemeHelper.registerIcon("project", "project-icon", this);

    }

    private void loadInitialProject() {

        String[] theArgs = PluginServices.getArguments();

        String lastArg = theArgs[theArgs.length - 1];

        if ((lastArg.toLowerCase().endsWith(Project.FILE_EXTENSION

            .toLowerCase()))) {

            PluginServices.getLogger().debug(

                "Intentando cargar el proyecto " + lastArg);

            // File projectFile = new File(lastArg);

            setProject(readProject(lastArg));

            PluginServices.getMainFrame().setTitle(p.getName());

            projectPath = lastArg;

        } else {

            setProject(ProjectManager.getInstance().createProject());

            p.setName(PluginServices.getText(this, "untitled"));

            p.setModified(false);

            PluginServices.getMainFrame().setTitle(

                PluginServices.getText(this, "sin_titulo"));

        }

    }

    /**

     * @see com.iver.mdiApp.plugins.IExtension#postInitialize()

     */

    public void postInitialize() {

        registerDocuments();

        loadInitialProject();

        getProjectFrame().setProject(p);

        showProjectWindow();

    }

    public ProjectWindow getProjectFrame() {

        if (projectFrame == null) {

            projectFrame = new ProjectWindow();

        }

        return projectFrame;

    }

    /**

     * Muestra la ventana con el gestor de proyectos.

     */

    public void showProjectWindow() {

        if (seedProjectWindow != null) {

            if (seedProjectWindow.isClosed()) {

                // if it was closed, we just don't open the window now

                seedProjectWindow.setClosed(false);

                return;

            }

            WindowInfo winProps = seedProjectWindow;

            seedProjectWindow = null;

            PluginServices.getMDIManager().addWindow(getProjectFrame());

            PluginServices.getMDIManager().changeWindowInfo(getProjectFrame(),

                winProps);

        } else {

            PluginServices.getMDIManager().addWindow(getProjectFrame());

        }

    }

    /**

     * Muestra la ventana con el gestor de proyectos, con las propiedades

     * de ventana especificadas.

     */

    public void showProjectWindow(WindowInfo wi) {

        seedProjectWindow = wi;

        showProjectWindow();

    }

    /**

     * Guarda el proyecto actual en disco.

     */

    private boolean guardar() {

        boolean saved = false;

        // if (p.getPath() == null) {

        if (projectPath == null) {

            saved = guardarDialogo();

        } else {

            long t1, t2;

            t1 = System.currentTimeMillis();

            saved = writeProject(new File(projectPath), p, false);

            t2 = System.currentTimeMillis();

            PluginServices.getLogger().info(

                "Project saved. " + (t2 - t1) + " miliseconds");

            getProjectFrame().refreshControls();

        }

        return saved;

    }

    private boolean guardarDialogo() {

        boolean saved = false;

        if (lastSavePath == null) {

            lastSavePath = projectPath;

        }

        Preferences prefs = Preferences.userRoot().node("gvsig.foldering");

        JFileChooser jfc =

            new JFileChooser(PROJECT_FILE_CHOOSER_ID, prefs.get(

                "ProjectsFolder", null));

        jfc.setDialogTitle(PluginServices.getText(this, "guardar_proyecto"));

        jfc.addChoosableFileFilter(new GenericFileFilter(

            Project.FILE_EXTENSION, MessageFormat.format(

                PluginServices.getText(this, "_gvSIG_file_project_({0})"),

                Project.FILE_EXTENSION)));

        if (jfc.showSaveDialog((Component) PluginServices.getMainFrame()) == JFileChooser.APPROVE_OPTION) {

            File file = jfc.getSelectedFile();

            if (!(file.getPath().toLowerCase().endsWith(Project.FILE_EXTENSION

                .toLowerCase()))) {

                file = new File(file.getPath() + Project.FILE_EXTENSION);

            }

            saved = writeProject(file, p);

            String filePath = file.getAbsolutePath();

            lastSavePath =

                filePath.substring(0, filePath.lastIndexOf(File.separatorChar));

            getProjectFrame().refreshControls();

        }

        return saved;

    }

    /**

     * Checks whether the project and related unsaved data is modified,

     * and allows the user to save it.

     *

     * @return true if the data has been correctly saved, false otherwise

     */

    private boolean askSave() {

        if (p != null && p.hasChanged()) {

            TerminationProcess process = Launcher.getTerminationProcess();

            UnsavedDataPanel panel = process.getUnsavedDataPanel();

            panel.setHeaderText(PluginServices.getText(this,

                "Select_resources_to_save_before_closing_current_project"));

            panel.setAcceptText(PluginServices.getText(this, "save_resources"),

                PluginServices.getText(this,

                    "Save_the_selected_resources_and_close_current_project"));

            panel.setCancelText(PluginServices.getText(this, "Dont_close"),

                PluginServices.getText(this, "Return_to_current_project"));

            int closeCurrProj;

            try {

                closeCurrProj = process.manageUnsavedData();

                if (closeCurrProj == JOptionPane.NO_OPTION) {

                    // the user chose to return to current project

                    return false;

                }

            } catch (Exception e) {

                LOG.error("Some data can not be saved", e);

            }

        }

        return true;

    }

    /**

     * @see com.iver.mdiApp.plugins.IExtension#updateUI(java.lang.String)

     */

    public void execute(String actionCommand) {

        if (actionCommand.equals("application-project-new")) {

            if (!askSave()) {

                return;

            }

            projectPath = null;

            // ProjectDocument.initializeNUMS();

            PluginServices.getMDIManager().closeAllWindows();

            setProject(ProjectManager.getInstance().createProject());

            getProjectFrame().setProject(p);

            showProjectWindow();

            PluginServices.getMainFrame().setTitle(

                PluginServices.getText(this, "sin_titulo"));

        } else

            if (actionCommand.equals("application-project-open")) {

                if (!askSave()) {

                    return;

                }

                Preferences prefs =

                    Preferences.userRoot().node("gvsig.foldering");

                JFileChooser jfc =

                    new JFileChooser(PROJECT_FILE_CHOOSER_ID, prefs.get(

                        "ProjectsFolder", null));

                jfc.addChoosableFileFilter(new GenericFileFilter(

                    Project.FILE_EXTENSION, PluginServices.getText(this,

                        "tipo_fichero_proyecto")));

                if (jfc.showOpenDialog((Component) PluginServices

                    .getMainFrame()) == JFileChooser.APPROVE_OPTION) {

                    // ProjectDocument.initializeNUMS();

                    PluginServices.getMDIManager().closeAllWindows();

                    File projectFile = jfc.getSelectedFile();

                    Project o = readProject(projectFile);

                    setPath(projectFile.getAbsolutePath());

                    // lastPath = getPath();

                    if (o != null) {

                        setProject(o);

                    }

                    getProjectFrame().setProject(p);

                    PluginServices.getMainFrame().setTitle(p.getName());

                    getProjectFrame().refreshControls();

                    // p.restoreWindowProperties();

                }

            } else

                if (actionCommand.equals("application-project-save")) {

                    guardar();

                } else

                    if (actionCommand.equals("application-project-save-as")) {

                        guardarDialogo();

                    } else

                        if (actionCommand.equals("application-exit")) {

                            Launcher.closeApplication();

                        }

    }

    /**

     * Escribe el proyecto en XML.

     *

     * @param file

     *            Fichero.

     * @param p

     *            Proyecto.

     */

    public boolean writeProject(File file, Project p) {

        return writeProject(file, p, true);

    }

    /**

     * Escribe el proyecto en XML. Pero permite decidir si se

     * pide confirmaci?n para sobreescribir

     *

     * @param file

     *            Fichero.

     * @param p

     *            Proyecto.

     * @param askConfirmation

     *            boolean

     */

    public boolean writeProject(File file, Project p, boolean askConfirmation) {

        if (askConfirmation && file.exists()) {

            int resp =

                JOptionPane.showConfirmDialog((Component) PluginServices

                    .getMainFrame(), PluginServices.getText(this,

                    "fichero_ya_existe_seguro_desea_guardarlo"), PluginServices

                    .getText(this, "guardar"), JOptionPane.YES_NO_OPTION);

            if (resp != JOptionPane.YES_OPTION) {

                return false;

            }

        }

        NotificationManager.addInfo(PluginServices.getText(this,

            "writinng_project") + ": " + file.getName());

        // write it out as XML

        try {

            fireBeforeSavingFileEvent(new SaveEvent(this,

                SaveEvent.BEFORE_SAVING, file));

            p.saveState(file);

            fireAfterSavingFileEvent(new SaveEvent(this,

                SaveEvent.AFTER_SAVING, file));

            PluginServices.getMainFrame().setTitle(p.getName());

            setPath(file.toString());

        } catch (PersistenceException e) {

            String messagestack = e.getLocalizedMessageStack();

            NotificationManager.addError(

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

                    + file.getName() + "\n" + messagestack, e);

            return false;

        } catch (Exception e) {

            NotificationManager.addError(

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

                    + file.getName(), e);

            return false;

        }

        NotificationManager.addInfo(PluginServices.getText(this,

            "wrote_project") + ": " + file.getName());

        return true;

    }

    public Project readProject(String path) {

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

        project.loadState(new File(path));

        return (Project) project;

    }

    /**

     * Lee del XML el proyecto.<br>

     * <br>

     *

     * Reads the XML of the project.<br>

     * It returns a project object holding all needed info that is not linked to

     * the Project Dialog. <br>

     * In case you want the project to be

     * linked to the window you must set this object to the extension:<br>

     *

     * <b>Example:</b><br>

     *

     * ...<br>

     * ...<br>

     * Project p = ProjectExtension.readProject(projectFile);<br>

     * ProjectExtension.setProject(p);

     * ...<br>

     * ...<br>

     *

     * @param file

     *            Fichero.

     *

     * @return Project

     *

     */

    public Project readProject(File file) {

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

        project.loadState(file);

        return (Project) project;

    }

    /**

     * Devuelve el proyecto.

     *

     * @return Proyecto.

     */

    public Project getProject() {

        return p;

    }

    /**

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

     */

    public boolean isEnabled() {

        return true;

    }

    /**

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

     */

    public boolean isVisible() {

        return true;

    }

    /**

     * Sets the project

     *

     * @param p

     */

    public void setProject(Project p) {

        getProjectFrame().setProject(p);

        this.p = p;

    }

    private void registerDocuments() {

        ViewManager.register();

    }

    private void initializeDocumentActionsExtensionPoint() {

        ExtensionPointManager epMan = ToolsLocator.getExtensionPointManager();

        epMan.add("DocumentActions_View",

            "Context menu options of the view document list"

                + " in the project window " + "(register instances of "

                + "org.gvsig.app.project.AbstractDocumentContextMenuAction)");

    }

    public static String getPath() {

        return projectPath;

    }

    public static void setPath(String path) {

        projectPath = path;

    }

    public IWindow getProjectWindow() {

        return getProjectFrame();

    }

    public IExtensionStatus getStatus() {

        return this;

    }

    public boolean hasUnsavedData() {

        return p.hasChanged();

    }

    public IUnsavedData[] getUnsavedData() {

        if (hasUnsavedData()) {

            UnsavedProject data = new UnsavedProject(this);

            IUnsavedData[] dataArray = { data };

            return dataArray;

        } else {

            return null;

        }

    }

    /**

     * Implements the IUnsavedData interface to show unsaved projects

     * in the Unsavad Data dialog.

     *

     * @author Cesar Martinez Izquierdo <cesar.martinez@iver.es>

     */

    public class UnsavedProject extends UnsavedData {

        public UnsavedProject(IExtension extension) {

            super(extension);

        }

        public String getDescription() {

            if (getPath() == null) {

                return PluginServices.getText(ProjectExtension.this,

                    "Unnamed_new_gvsig_project_");

            } else {

                return PluginServices.getText(ProjectExtension.this,

                    "Modified_project_");

            }

        }

        public String getResourceName() {

            if (getPath() == null) {

                return PluginServices.getText(ProjectExtension.this, "Unnamed");

            } else {

                return getPath();

            }

        }

        public boolean saveData() {

            return guardar();

        }

        public String getIcon() {

            return "project-icon";

        }

    }

    public IMonitorableTask[] getRunningProcesses() {

        // TODO Auto-generated method stub

        return null;

    }

    public boolean hasRunningProcesses() {

        // TODO Auto-generated method stub

        return false;

    }

    /**

     * Adds the specified before saving listener to receive

     * "before saving file events" from

     * this component.

     * If l is null, no exception is thrown and no action is performed.

     *

     * @author Pablo Piqueras Bartolom? <pablo.piqueras@iver.es>

     *

     * @param l

     *            the before saving listener.

     * @see SaveEvent

     * @see BeforeSavingListener

     * @see #removeListener(BeforeSavingListener)

     * @see #getBeforeSavingListeners

     */

    public synchronized void addListener(BeforeSavingListener l) {

        if (l == null) {

            return;

        }

        if (!this.beforeSavingListeners.contains(l)) {

            this.beforeSavingListeners.add(l);

        }

    }

    /**

     * Adds the specified after saving listener to receive

     * "after saving file events" from

     * this component.

     * If l is null, no exception is thrown and no action is performed.

     *

     * @author Pablo Piqueras Bartolom? <pablo.piqueras@iver.es>

     *

     * @param l

     *            the after saving listener.

     * @see SaveEvent

     * @see AfterSavingListener

     * @see #removeListener(AfterSavingListener)

     * @see #getAfterSavingListeners()

     */

    public synchronized void addListener(AfterSavingListener l) {

        if (l == null) {

            return;

        }

        if (!this.afterSavingListeners.contains(l)) {

            this.afterSavingListeners.add(l);

        }

    }

    /**

     * Returns an array of all the before saving listeners

     * registered on this component.

     *

     * @author Pablo Piqueras Bartolom? <pablo.piqueras@iver.es>

     *

     * @return all of this component's <code>BeforeSavingListener</code>s

     *         or an empty array if no key

     *         listeners are currently registered

     *

     * @see #addBeforeSavingListener(BeforeSavingListener)

     * @see #removeBeforeSavingListener(BeforeSavingListener)

     */

    public synchronized BeforeSavingListener[] getBeforeSavingListeners() {

        return this.beforeSavingListeners

            .toArray(new BeforeSavingListener[] {});

    }

    /**

     * Returns an array of all the after saving listeners

     * registered on this component.

     *

     * @author Pablo Piqueras Bartolom? <pablo.piqueras@iver.es>

     *

     * @return all of this component's <code>AfterSavingListener</code>s

     *         or an empty array if no key

     *         listeners are currently registered

     *

     * @see #addAfterSavingListener(AfterSavingListener)

     * @see #removeAfterSavingListener

     */

    public synchronized AfterSavingListener[] getAfterSavingListeners() {

        return this.afterSavingListeners.toArray(new AfterSavingListener[] {});

    }

    /**

     * Removes the specified before saving listener so that it no longer

     * receives save file events from this component. This method performs

     * no function, nor does it throw an exception, if the listener

     * specified by the argument was not previously added to this component.

     * If listener <code>l</code> is <code>null</code>,

     * no exception is thrown and no action is performed.

     *

     * @author Pablo Piqueras Bartolom? <pablo.piqueras@iver.es>

     *

     * @param l

     *            the before saving listener

     * @see SaveEvent

     * @see BeforeSavingListener

     * @see #addListener(BeforeSavingListener)

     * @see #getBeforeSavingListeners()

     */

    public synchronized void removeListener(BeforeSavingListener l) {

        if (l == null) {

            return;

        }

        this.beforeSavingListeners.remove(l);

    }

    /**

     * Removes the specified after saving listener so that it no longer

     * receives save file events from this component. This method performs

     * no function, nor does it throw an exception, if the listener

     * specified by the argument was not previously added to this component.

     * If listener <code>l</code> is <code>null</code>,

     * no exception is thrown and no action is performed.

     *

     * @author Pablo Piqueras Bartolom? <pablo.piqueras@iver.es>

     *

     * @param l

     *            the after saving listener

     * @see SaveEvent

     * @see AfterSavingListener

     * @see #addListener(AfterSavingListener)

     * @see #getAfterSavingListeners()

     */

    public synchronized void removeListener(AfterSavingListener l) {

        if (l == null) {

            return;

        }

        this.afterSavingListeners.remove(l);

    }

    /**

     * Reports a before saving file event.

     *

     * @author Pablo Piqueras Bartolom? <pablo.piqueras@iver.es>

     *

     * @param evt

     *            the before saving file event

     */

    protected void fireBeforeSavingFileEvent(SaveEvent evt) {

        if ((evt.getID() != SaveEvent.BEFORE_SAVING) || (evt.getFile() == null)) {

            return;

        }

        Iterator<BeforeSavingListener> iter =

            this.beforeSavingListeners.iterator();

        while (iter.hasNext()) {

            iter.next().beforeSaving(evt);

        }

    }

    /**

     * Reports a after saving file event.

     *

     * @author Pablo Piqueras Bartolom? <pablo.piqueras@iver.es>

     *

     * @param evt

     *            the after saving file event

     */

    protected void fireAfterSavingFileEvent(SaveEvent evt) {

        if ((evt.getID() != SaveEvent.AFTER_SAVING) || (evt.getFile() == null)) {

            return;

        }

        Iterator<AfterSavingListener> iter =

            this.afterSavingListeners.iterator();

        while (iter.hasNext()) {

            iter.next().afterSaving(evt);

        }

    }

}