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.ui.dialog.configurator;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.List;

import java.util.UUID;

import org.eclipse.jface.layout.GridLayoutFactory;

import org.eclipse.jface.viewers.ArrayContentProvider;

import org.eclipse.jface.viewers.CheckStateChangedEvent;

import org.eclipse.jface.viewers.CheckboxTableViewer;

import org.eclipse.jface.viewers.ICheckStateListener;

import org.eclipse.jface.viewers.LabelProvider;

import org.eclipse.swt.SWT;

import org.eclipse.swt.layout.GridData;

import org.eclipse.swt.layout.GridLayout;

import org.eclipse.swt.widgets.Button;

import org.eclipse.swt.widgets.Combo;

import org.eclipse.swt.widgets.Composite;

import org.eclipse.swt.widgets.Event;

import org.eclipse.swt.widgets.Label;

import org.eclipse.swt.widgets.Listener;

import eu.etaxonomy.cdm.api.application.CdmApplicationState;

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

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

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

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

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

import eu.etaxonomy.cdm.model.reference.OriginalSourceType;

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

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

import eu.etaxonomy.taxeditor.l10n.Messages;

import eu.etaxonomy.taxeditor.ui.combo.OriginalSourceTypeComparator;

/**

 * @author k.luther

 * @since 11.12.2019

 */

public class StructuredDescriptionAggregationConfigurationWizardPage extends AggregationConfigurationWizardPage<StructuredDescriptionAggregation> implements Listener {

    protected CheckboxTableViewer subTreeSelectionViewer;

    private List<TaxonNodeDto> selectedNodes;

    /**

     * @param pageName

     */

    protected StructuredDescriptionAggregationConfigurationWizardPage(StructuredDescriptionAggregationConfiguration configurator, List<TaxonNodeDto> nodes) {

        super(Messages.DistributionAggregationWizardPage_TITLE);

        this.configurator = configurator;

        this.setDescription(Messages.DistributionAggregationWizardPage_DESCRIPTION);

        selectedNodes = nodes;

    }

    @Override

