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

import java.util.Arrays;

import java.util.Collection;

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 javax.inject.Named;

import org.eclipse.core.runtime.IStatus;

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.services.IServiceConstants;

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.wizard.WizardDialog;

import org.eclipse.swt.SWT;

import org.eclipse.swt.events.ModifyEvent;

import org.eclipse.swt.events.ModifyListener;

import org.eclipse.swt.events.SelectionAdapter;

import org.eclipse.swt.events.SelectionEvent;

import org.eclipse.swt.widgets.Composite;

import org.eclipse.swt.widgets.Shell;

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

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

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

import eu.etaxonomy.cdm.api.service.config.FeatureNodeDeletionConfigurator;

import eu.etaxonomy.cdm.hibernate.HibernateProxyHelper;

import eu.etaxonomy.cdm.model.description.Feature;

import eu.etaxonomy.cdm.model.description.FeatureNode;

import eu.etaxonomy.cdm.model.description.FeatureTree;

import eu.etaxonomy.taxeditor.featuretree.AvailableFeaturesWizard;

import eu.etaxonomy.taxeditor.session.ICdmEntitySession;

import eu.etaxonomy.taxeditor.session.ICdmEntitySessionEnabled;

import eu.etaxonomy.taxeditor.store.CdmStore;

import eu.etaxonomy.taxeditor.ui.dialog.selection.FeatureTreeSelectionDialog;

/**

 *

 * @author pplitzner

 * @date 06.06.2017

 *

 */

public class FeatureTreeEditor implements ICdmEntitySessionEnabled,

		ModifyListener, ISelectionChangedListener {

    private ConversationHolder conversation;

    private ICdmEntitySession cdmEntitySession;

    @Inject

    private ESelectionService selService;

    @Inject

    private MDirtyable dirty;

    private Shell shell;

    private FeatureTree featureTree;

    private FeatureTreeEditorComposite composite;

    @Inject

    public FeatureTreeEditor(@Named(IServiceConstants.ACTIVE_SHELL)Shell shell) {

        this.shell = shell;

        if(conversation==null){

            conversation = CdmStore.createConversation();

        }

        if (CdmStore.isActive()) {

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

        }

    }

	/** {@inheritDoc} */

	@PostConstruct

	public void createControl(Composite parent, @Named(IServiceConstants.ACTIVE_SHELL)Shell shell) {

	    composite = new FeatureTreeEditorComposite(parent, SWT.NULL);

		composite.init(new FeatureNodeDragListener(composite.getViewer()), new FeatureNodeDropAdapter(this, composite.getViewer()), this, new SelectionAdapter() {

            @Override

            public void widgetSelected(SelectionEvent e) {

                FeatureTree tree = FeatureTreeSelectionDialog.select(shell, conversation, null);

                if(tree!=null){

                    setSelectedTree(tree);

                }

            }

        }, new AddButtonListener(), new RemoveSelectionListener());

        composite.getText_title().setEnabled(false);

	}

	public void setDirty(boolean isDirty){

	    this.dirty.setDirty(isDirty);

	}

	public boolean isDirty(){

	    return dirty.isDirty();

	}

    public void setSelectedTree(FeatureTree featureTree) {

		this.featureTree = HibernateProxyHelper.deproxy(featureTree, FeatureTree.class);

		this.featureTree.setRoot(HibernateProxyHelper.deproxy(featureTree.getRoot(), FeatureNode.class));

		composite.getViewer().setInput(featureTree);

		composite.getText_title().setEnabled(true);

		composite.getText_title().removeModifyListener(this);

		composite.getText_title().setText(featureTree.getTitleCache());

		composite.getText_title().addModifyListener(this);

	}

	/** {@inheritDoc} */

	@Override

	public void modifyText(ModifyEvent e) {

		featureTree.setTitleCache(composite.getText_title().getText(), true);

		setDirty(true);

	}

	/** {@inheritDoc} */

	@Override

	public void selectionChanged(SelectionChangedEvent event) {

		IStructuredSelection selection = (IStructuredSelection) event

				.getSelection();

		composite.getButton_add().setEnabled(selection.size() <= 1);

		composite.getButton_remove().setEnabled(selection.size() > 0);

		//propagate selection

		IStructuredSelection isel = (IStructuredSelection) event.getSelection();

		selService.setSelection((isel.size() == 1 ? isel.getFirstElement() : isel.toArray()));

	}

	@Focus

	public void focus(){

	    composite.getViewer().getControl().setFocus();

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

            conversation.bind();

        }

        if(cdmEntitySession != null) {

            cdmEntitySession.bind();

        }

	}

	@Persist

	public void save(){

        if (!conversation.isBound()) {

            conversation.bind();

        }

        // commit the conversation and start a new transaction immediately

        conversation.commit(true);

        CdmStore.getService(IFeatureTreeService.class).saveOrUpdate(featureTree);

        this.setDirty(false);

	}

	@PreDestroy

	public void dispose(){

        if(conversation!=null){

            conversation.close();

        }

        if(cdmEntitySession != null) {

            cdmEntitySession.dispose();

        }

	}

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

        List<FeatureTree> root = new ArrayList<>();

        root.add(featureTree);

        return root;

    }

	private class AddButtonListener extends SelectionAdapter {

		@Override

		public void widgetSelected(SelectionEvent e) {

			AvailableFeaturesWizard wizard = new AvailableFeaturesWizard(

					featureTree);

			WizardDialog dialog = new WizardDialog(shell, wizard);

			if (dialog.open() == IStatus.OK) {

                IStructuredSelection selection = (IStructuredSelection) composite.getViewer().getSelection();

                FeatureNode parent = (FeatureNode) (selection.getFirstElement() != null ? selection.getFirstElement()

                        : ((FeatureTree) composite.getViewer().getInput()).getRoot());

                Collection<Feature> additionalFeatures = wizard.getAdditionalFeatures();

                for (Feature feature : additionalFeatures) {

				    CdmStore.getService(IFeatureNodeService.class).addChildFeaturNode(parent, feature);

				}

				setDirty(true);

				composite.getViewer().refresh();

				composite.getViewer().expandToLevel(parent, 1);

			}

		}

	}

	private class RemoveSelectionListener extends SelectionAdapter {

		@Override

		public void widgetSelected(SelectionEvent e) {

			IStructuredSelection selection = (IStructuredSelection) composite.getViewer()

					.getSelection();

			for (Object selectedObject : selection.toArray()) {

				FeatureNode featureNode = (FeatureNode) selectedObject;

				CdmStore.getService(IFeatureNodeService.class).deleteFeatureNode(featureNode.getUuid(), new FeatureNodeDeletionConfigurator());

			}

			setDirty(true);

			composite.getViewer().refresh();

		}

	}

}