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.core.di.annotations.Optional;

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.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.model.AbstractUtility;

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 FeatureTreeEditorComposite composite;

    @Inject

    public FeatureTreeEditor() {

    }

	/** {@inheritDoc} */

    @PostConstruct

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

        if (CdmStore.isActive()){

            if(conversation == null){

                conversation = CdmStore.createConversation();

            }

            if(cdmEntitySession!=null){

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

            }

        }

        else{

            return;

        }

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

                            composite.setSelectedTree(tree, FeatureTreeEditor.this);

                        }

                    }

                }, new AddButtonListener(), new RemoveSelectionListener(), new FeatureTreeExportListener(shell, composite));

    }

	public void setDirty(boolean isDirty){

	    this.dirty.setDirty(isDirty);

	}

	public boolean isDirty(){

	    return dirty.isDirty();

	}

    public FeatureTree getSelectedFeatureTree(){

        return composite.getFeatureTree();

    }

	/** {@inheritDoc} */

	@Override

	public void modifyText(ModifyEvent e) {

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

		setDirty(true);

	}

	/** {@inheritDoc} */

	@Override

	public void selectionChanged(SelectionChangedEvent event) {

		IStructuredSelection selection = (IStructuredSelection) event

				.getSelection();

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

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

		//propagate selection

		selService.setSelection(AbstractUtility.getElementsFromSelectionChangedEvent(event));

	}

	@Focus

	public void focus(){

	    if(composite!=null){

	        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(composite.getFeatureTree());

        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(composite.getFeatureTree());

        return root;

    }

	private class AddButtonListener extends SelectionAdapter {

		@Override

		public void widgetSelected(SelectionEvent e) {

			AvailableFeaturesWizard wizard = new AvailableFeaturesWizard();

			WizardDialog dialog = new WizardDialog(e.widget.getDisplay().getActiveShell(), wizard);

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

                FeatureNode parent = ((FeatureTree) composite.getViewer().getInput()).getRoot();

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

                for (Feature feature : additionalFeatures) {

                    if(!getSelectedFeatureTree().getDistinctFeatures().contains(feature)){

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

		}

	}

}