    public void createControl(Composite parent) {

        final Composite control = new Composite(parent, SWT.NULL);

        GridLayout gridLayoutControl = new GridLayout();

        gridLayoutControl.numColumns = 2;

        control.setLayout(gridLayoutControl);

        control.setLayoutData(new GridData(SWT.FILL, SWT.LEFT, true, true));

        //select between taxonnode filter for the whole descriptive dataset or the selected taxonnode/subtree

        GridData gridDataRadioComposite = new GridData();

        gridDataRadioComposite.horizontalSpan = 2;

        GridLayout gridLayoutRadioComposite = new GridLayout();

        gridLayoutRadioComposite.numColumns = 1;

        Composite radioComposite = new Composite(control, SWT.NULL);

        radioComposite.setLayoutData(gridDataRadioComposite);

        radioComposite.setLayout(gridLayoutRadioComposite);

        checkUseSelectedSubtree= new Button(radioComposite,  SWT.RADIO);

        checkUseSelectedSubtree.setText(Messages.AggregationWizardPage_SUBTREE);

        checkUseSelectedSubtree.addListener(SWT.Selection, new Listener() {

            @Override

            public void handleEvent(Event e) {

                subTreeSelectionViewer.getTable().setEnabled(checkUseSelectedSubtree.getSelection());

                if (checkUseSelectedSubtree.getSelection() && subTreeSelectionViewer.getCheckedElements().length == 0){

                    setPageComplete(false);

                }else{

                    setPageComplete(true);

                }

           }

        });

        checkUseSelectedTaxonNode= new Button(radioComposite,  SWT.RADIO);

        checkUseSelectedTaxonNode.setText(Messages.StructuredDescriptionAggregationWizardPage_SELECT_SELECTED_TAXA_ONLY);

        checkUseSelectedTaxonNode.addListener(SWT.Selection, new Listener() {

            @Override

            public void handleEvent(Event e) {

                subTreeSelectionViewer.getTable().setEnabled(checkUseSelectedTaxonNode.getSelection());

                if (checkUseSelectedTaxonNode.getSelection() && subTreeSelectionViewer.getCheckedElements().length == 0){

                    setPageComplete(false);

                }else{

                    setPageComplete(true);

                }

            }

        });

//        configurator.getTaxonNodeFilter().getSubtreeFilter().stream().forEach(p ->uuidList.add(p.getUuid()));

//        List<String> propertyPaths = new ArrayList<>();

//        propertyPaths.add("taxon.name");

//        List<TaxonNode> nodeList = CdmApplicationState.getCurrentAppConfig().getTaxonNodeService().load(uuidList, propertyPaths);

        Label selectSubtreeLabel = new Label(control, SWT.NULL);

        selectSubtreeLabel.setText(Messages.StructuredDescriptionAggregationWizardPage_SELECT_SUBTREE);

        selectSubtreeLabel.setToolTipText(Messages.StructuredDescriptionAggregationWizardPage_TOOLTIP_SELECT_SUBTREE);

        subTreeSelectionViewer = CheckboxTableViewer.newCheckList(control, SWT.BORDER | SWT.SINGLE);

        subTreeSelectionViewer.setContentProvider(new ArrayContentProvider());

        subTreeSelectionViewer.setLabelProvider(new LabelProvider(){

            @Override

            public String getText(Object element){

                if (element instanceof TaxonNodeDto){

                   return ((TaxonNodeDto)element).getNameTitleCache();

                }else if (element instanceof TaxonNode){

                    return ((TaxonNode)element).getTaxon().getName().getTitleCache();

                }

                return null;

            }

        });

        if (selectedNodes == null || selectedNodes.isEmpty()){

            List<TaxonNode> nodeList;

            List<UUID> uuidList = new ArrayList<>();

            configurator.getTaxonNodeFilter().getSubtreeFilter().stream().forEach(p ->uuidList.add(p.getUuid()));

            List<String> propertyPaths = new ArrayList<>();

            propertyPaths.add("taxon.name");

            nodeList = CdmApplicationState.getCurrentAppConfig().getTaxonNodeService().load(uuidList, propertyPaths);

            subTreeSelectionViewer.setInput(nodeList);

        }else{

            subTreeSelectionViewer.setInput(selectedNodes);

        }

        subTreeSelectionViewer.setAllChecked(true);

        subTreeSelectionViewer.addCheckStateListener(new ICheckStateListener(){

            @Override

            public void checkStateChanged(    CheckStateChangedEvent event){

                Object[] checked =subTreeSelectionViewer.getCheckedElements();

                if (checked.length == 0){

                   setPageComplete(false);

                }else{

                    setPageComplete(true);

                }

            }

        });

        //use taxonnode filter of the descriptive data set

        checkUseHigherLevel = new Button(radioComposite,  SWT.RADIO);

        checkUseHigherLevel.setText(Messages.StructuredDescriptionAggregationWizardPage_SELECT_ALL_SUBTREES);

        checkUseHigherLevel.addListener(SWT.Selection, new Listener() {

           @Override

           public void handleEvent(Event e) {

               subTreeSelectionViewer.getTable().setEnabled(!checkUseHigherLevel.getSelection());

               updateHigherRankCombo();

               if (checkUseHigherLevel.getSelection()){

                   setPageComplete(true);

               }

           }

        });

        GridLayoutFactory.fillDefaults();

        Label higherRankLabel = new Label(control, SWT.NULL);

        higherRankLabel.setText(Messages.DistributionAggregationWizardPage_HIGHEST_RANK);

        comboHigherRank = new Combo(control, SWT.BORDER | SWT.READ_ONLY);

        updateHigherRankCombo();

        comboHigherRank.addListener(SWT.Selection, this);

        Label lowerRankLabel = new Label(control, SWT.NULL);

        lowerRankLabel.setText(Messages.DistributionAggregationWizardPage_LOWEST_RANK);

        comboLowerRank = new Combo(control, SWT.BORDER | SWT.READ_ONLY);

        updateLowerRankCombo();

        comboLowerRank.addListener(SWT.Selection, this);

        Label aggregationModeLabel = new Label(control, SWT.NULL);

        aggregationModeLabel.setText(Messages.DistributionAggregationWizardPage_AGGREGATION_MODE);

        aggregationModeLabel.setToolTipText(Messages.DistributionAggregationWizardPage_TOOLTIP_AGGR_MODE);

        aggregationModeViewer = CheckboxTableViewer.newCheckList(control, SWT.BORDER | SWT.SINGLE);

        aggregationModeViewer.setContentProvider(new ArrayContentProvider());

        aggregationModeViewer.setLabelProvider(new LabelProvider(){

            @Override

            public String getText(Object element){

                if (element instanceof AggregationMode){

                   return ((AggregationMode)element).getMessage();

                }

                return null;

            }

        });

        List<AggregationMode> aggregationModeList = new ArrayList<>();

        for (AggregationMode mode: AggregationMode.values()){

            aggregationModeList.add(mode);

        }

        aggregationModeViewer.setInput(aggregationModeList);

        aggregationModeViewer.addCheckStateListener(new ICheckStateListener(){

            @Override

            public void checkStateChanged(    CheckStateChangedEvent event){

                Object[] checked =aggregationModeViewer.getCheckedElements();

                boolean withinChecked = false;

                boolean toParentChecked = false;

                for (int i = 0; i<checked.length;i++){

                    if (checked[i] instanceof AggregationMode){

                        if (((AggregationMode)checked[i]).equals(AggregationMode.WithinTaxon)){

                            withinChecked = true;

                        }

                        if (((AggregationMode)checked[i]).equals(AggregationMode.ToParent)){

                            toParentChecked = true;

                        }

                    }

                }

                comboSourceModeChildParent.setEnabled(toParentChecked);

                comboSourceModeWithinTaxon.setEnabled(withinChecked);

//                AggregationSourceMode areaMode = (AggregationSourceMode)comboSourceModeSubAreaSuperArea.getData(comboSourceModeSubAreaSuperArea.getText());

                AggregationSourceMode taxonMode = (AggregationSourceMode)comboSourceModeChildParent.getData(comboSourceModeChildParent.getText());

//                getSourceTypeViewer().getTable().setEnabled(((areachecked && (areaMode.equals(AggregationSourceMode.ALL) || areaMode.equals(AggregationSourceMode.ALL_SAMEVALUE) ) )|| toParentChecked && (taxonMode.equals(AggregationSourceMode.ALL) || taxonMode.equals(AggregationSourceMode.ALL_SAMEVALUE) )) );

                getWizard().getContainer().updateButtons();

              }

        });

        Label sourceModeLabelWithin = new Label(control, SWT.NULL);

        sourceModeLabelWithin.setText(Messages.DistributionAggregationWizardPage_SOURCEMODE_WITHIN_TAXON);

        sourceModeLabelWithin.setToolTipText(Messages.DistributionAggregationWizardPage_TOOLTIP_SOURCEMODE_WITHIN_TAXON);

        comboSourceModeWithinTaxon = new Combo(control,  SWT.BORDER| SWT.READ_ONLY);

        comboSourceModeWithinTaxon.setText(Messages.DistributionAggregationWizardPage_AGGREGATION_MODE);

        for (AggregationSourceMode mode :AggregationSourceMode.list(AggregationMode.WithinTaxon, AggregationType.StructuredDescription) ){

            comboSourceModeWithinTaxon.add(mode.getMessage());

            comboSourceModeWithinTaxon.setData(mode.getMessage(), mode);

        }

        comboSourceModeWithinTaxon.setEnabled(false);

        comboSourceModeWithinTaxon.select(0);

        comboSourceModeWithinTaxon.addListener(SWT.Selection, this);

        Label sourceModeLabel = new Label(control, SWT.NULL);

        sourceModeLabel.setText(Messages.DistributionAggregationWizardPage_SOURCEMODE_CHILD_PARENT);

        sourceModeLabel.setToolTipText(Messages.DistributionAggregationWizardPage_TOOLTIP_SOURCEMODE_CHILD_PARENT);

        comboSourceModeChildParent = new Combo(control,  SWT.BORDER| SWT.READ_ONLY);

        comboSourceModeChildParent.setText(Messages.DistributionAggregationWizardPage_AGGREGATION_MODE);

        for (AggregationSourceMode mode :AggregationSourceMode.list(AggregationMode.ToParent, AggregationType.StructuredDescription) ){

            comboSourceModeChildParent.add(mode.getMessage());

            comboSourceModeChildParent.setData(mode.getMessage(), mode);

        }

        comboSourceModeChildParent.setEnabled(false);

        comboSourceModeChildParent.select(0);

        comboSourceModeChildParent.addListener(SWT.Selection, this);

        GridLayoutFactory.fillDefaults();

        Label sourceTypeLabel = new Label(control, SWT.NULL);

        sourceTypeLabel.setText(Messages.DistributionAggregationWizardPage_SOURCE_TYPE);

        sourceTypeLabel.setToolTipText(Messages.DistributionAggregationWizardPage_TOOLTIP_SOURCE_TYPE);

        sourceTypeViewer = CheckboxTableViewer.newCheckList(control, SWT.BORDER | SWT.SINGLE);

        sourceTypeViewer.setContentProvider(new ArrayContentProvider());

        sourceTypeViewer.addCheckStateListener(new ICheckStateListener(){

            @Override

            public void checkStateChanged(    CheckStateChangedEvent event){

//                checkedElements=sourceTypeViewer.getCheckedElements();

                getWizard().getContainer().updateButtons();

              }

        });

        List<String> typeStrings = new ArrayList<>();

        OriginalSourceType[] typeArray = OriginalSourceType.values();

        Arrays.sort(typeArray, new OriginalSourceTypeComparator(null));

        Arrays.stream(typeArray).forEach(p ->typeMap.put(p.getMessage(), p));

        Arrays.stream(typeArray).forEach(p ->typeStrings.add(p.getMessage()));

        typeStrings.remove(OriginalSourceType.NomenclaturalReference.getMessage());

        sourceTypeViewer.setInput(typeStrings);

        sourceTypeViewer.setChecked(OriginalSourceType.PrimaryMediaSource.getMessage(), true);

        sourceTypeViewer.setChecked(OriginalSourceType.PrimaryTaxonomicSource.getMessage(), true);

        sourceTypeViewer.getTable().setEnabled(false);

        setControl(control);

    }

