EDIT

/**

 * Copyright (C) 2007 EDIT

 * European Distributed Institute of Taxonomy

 * http://www.e-taxonomy.eu

 *

 * The contents of this file are subject to the Mozilla Public License Version 1.1

 * See LICENSE.TXT at the top of this package for the full license terms.

 */

package eu.etaxonomy.taxeditor.navigation.navigator;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.Comparator;

import java.util.HashMap;

import java.util.HashSet;

import java.util.List;

import java.util.Map;

import java.util.Observable;

import java.util.Observer;

import java.util.Set;

import java.util.UUID;

import org.eclipse.core.runtime.IAdaptable;

import org.eclipse.core.runtime.IProgressMonitor;

import org.eclipse.jface.viewers.DoubleClickEvent;

import org.eclipse.jface.viewers.TreePath;

import org.eclipse.ui.IMemento;

import org.eclipse.ui.IViewSite;

import org.eclipse.ui.PartInitException;

import org.eclipse.ui.navigator.CommonNavigator;

import eu.etaxonomy.cdm.api.application.CdmApplicationState;

import eu.etaxonomy.cdm.api.application.CdmChangeEvent;

import eu.etaxonomy.cdm.api.application.ICdmChangeListener;

import eu.etaxonomy.cdm.api.conversation.ConversationHolder;

import eu.etaxonomy.cdm.api.conversation.IConversationEnabled;

import eu.etaxonomy.cdm.api.service.IClassificationService;

import eu.etaxonomy.cdm.model.common.CdmBase;

import eu.etaxonomy.cdm.model.taxon.Classification;

import eu.etaxonomy.cdm.model.taxon.TaxonNaturalComparator;

import eu.etaxonomy.cdm.model.taxon.TaxonNode;

import eu.etaxonomy.cdm.model.taxon.TaxonNodeByNameComparator;

import eu.etaxonomy.cdm.model.taxon.TaxonNodeByRankAndNameComparator;

import eu.etaxonomy.cdm.persistence.dto.TaxonNodeDto;

import eu.etaxonomy.cdm.persistence.hibernate.CdmDataChangeMap;

import eu.etaxonomy.taxeditor.model.DataChangeBridge;

import eu.etaxonomy.taxeditor.model.IDataChangeBehavior;

import eu.etaxonomy.taxeditor.model.MessagingUtils;

import eu.etaxonomy.taxeditor.navigation.NavigationUtil;

import eu.etaxonomy.taxeditor.navigation.l10n.Messages;

import eu.etaxonomy.taxeditor.operation.IPostOperationEnabled;

import eu.etaxonomy.taxeditor.preference.PreferencesUtil;

import eu.etaxonomy.taxeditor.session.ICdmEntitySession;

import eu.etaxonomy.taxeditor.session.ICdmEntitySessionEnabled;

import eu.etaxonomy.taxeditor.store.CdmStore;

import eu.etaxonomy.taxeditor.store.LoginManager;

/**

 * Taxonomic tree implementation using Common Navigator Framework.

 *

 * @author p.ciardelli

 * @author n.hoffmann

 * @created 02.06.2009

 * @version 1.0+-

 */

