ref #6774 Adapt to service layer changes
[taxeditor.git] / eu.etaxonomy.taxeditor.store / src / main / java / eu / etaxonomy / taxeditor / featuretree / e4 / FeatureTreeEditor.java
index bdfde16a0bb2b3fbe3e212ca7dd16d9acc220b2f..26969d86979201133986ca20c12fd0ef245b75e1 100644 (file)
@@ -10,6 +10,7 @@
 package eu.etaxonomy.taxeditor.featuretree.e4;
 
 import java.util.Arrays;
+import java.util.Collections;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
@@ -21,17 +22,24 @@ import javax.inject.Inject;
 import org.eclipse.core.runtime.IProgressMonitor;
 import org.eclipse.e4.ui.di.Focus;
 import org.eclipse.e4.ui.di.Persist;
+import org.eclipse.e4.ui.di.UISynchronize;
 import org.eclipse.e4.ui.model.application.ui.MDirtyable;
 import org.eclipse.e4.ui.model.application.ui.basic.MPart;
 import org.eclipse.e4.ui.services.EMenuService;
 import org.eclipse.e4.ui.workbench.modeling.EPartService;
 import org.eclipse.e4.ui.workbench.modeling.ESelectionService;
+import org.eclipse.jface.util.LocalSelectionTransfer;
+import org.eclipse.jface.viewers.ISelection;
 import org.eclipse.jface.viewers.ISelectionChangedListener;
 import org.eclipse.jface.viewers.IStructuredSelection;
 import org.eclipse.jface.viewers.SelectionChangedEvent;
+import org.eclipse.jface.viewers.StructuredSelection;
 import org.eclipse.jface.viewers.TreeViewer;
+import org.eclipse.swt.SWT;
 import org.eclipse.swt.dnd.DND;
 import org.eclipse.swt.dnd.Transfer;
+import org.eclipse.swt.events.KeyAdapter;
+import org.eclipse.swt.events.KeyEvent;
 import org.eclipse.swt.layout.FillLayout;
 import org.eclipse.swt.widgets.Composite;
 import org.eclipse.ui.IMemento;
@@ -39,21 +47,26 @@ import org.eclipse.ui.IMemento;
 import eu.etaxonomy.cdm.api.conversation.ConversationHolder;
 import eu.etaxonomy.cdm.api.conversation.IConversationEnabled;
 import eu.etaxonomy.cdm.api.service.IFeatureTreeService;
-import eu.etaxonomy.cdm.model.description.FeatureNode;
-import eu.etaxonomy.cdm.model.description.FeatureTree;
+import eu.etaxonomy.cdm.api.service.ITermService;
+import eu.etaxonomy.cdm.model.description.Character;
+import eu.etaxonomy.cdm.model.term.FeatureNode;
+import eu.etaxonomy.cdm.model.term.FeatureTree;
+import eu.etaxonomy.cdm.model.term.TermType;
 import eu.etaxonomy.cdm.persistence.hibernate.CdmDataChangeMap;
 import eu.etaxonomy.taxeditor.editor.definedterm.TermTransfer;
 import eu.etaxonomy.taxeditor.featuretree.FeatureNodeTransfer;
 import eu.etaxonomy.taxeditor.featuretree.FeatureTreeContentProvider;
 import eu.etaxonomy.taxeditor.featuretree.FeatureTreeLabelProvider;
+import eu.etaxonomy.taxeditor.featuretree.e4.operation.AddFeatureOperation;
+import eu.etaxonomy.taxeditor.model.AbstractUtility;
 import eu.etaxonomy.taxeditor.model.IContextListener;
 import eu.etaxonomy.taxeditor.model.IDirtyMarkable;
 import eu.etaxonomy.taxeditor.model.IPartContentHasDetails;
 import eu.etaxonomy.taxeditor.model.IPartContentHasSupplementalData;
 import eu.etaxonomy.taxeditor.session.ICdmEntitySession;
