EDIT

/**

 * Copyright (C) 2014 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.preference.wizard;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.Collection;

import java.util.HashMap;

import java.util.HashSet;

import java.util.List;

import java.util.Map;

import java.util.Set;

import java.util.UUID;

import org.apache.commons.lang.StringUtils;

import org.eclipse.jface.preference.IPreferenceStore;

import org.eclipse.jface.viewers.CheckStateChangedEvent;

import org.eclipse.jface.viewers.CheckboxTreeViewer;

import org.eclipse.jface.viewers.ICheckStateListener;

import org.eclipse.jface.viewers.ITreeViewerListener;

import org.eclipse.jface.viewers.StyledString.Styler;

import org.eclipse.jface.viewers.TreeExpansionEvent;

import org.eclipse.swt.SWT;

import org.eclipse.swt.graphics.TextStyle;

import org.eclipse.swt.layout.GridData;

import org.eclipse.swt.layout.GridLayout;

import org.eclipse.swt.widgets.Composite;

import org.eclipse.swt.widgets.Display;

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

import eu.etaxonomy.cdm.model.common.CdmBase;

import eu.etaxonomy.cdm.model.common.DefinedTermBase;

import eu.etaxonomy.cdm.model.common.TermType;

import eu.etaxonomy.cdm.model.common.TermVocabulary;

import eu.etaxonomy.cdm.model.metadata.PreferencePredicate;

import eu.etaxonomy.taxeditor.l10n.Messages;

import eu.etaxonomy.taxeditor.preference.PreferencesUtil;

import eu.etaxonomy.taxeditor.session.ICdmEntitySession;

import eu.etaxonomy.taxeditor.session.ICdmEntitySessionEnabled;

import eu.etaxonomy.taxeditor.store.CdmStore;

/**

 * @author a.oppermann

 * @date 21.07.2014

 *

 */

public class AvailableDistributionPage extends AbstractTermSelectionWizardPage implements ICdmEntitySessionEnabled {

    /**

     * @param pageName

     */

    protected AvailableDistributionPage(String pageName) {

        super(pageName, TermType.NamedArea);

        this.localPref = true;

    }

    /** {@inheritDoc} */

    @Override