public class TaxonNavigator extends CommonNavigator implements

		IPostOperationEnabled, IConversationEnabled, Observer,

		ICdmEntitySessionEnabled, ICdmChangeListener {

    private static final String RESTORING_TAXON_NAVIGATOR = Messages.TaxonNavigator_RESTORE;

    /**

	 * Constant

	 * <code>ID="eu.etaxonomy.taxeditor.navigation.navig"{trunked}</code>

	 */

	public static final String ID = "eu.etaxonomy.taxeditor.navigation.navigator"; //$NON-NLS-1$

	private static final String TREE_PATH = "treepath"; //$NON-NLS-1$

	private static final String TREE_PATHS = "treepaths"; //$NON-NLS-1$

	private ConversationHolder conversation;

	private ICdmEntitySession cdmEntitySession;

	private String partNameCache;

	private IDataChangeBehavior dataChangeBehavior;

	private Root root;

	/*

	 * (non-Javadoc)

	 *

	 * @see org.eclipse.ui.navigator.CommonNavigator#getInitialInput()

	 */

	/** {@inheritDoc} */

	@Override

	protected IAdaptable getInitialInput() {

		Comparator<TaxonNode> comparator;

		if (PreferencesUtil.getSortNodesNaturally()){

			comparator = new TaxonNaturalComparator();

		} else if (PreferencesUtil.getSortNodesStrictlyAlphabetically()){

			comparator = new TaxonNodeByNameComparator();

		}else {

			comparator = new TaxonNodeByRankAndNameComparator();

		}

//		TaxonNodeNavigatorComparator viewerComparator = new TaxonNodeNavigatorComparator(comparator);

//		this.getCommonViewer().setComparator(viewerComparator);

//		//this.getCommonViewer().setAutoExpandLevel(2);

        setLinkingEnabled(true);

//		this.getCommonViewer().addSelectionChangedListener(new ISelectionChangedListener() {

//

//			@Override

//			public void selectionChanged(SelectionChangedEvent arg0) {

//				IStructuredSelection selection = (IStructuredSelection) getCommonViewer().getSelection();

//

//				Object firstElement =  selection.getFirstElement();

//				//

//				if (!(firstElement instanceof Classification)){

//					//NavigationUtil.selectInNavigator(firstElement, null);

//					NavigationUtil.openEditor(firstElement);

//				}

//

//			}

//		} );

		if (CdmStore.isActive()) {

			// TODO when closing and reopening the taxon navigator

			// we do not preserve state. Closing the view, in contrary to

			// closing the whole application

			// should be handled by the state manager too

		    root = new Root(conversation);

			return root;

		}

		return new EmptyRoot();

	}

	/** {@inheritDoc} */

	@Override

	public void init(IViewSite site) throws PartInitException {

		super.init(site);

		init();

	}

	/**

	 * <p>

	 * init

	 * </p>

	 */

	public void init() {

		if (CdmStore.isActive() && conversation == null) {

			conversation = CdmStore.createConversation();

			conversation.registerForDataStoreChanges(TaxonNavigator.this);

		}

		if (CdmStore.isActive()) {

		    cdmEntitySession = CdmStore.getCurrentSessionManager().newSession(this, true);

		    CdmApplicationState.getCurrentDataChangeService().register(this);

		}

		CdmStore.getLoginManager().addObserver(this);

	}

	/**

	 * Refresh this navigators viewer

	 */

	public void refresh() {

		if(getConversationHolder() != null){

			getConversationHolder().bind();

			//FIXME : Need to make sure this is a stable fix (ticket 3822)

			if(!getConversationHolder().isCompleted()){

			    getConversationHolder().commit();

			}

		}

		if(!getCommonViewer().getTree().isDisposed()){

		    getCommonViewer().refresh();

		}

	}

	   /**

     * Refresh this navigators viewer

     */

    public void refresh(Set<?> objects) {

        for(Object obj : objects) {

            getCommonViewer().refresh(obj);

        }

    }

	/**

	 * Removes all content

	 */

	public void clear() {

		getCommonViewer().setInput(new EmptyRoot());

	}

	/**

	 * <p>

	 * restore

	 * </p>

	 *

	 * @param memento

	 *            a {@link org.eclipse.ui.IMemento} object.

	 * @param monitor

	 *            a {@link org.eclipse.core.runtime.IProgressMonitor} object.

	 */

	public void restore(IMemento memento, IProgressMonitor monitor) {

	    root = new Root(conversation);

		if (memento == null) {

			getCommonViewer().setInput(root);

			return;

		}

		int mementoWork = 0;

		Set<TreePath> treePaths = new HashSet<TreePath>();

		IMemento[] treePathMementos = null;

		IMemento treePathsMemento = memento.getChild(TREE_PATHS);

		if (treePathsMemento != null) {

			treePathMementos = treePathsMemento.getChildren(TREE_PATH);

			mementoWork = treePathMementos.length;

		}

		// begin the monitor with steps for all tree paths and steps for

		// creating

		// conversation s.o., refreshing the tree and setting the paths

		IProgressMonitor subProgressMonitor = NavigationUtil

				.getSubProgressMonitor(monitor, 1);

		subProgressMonitor.beginTask(RESTORING_TAXON_NAVIGATOR,

				1 + mementoWork + 5);

		subProgressMonitor.subTask(RESTORING_TAXON_NAVIGATOR);

		subProgressMonitor.worked(1);

		conversation = CdmStore.createConversation();

		subProgressMonitor.worked(1);

		conversation.registerForDataStoreChanges(TaxonNavigator.this);

		subProgressMonitor.worked(1);

		getCommonViewer().setInput(root);

		subProgressMonitor.worked(1);

		getCommonViewer().refresh();

		subProgressMonitor.worked(1);

		if (treePathMementos != null && treePathMementos.length > 0) {

			for (IMemento treePathMemento : treePathMementos) {

				TreePath treePath = createTreePathFromString(treePathMemento

						.getID());

				if (!subProgressMonitor.isCanceled() && treePath != null) {

					treePaths.add(treePath);

					subProgressMonitor.worked(1);

				}

			}

		}

		if (treePaths.size() > 0) {

			TaxonNavigator.this.getCommonViewer().setExpandedTreePaths(

					treePaths.toArray(new TreePath[0]));

			subProgressMonitor.worked(1);

		}

		subProgressMonitor.done();

	}

	/**

	 * @param string

	 * @return

	 */

	private TreePath createTreePathFromString(String string) {

		List<CdmBase> pathList = new ArrayList<CdmBase>();

		if (string.length() == 0) {

            return null;

        }

		for (String uuid : string.split(" ")) { //$NON-NLS-1$

			CdmBase cdmBaseObject = CdmStore.getService(

					IClassificationService.class).getTaxonNodeByUuid(

					UUID.fromString(uuid));

			if (cdmBaseObject == null) {

				// is this a tree uuid?

				cdmBaseObject = CdmStore.getService(

						IClassificationService.class).load(

						UUID.fromString(uuid));

				if (cdmBaseObject == null) {

                    return null;

                }

			}

			pathList.add(cdmBaseObject);

		}

		return new TreePath(pathList.toArray());

	}

	/** {@inheritDoc} */

	@Override

	public void saveState(IMemento aMemento) {

		//

	}

	/**

	 * <p>

	 * saveTreeState

	 * </p>

	 *

	 * @param memento

	 *            a {@link org.eclipse.ui.IMemento} object.

	 * @param progressMonitor

	 *            a {@link org.eclipse.core.runtime.IProgressMonitor} object.

	 */

	public void saveTreeState(IMemento memento, IProgressMonitor progressMonitor) {

		if (memento == null) {

			return;

		}

		IProgressMonitor monitor = NavigationUtil.getSubProgressMonitor(

				progressMonitor, 1);

		super.saveState(memento);

		memento = memento.createChild(TREE_PATHS);

		TreePath[] treePaths = this.getCommonViewer().getExpandedTreePaths();

		monitor.beginTask(Messages.TaxonNavigator_SAVE_TASK, treePaths.length);

		for (TreePath treePath : treePaths) {

			int pathLength = treePath.getSegmentCount();

			String path = ""; //$NON-NLS-1$

			for (int i = 0; i < pathLength; i++) {

				Object segment = treePath.getSegment(i);

				if (segment instanceof CdmBase) {

					path += ((CdmBase) segment).getUuid().toString() + " "; //$NON-NLS-1$

					monitor.worked(1);

				} else {

					MessagingUtils.warn(getClass(),

							"Non-taxon tree path segment " + segment); //$NON-NLS-1$

				}

			}

			memento.createChild(TREE_PATH, path.trim());

		}

		monitor.done();

	}

	/*

	 * (non-Javadoc)

	 *

	 * @see

	 * eu.etaxonomy.cdm.api.conversation.IConversationEnabled#getConversationHolder

	 * ()

	 */

	/**

	 * <p>

	 * getConversationHolder

	 * </p>

	 *

	 * @return a {@link eu.etaxonomy.cdm.api.conversation.ConversationHolder}

	 *         object.

	 */

	@Override

	public ConversationHolder getConversationHolder() {

		return conversation;

	}

	/*

	 * (non-Javadoc)

	 *

	 * @see

	 * eu.etaxonomy.cdm.persistence.hibernate.ICdmPostDataChangeObserver#update

	 * (eu.etaxonomy.cdm.persistence.hibernate.CdmDataChangeMap)

	 */

	/** {@inheritDoc} */

	@Override

	public void update(CdmDataChangeMap changeEvents) {

		if (dataChangeBehavior == null) {

			dataChangeBehavior = new TaxonNavigatorDataChangeBehavior(this);

		}

		DataChangeBridge.handleDataChange(changeEvents, dataChangeBehavior);

	}

	/** {@inheritDoc} */

	@Override

	public String getFrameToolTipText(Object element) {

		if (element instanceof Root) {

			return Messages.TaxonNavigator_TAXON_TREE;

		}

		return super.getFrameToolTipText(element);

	}

	/*

	 * (non-Javadoc)

	 *

	 * @see org.eclipse.ui.part.WorkbenchPart#dispose()

	 */

	/** {@inheritDoc} */

	@Override

	public void dispose() {

		super.dispose();

		dataChangeBehavior = null;

		if (conversation != null) {

			conversation.unregisterForDataStoreChanges(this);

		}

		if(cdmEntitySession != null) {

		    cdmEntitySession.dispose();

		}

		if(CdmApplicationState.getCurrentDataChangeService() != null) {

		    CdmApplicationState.getCurrentDataChangeService().unregister(this);

		}

	}

	/*

	 * (non-Javadoc)

	 *

	 * @see org.eclipse.ui.navigator.CommonNavigator#setFocus()

	 */

	/** {@inheritDoc} */

	@Override

	public void setFocus() {

		// logger.warn("Setting focus to navigator");

		super.setFocus();

		if (getConversationHolder() != null) {

			getConversationHolder().bind();

		}

		if(cdmEntitySession != null) {

		    cdmEntitySession.bind();

		}

	}

	/*

	 * (non-Javadoc)

	 *

	 * @see

	 * eu.etaxonomy.taxeditor.operations.IPostOperationEnabled#postOperation

	 * (eu.etaxonomy.cdm.model.common.CdmBase)

	 */

	/** {@inheritDoc} */

	@Override

	public boolean postOperation(CdmBase objectAffectedByOperation) {

		// nothing to do here

		return true;

	}

	/**

	 * <p>

	 * save

	 * </p>

	 *

	 * @param memento

	 *            a {@link org.eclipse.ui.IMemento} object.

	 * @param monitor

	 *            a {@link org.eclipse.core.runtime.IProgressMonitor} object.

	 */

	public void save(IMemento memento, IProgressMonitor monitor) {

		saveTreeState(memento, monitor);

		if (conversation != null) {

			conversation.unregisterForDataStoreChanges(this);

			conversation = null;

		}

	}

	/** {@inheritDoc} */

	@Override

	protected void handleDoubleClick(DoubleClickEvent event) {

//		ISelection selection = event.getSelection();

//		if(selection instanceof IStructuredSelection){

//			Object firstElement = ((IStructuredSelection) selection).getFirstElement();

//			if(firstElement instanceof ICdmBase){

//				NavigationUtil.openEditor((ICdmBase) firstElement, PlatformUI.getWorkbench().getActiveWorkbenchWindow().getShell());

//			}

//		}

		// If the double click is passed up to the super-class it will

		// expand/collapse trees.

		// We do not want that

		// super.handleDoubleClick(anEvent);

	}

	/**

	 * <p>

	 * onComplete

	 * </p>

	 *

	 * @return a boolean.

	 */

	@Override

	public boolean onComplete() {

		return true;

	}

	/*

	 * (non-Javadoc)

	 *

	 * @see org.eclipse.ui.part.WorkbenchPart#showBusy(boolean)

	 */

	/** {@inheritDoc} */

	@Override

	public void showBusy(boolean busy) {

		super.showBusy(busy);

		getCommonViewer().getControl().setEnabled(!busy);

		if (busy) {

			partNameCache = getPartName();

			setPartName(Messages.TaxonNavigator_LOAD_DATASOURCE);

		} else {

			if (partNameCache != null) {

				setPartName(partNameCache);

			}

		}

	}

	/* (non-Javadoc)

	 * @see java.util.Observer#update(java.util.Observable, java.lang.Object)

	 */

	@Override

	public void update(Observable o, Object arg) {

		if(o instanceof LoginManager){

			refresh();

		}

	}

    /* (non-Javadoc)

     * @see eu.etaxonomy.taxeditor.session.ICdmEntitySessionEnabled#getCdmEntitySession()

     */

    @Override

    public ICdmEntitySession getCdmEntitySession() {

       return cdmEntitySession;

    }

    /* (non-Javadoc)

     * @see eu.etaxonomy.taxeditor.session.ICdmEntitySessionEnabled#getRootEntities()

     */

    @Override

    public List<TaxonNodeDto> getRootEntities() {

        if(root != null) {

            return root.getParentBeans();

        }

        return null;

    }

    /* (non-Javadoc)

     * @see eu.etaxonomy.cdm.api.application.ICdmChangeListener#onChange(eu.etaxonomy.cdm.api.application.CdmChangeEvent)

     */

    @Override

    public void onChange(CdmChangeEvent event) {

//        if(event.getAction() == Action.Delete && Classification.class.equals(event.getEntityType())) {

//            refresh();

//            return;

//        }

        refresh();

        for(CdmBase cb : event.getChangedObjects()) {

        	if(cb instanceof TaxonNode) {

                TaxonNode tn = (TaxonNode)cb;

                if(tn.getTaxon() == null) {

                    getCommonViewer().refresh(tn.getClassification());

                } else {

                    getCommonViewer().refresh(cb);

                }

            } else if (cb instanceof Classification) {

                getCommonViewer().refresh();

            }

        }

    }

    /* (non-Javadoc)

     * @see eu.etaxonomy.taxeditor.session.ICdmEntitySessionEnabled#getPropertyPathsMap()

     */

    @Override

    public Map<Object, List<String>> getPropertyPathsMap() {

        Map<Object, List<String>> propertyPathsMap = new HashMap<Object, List<String>>();

        List<String> taxonNodePropertyPaths = Arrays.asList(new String[] {

                "taxon.name" //$NON-NLS-1$

        });

         propertyPathsMap.put("childNodes", taxonNodePropertyPaths); //$NON-NLS-1$

        return propertyPathsMap;

    }

}

