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

import javax.annotation.PostConstruct;

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

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

/**

 *

 * @author pplitzner

 * @date 06.06.2017

 *

 */

public class FeatureTreeEditor implements

		ModifyListener, ISelectionChangedListener {

    private ConversationHolder conversation;

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

        }

    }

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

        }

	}

	@Persist

	public void save(){

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

	    setDirty(false);

	}

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

					FeatureNode child = FeatureNode.NewInstance(feature);

					parent.addChild(child);

					CdmStore.getService(IFeatureNodeService.class).save(child);

				}

				setDirty(true);

				composite.getViewer().refresh();

			}

		}

	}

	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;

				FeatureNode parent = featureNode.getParent();

				parent.removeChild(featureNode);

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

			}

			setDirty(true);

			composite.getViewer().refresh();

		}

	}

}