    public void createControl(Composite parent) {

        IPreferenceStore preferenceStore = PreferencesUtil.getPreferenceStore();

        String checkedValues = preferenceStore.getString(PreferencesUtil.DISTRIBUTION_AREA_OCCURENCE_STATUS);

        String grayedValues = preferenceStore.getString(PreferencesUtil.DISTRIBUTION_AREA_OCCURENCE_STATUS_GRAYED);

        setTitle(Messages.AvailableDistributionPage_PAGE_TITLE);

        setDescription(Messages.AvailableDistributionPage_PAGE_DESCRIPTION);

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

        composite.setLayout(new GridLayout());

        setViewer(new CheckboxTreeViewer(composite, SWT.NULL));

        ((CheckboxTreeViewer)getViewer()).getTree().setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true));

        getViewer().setContentProvider(new DistributionContentProvider());

        Styler styler = new Styler() {

            @Override

            public void applyStyles(TextStyle textStyle) {

                textStyle.foreground = Display.getCurrent().getSystemColor(SWT.COLOR_BLACK);

            }

        };

        getViewer().setLabelProvider(new AreaLabelProvider(styler));

        getViewer().setComparator(new AreaViewerComparator());

        ((CheckboxTreeViewer)getViewer()).addCheckStateListener(new ICheckStateListener() {

            private boolean ignoreCheckEvent = false;

            @Override

            public void checkStateChanged(CheckStateChangedEvent event) {

                if (ignoreCheckEvent ) {

                    return;

                }

                ignoreCheckEvent = true;

                try {

                    checkChildren(event.getElement(), event.getChecked());

                    grayParents(event.getElement(), event.getChecked());

                   // ((CheckboxTreeViewer)getViewer()).refresh();

                }

                finally {

                    ignoreCheckEvent = false;

                }

                checkNoneChecked();

            }

        });

        ((CheckboxTreeViewer)getViewer()).addTreeListener(new ITreeViewerListener() {

            @Override

            public void treeCollapsed(TreeExpansionEvent event) {

            }

            @Override

            public void treeExpanded(TreeExpansionEvent event) {

                final Object element = event.getElement();

                final Object[] children = ((DistributionContentProvider)getViewer().getContentProvider()).getChildren(element);

                for (Object child : children) {

                    if (child instanceof NamedAreaWrapper){

                        if (getListCheckedTerms().contains(((NamedAreaWrapper) child).getNamedArea()) ){

                            ((CheckboxTreeViewer)getViewer()).setChecked(child, true);

                        }

                        if (getListGrayedTerms().contains(((NamedAreaWrapper) child).getNamedArea())&& !((NamedAreaWrapper) child).isBaseElement){

                            ((CheckboxTreeViewer)getViewer()).setGrayChecked(child, true);

                        }

                    }

                }

            }

        });

        // TODO: write listener or toggle button to select all named areas

        rememberCheckedValues(checkedValues, grayedValues);

       // ((CheckboxTreeViewer)getViewer()).expandToLevel(2);

        setControl(composite);

    }

    private boolean checkNoneChecked() {

        if (((CheckboxTreeViewer)getViewer()).getCheckedElements().length == 0) {

            setMessage(Messages.AvailableDistributionPage_CHECK_MESSAGE, WARNING);

            return true;

        } else {

            setMessage(null);

            return false;

        }

    }

    /**

    *

    * @param path

    * @param checked

    */

   private void checkChildren(final Object element, final boolean checked) {

       if (element == null) {

           return;

       }

       if (((CheckboxTreeViewer)getViewer()).isExpandable(element)){

           ((CheckboxTreeViewer)getViewer()).expandToLevel(element, CheckboxTreeViewer.ALL_LEVELS);

           ((CheckboxTreeViewer)getViewer()).setSubtreeChecked(element, checked);

           ((CheckboxTreeViewer)getViewer()).setChecked(element, checked);

           if (((CheckboxTreeViewer)getViewer()).getGrayed(element)){

               ((CheckboxTreeViewer)getViewer()).setGrayChecked(element, !checked);

           }

//           if (element instanceof NamedAreaWrapper){

//               NamedAreaWrapper baseElement = new NamedAreaWrapper(((NamedAreaWrapper)element).getNamedArea(), true, ((NamedAreaWrapper)element).parent);

//               ((CheckboxTreeViewer)getViewer()).setChecked(baseElement, !checked);

//               ((CheckboxTreeViewer)getViewer()).setGrayed(element, checked);

//           }

       }else{

           ((CheckboxTreeViewer)getViewer()).setChecked(element, checked);

           ((CheckboxTreeViewer)getViewer()).setGrayed(element, false);

       }

   }

   /**

   *

   * @param path

   * @param checked

   */

  private void grayParents(final Object element, boolean checked) {

      DistributionContentProvider tcp = new DistributionContentProvider();

      if (element == null) {

          return;

      }

   //   final Object element = path.getLastSegment();

      Object parent = tcp.getParent(element);

      if (parent != null) {

          boolean allUnchecked = true;

          boolean allChecked = true;

          //if element is checked then the parent should be grayed if it is unchecked it have to be computed whether the parent should be checked

//          if (checked){

//              if (!((CheckboxTreeViewer)getViewer()).getChecked(parent)){

//                  ((CheckboxTreeViewer)getViewer()).setGrayChecked(parent, true);

//              }

//

//          }else{

              for (final Object fieldElement : tcp.getChildren(parent)) {

                  allChecked = allChecked && ((CheckboxTreeViewer)getViewer()).getChecked(fieldElement);

                  allUnchecked = allUnchecked && !((CheckboxTreeViewer)getViewer()).getChecked(fieldElement);

                  allUnchecked = allUnchecked && !((CheckboxTreeViewer)getViewer()).getGrayed(fieldElement);

              }

              if (allUnchecked) {

//                  if (!((CheckboxTreeViewer)getViewer()).getChecked(parent)){

                      ((CheckboxTreeViewer)getViewer()).setChecked(parent, false);

                      ((CheckboxTreeViewer)getViewer()).setGrayed(parent, false);

//                  }else{

//                      checked = true;

//                  }

              } else if (allChecked){

                  ((CheckboxTreeViewer)getViewer()).setChecked(parent, true);

                  ((CheckboxTreeViewer)getViewer()).setGrayed(parent, false);

              }else {

//                if (!((CheckboxTreeViewer)getViewer()).getChecked(parent)){

                    ((CheckboxTreeViewer)getViewer()).setGrayChecked(parent, true);

                    ((CheckboxTreeViewer)getViewer()).setChecked(parent, true);

//                }

                checked = true;

              }

//          }

        grayParents(parent, checked);

      }

  }

    @Override

    public void dispose() {

        CdmStore.getCurrentSessionManager().dispose(this);

        super.dispose();

    }

    /* (non-Javadoc)

     * @see eu.etaxonomy.taxeditor.session.ICdmEntitySessionEnabled#getCdmEntitySession()

     */

    @Override

    public ICdmEntitySession getCdmEntitySession() {

        return CdmStore.getCurrentSessionManager().getNullSession();

    }

    /* (non-Javadoc)

     * @see eu.etaxonomy.taxeditor.session.ICdmEntitySessionEnabled#getRootEntities()

     */

    @Override

    public <T extends CdmBase> Collection<T> getRootEntities() {

        return null;

    }

    /* (non-Javadoc)

     * @see eu.etaxonomy.taxeditor.session.ICdmEntitySessionEnabled#getPropertyPathsMap()

     */

    @Override

    public Map<Object, List<String>> getPropertyPathsMap() {

        Map<Object, List<String>> propertyPathsMap = new HashMap<Object, List<String>>();

        List<String> termsPropertyPaths = Arrays.asList(new String[] {

                "includes" //$NON-NLS-1$

        });

        propertyPathsMap.put("includes", termsPropertyPaths); //$NON-NLS-1$

        propertyPathsMap.put("terms", termsPropertyPaths); //$NON-NLS-1$

        return propertyPathsMap;

    }

    @Override

    protected  List<TermVocabulary<DefinedTermBase>> getVocabulariesFromPreference(){

        List<TermVocabulary<DefinedTermBase>> vocs = new ArrayList();

        if (PreferencesUtil.getPreferenceFromDB(PreferencePredicate.AvailableDistributionAreaVocabularies) == null && PreferencesUtil.getPreferenceStore().getString(PreferencesUtil.DISTRIBUTION_VOCABULARIES) == null){

            vocs = CdmStore.getService(IVocabularyService.class).findByTermType(

                    type, null);

        }else{

            String vocString = PreferencesUtil.getPreferenceStore().getString(PreferencesUtil.DISTRIBUTION_VOCABULARIES);

            String[] arrayVocs = vocString.split(";"); //$NON-NLS-1$

            Set<UUID> uuidVocs = new HashSet();

            for (String voc: arrayVocs){

                if (!StringUtils.isBlank(voc)){

                    uuidVocs.add(UUID.fromString(voc));

                }

            }

            List<TermVocabulary> tempVocs = CdmStore.getService(IVocabularyService.class).find(uuidVocs);

            for (TermVocabulary voc: tempVocs){

                vocs.add(voc);

            }

        }

        return vocs;

    }

}