-import eu.etaxonomy.taxeditor.session.ICdmEntitySessionEnabled;
+import eu.etaxonomy.taxeditor.store.AppModelId;
 import eu.etaxonomy.taxeditor.store.CdmStore;
-import eu.etaxonomy.taxeditor.workbench.part.IE4SavablePart;
+import eu.etaxonomy.taxeditor.store.StoreUtil;
 import eu.etaxonomy.taxeditor.workbench.part.IE4ViewerPart;
 
 /**
@@ -62,8 +75,8 @@ import eu.etaxonomy.taxeditor.workbench.part.IE4ViewerPart;
  * @date 06.06.2017
  *
  */
-public class FeatureTreeEditor implements ICdmEntitySessionEnabled, ISelectionChangedListener,
-        IE4ViewerPart, IE4SavablePart, IPartContentHasDetails, IPartContentHasSupplementalData,
+public class FeatureTreeEditor implements IFeatureTreeEditor, ISelectionChangedListener,
+        IE4ViewerPart, IPartContentHasDetails, IPartContentHasSupplementalData,
         IContextListener, IConversationEnabled, IDirtyMarkable {
 
     private ConversationHolder conversation;
@@ -76,7 +89,8 @@ public class FeatureTreeEditor implements ICdmEntitySessionEnabled, ISelectionCh
     @Inject
     private MDirtyable dirty;
 
-//    private FeatureTreeEditorComposite composite;
+    @Inject
+    private UISynchronize sync;
 
     @Inject
     private MPart thisPart;
@@ -88,7 +102,6 @@ public class FeatureTreeEditor implements ICdmEntitySessionEnabled, ISelectionCh
         CdmStore.getContextManager().addContextListener(this);
     }
 
