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.featuretree.e4;

import java.util.Arrays;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import javax.annotation.PostConstruct;

import javax.annotation.PreDestroy;

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.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.viewers.ISelectionChangedListener;

import org.eclipse.jface.viewers.IStructuredSelection;

import org.eclipse.jface.viewers.SelectionChangedEvent;

import org.eclipse.jface.viewers.TreeViewer;

import org.eclipse.swt.dnd.DND;

import org.eclipse.swt.dnd.Transfer;

import org.eclipse.swt.layout.FillLayout;

import org.eclipse.swt.widgets.Composite;

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.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.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.CdmStore;

import eu.etaxonomy.taxeditor.workbench.part.IE4SavablePart;

import eu.etaxonomy.taxeditor.workbench.part.IE4ViewerPart;

/**

 *

 * @author pplitzner

 * @date 06.06.2017

 *

 */

public class FeatureTreeEditor implements ICdmEntitySessionEnabled, ISelectionChangedListener,

        IE4ViewerPart, IE4SavablePart, IPartContentHasDetails, IPartContentHasSupplementalData,

        IContextListener, IConversationEnabled, IDirtyMarkable {

    private ConversationHolder conversation;

    private ICdmEntitySession cdmEntitySession;

    @Inject

    private ESelectionService selService;

    @Inject

    private MDirtyable dirty;

//    private FeatureTreeEditorComposite composite;

    @Inject

    private MPart thisPart;

    private TreeViewer viewer;

    @Inject

    public FeatureTreeEditor() {

        CdmStore.getContextManager().addContextListener(this);

    }

	/** {@inheritDoc} */

    @PostConstruct

    public void createControl(Composite parent, EMenuService menuService){

        if (CdmStore.isActive()){

            initSession();

        }

        else{

            return;

        }

        parent.setLayout(new FillLayout());

        viewer = new TreeViewer(parent);

        viewer.setContentProvider(new FeatureTreeContentProvider());

        viewer.setLabelProvider(new FeatureTreeLabelProvider());

        int ops = DND.DROP_COPY | DND.DROP_MOVE;

        Transfer[] transfers = new Transfer[] { FeatureNodeTransfer

                .getInstance(), TermTransfer.getInstance() };

        viewer.addDragSupport(ops, transfers, new FeatureNodeDragListener(viewer));

        viewer.addDropSupport(ops, transfers, new FeatureNodeDropAdapter(dirty, viewer));

        viewer.addSelectionChangedListener(this);

        List<FeatureTree> trees = CdmStore.getService(IFeatureTreeService.class).list(FeatureTree.class, null, null, null, null);

        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");

    }

    private void initSession(){

        if(conversation == null){

            conversation = CdmStore.createConversation();

        }

        if(cdmEntitySession!=null){

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

        }

    }

    private void clearSession() {

        if(conversation!=null){

            conversation.close();

            conversation = null;

        }

        if(cdmEntitySession != null) {

            cdmEntitySession.dispose();

            cdmEntitySession = null;

        }

        dirty.setDirty(false);

    }

	public void setDirty(boolean isDirty){

	    this.dirty.setDirty(isDirty);

	}

	public boolean isDirty(){

	    return dirty.isDirty();

	}

	/** {@inheritDoc} */

	@Override

	public void selectionChanged(SelectionChangedEvent event) {

		//propagate selection

		selService.setSelection(event.getSelection());

	}

	@Focus

	public void focus(){

	    if(viewer!=null){

	        viewer.getControl().setFocus();

	    }

        if(conversation!=null && !conversation.isBound()){

            conversation.bind();

        }

        if(cdmEntitySession != null) {

            cdmEntitySession.bind();

        }

	}

	@Override

    public void refresh(){

	    viewer.refresh();

	}

	public TreeViewer getViewer(){

	    return viewer;

	}

	/**

	 * {@inheritDoc}

	 */

	@Override

	public IStructuredSelection getSelection() {

	    return (IStructuredSelection) viewer.getSelection();

	}

	/**

	 * {@inheritDoc}

	 */

	@Override

	public ConversationHolder getConversationHolder() {

	    return conversation;

	}

	@Override

    @Persist

	public void save(IProgressMonitor monitor){

        if (!conversation.isBound()) {

            conversation.bind();

        }

        // 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);

        this.setDirty(false);

	}

	@PreDestroy

	public void dispose(){

	    selService.setSelection(null);

	    clearSession();

	}

    @Override

    public ICdmEntitySession getCdmEntitySession() {

        return cdmEntitySession;

    }

    @Override

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

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

                "children", //$NON-NLS-1$

                "feature", //$NON-NLS-1$

                "featureTree", //$NON-NLS-1$

        });

        Map<Object, List<String>> propertyPathMap =

                new HashMap<Object, List<String>>();

        propertyPathMap.put(FeatureNode.class,propertyPaths);

        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

        try{

            thisPart.getContext().get(EPartService.class).hidePart(thisPart);

        }

        catch(Exception e){

            //nothing

        }

    }

    /**

     * {@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);

    }

}