EDIT

/**

* Copyright (C) 2019 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.navigation.navigator.e4.handler;

import java.util.Collection;

import java.util.Iterator;

import javax.inject.Named;

import org.eclipse.core.commands.operations.AbstractOperation;

import org.eclipse.core.runtime.IStatus;

import org.eclipse.core.runtime.NullProgressMonitor;

import org.eclipse.core.runtime.Status;

import org.eclipse.e4.core.di.annotations.CanExecute;

import org.eclipse.e4.ui.model.application.ui.basic.MPart;

import org.eclipse.e4.ui.model.application.ui.menu.MHandledMenuItem;

import org.eclipse.e4.ui.services.IServiceConstants;

import org.eclipse.jface.dialogs.MessageDialog;

import org.eclipse.jface.viewers.IStructuredSelection;

import org.eclipse.jface.viewers.TreeSelection;

import org.eclipse.jface.window.Window;

import org.eclipse.jface.wizard.WizardDialog;

import org.eclipse.swt.widgets.Shell;

import eu.etaxonomy.cdm.api.service.description.AggregationMode;

import eu.etaxonomy.cdm.api.service.description.DistributionAggregationConfiguration;

import eu.etaxonomy.cdm.filter.TaxonNodeFilter;

import eu.etaxonomy.cdm.model.taxon.Classification;

import eu.etaxonomy.cdm.persistence.dto.TaxonNodeDto;

import eu.etaxonomy.taxeditor.editor.EditorUtil;

import eu.etaxonomy.taxeditor.navigation.l10n.Messages;

import eu.etaxonomy.taxeditor.navigation.navigator.TaxonNavigatorLabels;

import eu.etaxonomy.taxeditor.operation.e4.RemotingCdmHandlerE4;

import eu.etaxonomy.taxeditor.ui.dialog.configurator.DistributionAggregationConfiguratorWizard;

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

/**

 * @author k.luther

 * @since 15.11.2019

 */

public class AggregateDistributionForSubTreeHandler extends RemotingCdmHandlerE4 {

    private TaxonNodeDto taxonNode;

    private DistributionAggregationConfiguration configurator;

    private static final String NO = Messages.NO;

    private static final String CANCEL = Messages.RemotingMoveTaxonNodeHandler_CANCEL;

    private static final String YES = Messages.YES;

    /**

     * @param label

     */

    public AggregateDistributionForSubTreeHandler() {

        super("");

    }

    @Override

    public IStatus allowOperations(IStructuredSelection selection, Shell shell, MPart activePart,

            MHandledMenuItem menuItem) {

     // check that only a single taxon tree node has been selected

        if(selection.size() > 1) {  }

        // check for no taxon tree node selected

        if(selection.size() == 0) {

            return new Status(IStatus.ERROR,

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

                    TaxonNavigatorLabels.NO_TAXON_SELECTION_MESSAGE);

        }

        // check that selected object is a taxon node

        Object obj = selection.iterator().next();

        if(obj instanceof TaxonNodeDto) {

             taxonNode = (TaxonNodeDto)obj;

        } else if(obj instanceof Classification){

            taxonNode = new TaxonNodeDto(((Classification)obj).getRootNode());

        }else{

            return new Status(IStatus.ERROR,

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

                    TaxonNavigatorLabels.SELECTED_OBJECT_NOT_TREE_NODE_MESSAGE);

        }

        Collection<IE4SavablePart> dirtyParts = EditorUtil.checkForTaxonChanges(null, partService);

        if (dirtyParts != null && !dirtyParts.isEmpty()){

            String[] buttonLables = {YES, NO,CANCEL};

            MessageDialog dialog = new MessageDialog(null,  Messages.SetPublishFlagForSubtreeHandlerE4_UnsavedChanges, null, Messages.SetPublishFlagForSubtreeHandlerE4_UnsavedChangesQuestion, MessageDialog.QUESTION_WITH_CANCEL, buttonLables, 0);

            dialog.open();

            int returnCode = dialog.getReturnCode();

            boolean proceed = false;

            if (returnCode == 0){

                proceed = true;

            }else if (returnCode == 2){

                return Status.CANCEL_STATUS;

            }

            Iterator<IE4SavablePart> partIterator = dirtyParts.iterator();

            while( partIterator.hasNext() ){

                IE4SavablePart part = partIterator.next();

                if (proceed) {

                    if (part != null){

                        part.save(new NullProgressMonitor());

                    }

                }

            }

        }

        configurator = DistributionAggregationConfiguration.NewInstance(AggregationMode.byAreasAndRanks(), null, TaxonNodeFilter.NewSubtreeInstance(taxonNode.getUuid()), null);

        DistributionAggregationConfiguratorWizard wizard = new DistributionAggregationConfiguratorWizard(configurator);

        WizardDialog dialog = new WizardDialog(shell, wizard);

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

            return Status.OK_STATUS;

        }else{

            return Status.CANCEL_STATUS;

        }

    }

    @Override

    public AbstractOperation prepareOperation(IStructuredSelection selection, Shell shell, MPart activePart,

            MHandledMenuItem menuItem) {

        // TODO Auto-generated method stub

        return null;

    }

    @Override

    public void onComplete() {

        // TODO Auto-generated method stub

    }

    @Override

    protected Object getTrigger() {

        // TODO Auto-generated method stub

        return null;

    }

    @CanExecute

    private boolean canExecute(@Named(IServiceConstants.ACTIVE_SELECTION)TreeSelection selection, MHandledMenuItem menuItem){

        boolean canExecute = false;

        canExecute = selection.size()==1

                && selection.getFirstElement() instanceof TaxonNodeDto;

        menuItem.setEnabled(canExecute);

        return canExecute;

    }

}