-       /** {@inheritDoc} */
     @PostConstruct
     public void createControl(Composite parent, EMenuService menuService){
         if (CdmStore.isActive()){
@@ -103,59 +116,65 @@ public class FeatureTreeEditor implements ICdmEntitySessionEnabled, ISelectionCh
         viewer.setLabelProvider(new FeatureTreeLabelProvider());
 
         int ops = DND.DROP_COPY | DND.DROP_MOVE;
-        Transfer[] transfers = new Transfer[] { FeatureNodeTransfer
-                .getInstance(), TermTransfer.getInstance() };
+        Transfer[] transfers = new Transfer[] {
+                FeatureNodeTransfer.getInstance(),
+                TermTransfer.getInstance(),
+                LocalSelectionTransfer.getTransfer()};
         viewer.addDragSupport(ops, transfers, new FeatureNodeDragListener(viewer));
-        viewer.addDropSupport(ops, transfers, new FeatureNodeDropAdapter(dirty, viewer));
+        viewer.addDropSupport(ops, transfers, new FeatureTreeDropAdapter(this, viewer, sync));
         viewer.addSelectionChangedListener(this);
+        viewer.getTree().addKeyListener(new KeyAdapter() {
+            @Override
+            public void keyPressed(KeyEvent e) {
+                if(e.stateMask == SWT.MOD1 && e.keyCode == 'c'){
+                    copy(viewer.getStructuredSelection());
+                }
+                else if(e.stateMask == SWT.MOD1 && e.keyCode == 'v'){
+                    paste(viewer.getStructuredSelection());
+                }
+            }
+        });
 
         List<FeatureTree> trees = CdmStore.getService(IFeatureTreeService.class).list(FeatureTree.class, null, null, null, null);
+        Collections.sort(trees, (tree1, tree2) -> tree1.getTitleCache().compareTo(tree2.getTitleCache()));
         viewer.setInput(trees);
 
-//        composite = new FeatureTreeEditorComposite(parent, SWT.NULL);
-//        composite.init(new FeatureNodeDragListener(composite.getViewer()),
-//                new FeatureNodeDropAdapter(dirty, composite.getViewer()), this, new SelectionAdapter() {
-//            @Override
-//            public void widgetSelected(SelectionEvent e) {
-//                if(isDirty()){
-//                    MessageDialog dialog;
-//                    String[] buttonLables = {IDialogConstants.YES_LABEL, IDialogConstants.NO_LABEL, IDialogConstants.CANCEL_LABEL};
-//                    dialog = new MessageDialog(null, "Unsaved changes", null, "You have unsaved changes. Do you want to save?",
-//                            MessageDialog.QUESTION_WITH_CANCEL, 0, buttonLables);
-//                    int returnCode = dialog.open();
-//
-//                    if (returnCode == 0){
-//                        //YES
-//                        save(new NullProgressMonitor());
-//                    } else if (returnCode == 1){
-//                        //NO
-//                        if(CdmStore.isActive()){
-//                            clearSession();
-//                            initSession();
-//                            composite.setSelectedTree(null);
-//                        }
-//                    } else if (returnCode == 2){
-//                        //CANCEL
-//                        return;
-//                    }
-//                }
-//                FeatureTree tree = FeatureTreeSelectionDialog.select(composite.getDisplay().getActiveShell(), //conversation,
-//                        null);
-//                if (tree != null) {
-//                    composite.setSelectedTree(tree);
-//                }
-//            }
-//        }, this);
-
         //create context menu
-        menuService.registerContextMenu(viewer.getControl(), "eu.etaxonomy.taxeditor.store.popupmenu.featureTreeEditor");
+        menuService.registerContextMenu(viewer.getControl(), AppModelId.POPUPMENU_EU_ETAXONOMY_TAXEDITOR_STORE_POPUPMENU_FEATURETREEEDITOR);
+    }
+
+    public void paste(IStructuredSelection selection) {
+        if (StoreUtil.promptCheckIsDirty(this)) {
+            return;
+        }
+
+        ISelection clipBoardSelection = LocalSelectionTransfer.getTransfer().getSelection();
+        Object firstElement = selection.getFirstElement();
+        FeatureNode parentNode = null;
+        if(firstElement instanceof FeatureNode){
+            parentNode = (FeatureNode) firstElement;
+        }
+        else if(firstElement instanceof FeatureTree){
+            parentNode = ((FeatureTree)firstElement).getRoot();
+        }
+        if(parentNode!=null){
+            FeatureNode copiedNode = (FeatureNode) ((IStructuredSelection)clipBoardSelection).getFirstElement();
+
+            AddFeatureOperation operation = new AddFeatureOperation(copiedNode.getTerm().getUuid(), parentNode, this, this);
+            AbstractUtility.executeOperation(operation, sync);
+        }
+
+    }
+
+    public void copy(IStructuredSelection selection) {
+        LocalSelectionTransfer.getTransfer().setSelection(selection);
     }
 
     private void initSession(){
         if(conversation == null){
             conversation = CdmStore.createConversation();
         }
-        if(cdmEntitySession!=null){
+        if(cdmEntitySession==null){
             cdmEntitySession = CdmStore.getCurrentSessionManager().newSession(this, true);
         }
     }
@@ -176,11 +195,11 @@ public class FeatureTreeEditor implements ICdmEntitySessionEnabled, ISelectionCh
            this.dirty.setDirty(isDirty);
        }
 
-       public boolean isDirty(){
+       @Override
+    public boolean isDirty(){
            return dirty.isDirty();
        }
 
-       /** {@inheritDoc} */
        @Override
        public void selectionChanged(SelectionChangedEvent event) {
                //propagate selection
@@ -209,17 +228,11 @@ public class FeatureTreeEditor implements ICdmEntitySessionEnabled, ISelectionCh
            return viewer;
        }
 
-       /**
-        * {@inheritDoc}
-        */
        @Override
        public IStructuredSelection getSelection() {
            return (IStructuredSelection) viewer.getSelection();
        }
 
-       /**
-        * {@inheritDoc}
-        */
        @Override
        public ConversationHolder getConversationHolder() {
            return conversation;
@@ -235,12 +248,30 @@ public class FeatureTreeEditor implements ICdmEntitySessionEnabled, ISelectionCh
         // commit the conversation and start a new transaction immediately
         conversation.commit(true);
 
-        List<FeatureTree> trees = (List<FeatureTree>) viewer.getInput();
-        CdmStore.getService(IFeatureTreeService.class).merge(trees);
+        CdmStore.getService(IFeatureTreeService.class).saveOrUpdate(getRootEntities());
+
+        List<FeatureTree> rootEntities = getRootEntities();
+        for (FeatureTree featureTree : rootEntities) {
+            if(featureTree.getTermType().equals(TermType.Character)){
+                FeatureTree<Character> characterTree = featureTree;
+                //save characters because they can be modified in this editor
+                characterTree.getDistinctFeatures().forEach(character->CdmStore.getService(ITermService.class).merge(character,true));
+            }
+        }
+
+        initializeTrees();
 
         this.setDirty(false);
        }
 
+    private void initializeTrees() {
+        Object[] expandedElements = viewer.getExpandedElements();
+        viewer.getTree().removeAll();
+        List<FeatureTree> trees = CdmStore.getService(IFeatureTreeService.class).list(FeatureTree.class, null, null, null, null);
+        viewer.setInput(trees);
+        viewer.setExpandedElements(expandedElements);
+    }
+
        @PreDestroy
        public void dispose(){
            selService.setSelection(null);
@@ -265,24 +296,15 @@ public class FeatureTreeEditor implements ICdmEntitySessionEnabled, ISelectionCh
         return propertyPathMap;
     }
 
-    /**
-     * {@inheritDoc}
-     */
     @Override
     public List<FeatureTree> getRootEntities() {
         return (List<FeatureTree>) viewer.getInput();
     }
 
-    /**
-     * {@inheritDoc}
-     */
     @Override
     public void contextAboutToStop(IMemento memento, IProgressMonitor monitor) {
     }
 
-    /**
-     * {@inheritDoc}
-     */
     @Override
     public void contextStop(IMemento memento, IProgressMonitor monitor) {
         //close view when workbench closes
@@ -294,49 +316,51 @@ public class FeatureTreeEditor implements ICdmEntitySessionEnabled, ISelectionCh
         }
     }
 
-    /**
-     * {@inheritDoc}
-     */
     @Override
     public void contextStart(IMemento memento, IProgressMonitor monitor) {
     }
 
-    /**
-     * {@inheritDoc}
-     */
     @Override
     public void contextRefresh(IProgressMonitor monitor) {
     }
 
-    /**
-     * {@inheritDoc}
-     */
     @Override
     public void workbenchShutdown(IMemento memento, IProgressMonitor monitor) {
     }
 
-    /**
-     * {@inheritDoc}
-     */
     @Override
     public void update(CdmDataChangeMap arg0) {
     }
 
-    /**
-     * {@inheritDoc}
-     */
     @Override
     public void changed(Object element) {
         dirty.setDirty(true);
         viewer.refresh();
     }
 
-    /**
-     * {@inheritDoc}
-     */
     @Override
     public void forceDirty() {
         dirty.setDirty(true);
     }
 
+    @Override
+    public boolean postOperation(Object objectAffectedByOperation) {
+        initializeTrees();
+        viewer.refresh();
+        if(objectAffectedByOperation instanceof FeatureNode){
+            FeatureNode node = (FeatureNode)objectAffectedByOperation;
+            viewer.expandToLevel(node.getFeatureTree(), 1);
+        }
+        if(objectAffectedByOperation!=null){
+            StructuredSelection selection = new StructuredSelection(objectAffectedByOperation);
+            viewer.setSelection(selection);
+        }
+        return true;
+    }
+
+    @Override
+    public boolean onComplete() {
+        return false;
+    }
+
 }