/**

* Copyright (C) 2007 EDIT

* European Distributed Institute of Taxonomy

* http://www.e-taxonomy.eu

*

* The contents of this file are subject to the Mozilla Public License Version 1.1

* See LICENSE.TXT at the top of this package for the full license terms.

*/

package eu.etaxonomy.taxeditor.navigation.navigator;

import java.util.HashSet;

import java.util.Set;

import org.eclipse.core.runtime.IProgressMonitor;

import org.eclipse.core.runtime.IStatus;

import org.eclipse.core.runtime.Status;

import org.eclipse.core.runtime.jobs.Job;

import org.eclipse.swt.widgets.Display;

import eu.etaxonomy.cdm.hibernate.HibernateProxyHelper;

import eu.etaxonomy.cdm.model.common.CdmBase;

import eu.etaxonomy.cdm.model.name.TaxonName;

import eu.etaxonomy.cdm.model.taxon.ITaxonTreeNode;

import eu.etaxonomy.cdm.model.taxon.Taxon;

import eu.etaxonomy.cdm.persistence.hibernate.CdmDataChangeEvent;

import eu.etaxonomy.cdm.persistence.hibernate.CdmDataChangeEvent.EventType;

import eu.etaxonomy.cdm.persistence.hibernate.CdmDataChangeMap;