    @Override

    public void handleEvent(Event event) {

        if (event.widget.equals(comboHigherRank)){

            updateLowerRankCombo();

        }

        if (event.widget.equals(this.checkUseHigherLevel)){

            updateHigherRankCombo();

            updateLowerRankCombo();

        }

        if (event.widget.equals(this.checkUseSelectedSubtree)){

            updateHigherRankCombo();

            updateLowerRankCombo();

        }

        if (event.widget.equals(this.checkUseSelectedTaxonNode)){

            updateHigherRankCombo();

            updateLowerRankCombo();

        }

        if (event.widget.equals(this.comboSourceModeChildParent) || event.widget.equals(this.comboSourceModeWithinTaxon)){

            if (getSourceModeChildParent().equals(AggregationSourceMode.NONE) && getSourceModeWithinTaxon().equals(AggregationSourceMode.NONE)){

                sourceTypeViewer.getTable().setEnabled(false);

            }else {

                sourceTypeViewer.getTable().setEnabled(true);

            }

        }

    }

    public List<TaxonNode> getSelectedTaxonNodes(){

        List<TaxonNode> result = new ArrayList<>();

        for (Object o: subTreeSelectionViewer.getCheckedElements()){

            if (o instanceof TaxonNode){

                result.add((TaxonNode)o);

            }

        }

        return result;

    }

}