import eu.etaxonomy.taxeditor.model.AbstractDataChangeBehaviour;

import eu.etaxonomy.taxeditor.model.IDataChangeBehavior;

import eu.etaxonomy.taxeditor.navigation.l10n.Messages;

/**

 * <p>TaxonNavigatorDataChangeBehavior class.</p>

 *

 * @author n.hoffmann

 * @created 01.04.2009

 * @version 1.0

 */

public class TaxonNavigatorDataChangeBehavior extends AbstractDataChangeBehaviour implements

		IDataChangeBehavior {

    private static final String UPDATING_TAXON_NAVIGATOR = Messages.TaxonNavigatorDataChangeBehavior_UPDATE_NAVIGATOR;

    private final TaxonNavigator source;

	private Set<CdmBase> staleObjects;

	/**

	 * <p>Constructor for TaxonNavigatorDataChangeBehavior.</p>

	 *

	 * @param taxonNavigator a {@link eu.etaxonomy.taxeditor.navigation.navigator.TaxonNavigator} object.

	 */

	public TaxonNavigatorDataChangeBehavior(TaxonNavigator taxonNavigator) {

		source = taxonNavigator;

	}

	/* (non-Javadoc)

	 * @see eu.etaxonomy.taxeditor.store.model.IDataChangeBehavior#isRelevant(java.lang.Object, eu.etaxonomy.cdm.persistence.hibernate.CdmCrudEvent)

	 */

	/**

	 * <p>isRelevant</p>

	 *

	 * @param events a {@link eu.etaxonomy.cdm.persistence.hibernate.CdmDataChangeMap} object.

	 * @return a boolean.

	 */

	public boolean isRelevant(CdmDataChangeMap events) {

		// TODO react only on insert/update/delete of taxon and synonym objects

		// and on update of name objects

		boolean relevant = false;

		staleObjects = new HashSet<CdmBase>();

		for(CdmDataChangeEvent event : events.getAllEvents()){

			EventType eventType = event.getEventType();

			CdmBase eventEntity = event.getEntity();

			Set<CdmBase> affectedObjects = event.getAffectedObjects();

			if(affectedObjects != null) {

			    for(CdmBase cb : affectedObjects) {

			        staleObjects.add(HibernateProxyHelper.deproxy(cb));

			    }

			}

			// all tree node changes are relevant

			if((eventType == EventType.INSERT || eventType == EventType.DELETE || eventType == EventType.UPDATE)

					&& event.getEntity() instanceof ITaxonTreeNode){

				return true;

			}

			if (eventType == EventType.DELETE){

				return true;

			}

			if(eventType == EventType.UPDATE && event.getEntity() instanceof Taxon){

				TaxonName name = null;

				if(eventEntity instanceof Taxon){

					name = ((Taxon) eventEntity).getName();

				}else{

					continue;

				}

				// Set<IEditorPart> openEditors = NavigationUtil.getOpenEditors();

				/*for(IEditorPart editor : openEditors){

					if(name.equals(((TaxonEditorInput) editor.getEditorInput()).getTaxon().getName())){

						return true;

					}

				}*/

			}

//			if(eventType == EventType.UPDATE){

//				relevant = true;

//				CdmBase entity = event.getEntity();

//				if((entity instanceof TaxonNameBase)

//						|| (entity instanceof Taxon)

//						|| (entity instanceof Synonym)){

//					staleObjects.add(entity);

//				}

//			}

		}

		return false;

		// @deprecated

		// react on everything except load

//		if(events.sizeByEventType(EventType.INSERT) > 0){

//			return true;

//		}else if(events.sizeByEventType(EventType.UPDATE) > 0){

//			return true;

//		}else if(events.sizeByEventType(EventType.DELETE) > 0){

//			return true;

//		}else{

//			return false;

//		}

	}

	/** {@inheritDoc} */

	@Override

	public void reactOnDataChange(CdmDataChangeMap events) {

		if(isRelevant(events)){

			final Display display = Display.getCurrent();

			Job job = new Job(UPDATING_TAXON_NAVIGATOR) {

				@Override

				protected IStatus run(IProgressMonitor monitor) {

					monitor.beginTask(UPDATING_TAXON_NAVIGATOR, 3);

					monitor.worked(1);

					// clear the session completely

					monitor.subTask(Messages.TaxonNavigatorDataChangeBehavior_CLEAR_SESSION);

					display.asyncExec(new Runnable() {

						 @Override

						 public void run() {

							 source.getConversationHolder().clear();

						 }

					});

					// FIXME completely clearing the session is a brute force approach.

					// It would be much more elegant to clear only those elements that have been changed.

					// I could not get that to work but we should consider workin on this because we might

					// run into serious performance issues, especially when it comes to large trees

					//

					// at least, we moved this to a job so it can run in a background thred

					// seems to improve the situation but not sure if final solution

					monitor.worked(1);

					monitor.subTask(Messages.TaxonNavigatorDataChangeBehavior_REFRESH_VIEWER);

					display.asyncExec(new Runnable() {

					    @Override

					    public void run() {

					        if(staleObjects != null && staleObjects.size() > 0) {

					           source.refresh(staleObjects);

					        } else {

					            source.refresh();

					        }

					    }

					});

					monitor.worked(1);

					monitor.done();

					return Status.OK_STATUS;

				}

			};

			job.setPriority(Job.SHORT);

			job.schedule();

		}

	}

}