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

import java.util.Map;

import java.util.Set;

import java.util.UUID;

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

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

import org.eclipse.jface.viewers.TreePath;

import org.eclipse.jface.wizard.WizardPage;

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

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.taxeditor.editor.definedterm.TermContentProvider;

import eu.etaxonomy.taxeditor.editor.definedterm.TermLabelProvider;

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 WizardPage implements ICdmEntitySessionEnabled {

    private CheckboxTreeViewer viewer;

    private Set<TermVocabulary<DefinedTermBase>> vocabularies;

    /**

     * @param pageName

     */

    protected AvailableDistributionPage(String pageName) {

        super(pageName);

        // TODO check if configuration exists

        CdmStore.getCurrentSessionManager().bindNullSession();

    }

    /*

     * (non-Javadoc)

     *

     * @see

     * org.eclipse.jface.dialogs.IDialogPage#createControl(org.eclipse.swt.widgets

     * .Composite)

     */

    /** {@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("Select areas for Distribution Editor");

        setDescription("In order to be able to modify and see the distribution status of taxa,\n"

                + "you have to select the areas which you like to see.");

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

        composite.setLayout(new GridLayout());

        viewer = new CheckboxTreeViewer(composite, SWT.NULL);

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

        viewer.setContentProvider(new TermContentProvider());

        Styler styler = new Styler() {

            @Override

            public void applyStyles(TextStyle textStyle) {

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

            }

        };

        viewer.setLabelProvider(new TermLabelProvider(styler));

        viewer.addCheckStateListener(new ICheckStateListener() {

            private boolean ignoreCheckEvent = false;

            @Override

            public void checkStateChanged(CheckStateChangedEvent event) {

                if (ignoreCheckEvent ) {

                    return;

                }

                ignoreCheckEvent = true;

                try {

                    TreePath path  = ((ITreeSelection) viewer.getSelection()).getPaths()[0];

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

                    checkParents(path.getParentPath());

                }

                finally {

                    ignoreCheckEvent = false;

                }

                checkNoneChecked();

            }

        });

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

        rememberCheckedValues(checkedValues, grayedValues);

        setControl(composite);

    }

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

        if (vocabularies != null) {

            vocabularies.clear();

        }

        List<TermVocabulary<DefinedTermBase>> vocs = CdmStore.getService(IVocabularyService.class).findByTermType(

                TermType.NamedArea, null);

        return vocs;

    }

    private void rememberCheckedValues(String checkedValues, String grayedValues) {

        viewer.setInput(initialiseVocabularies());

        if (grayedValues != null && grayedValues != "") {

            String[] listGrayed = grayedValues.split(",");

            ArrayList listGrayedTerms = new ArrayList();

            getTermsFromStringValues(listGrayed, listGrayedTerms);

            for(Object element : listGrayedTerms){

                if(element != null){

                    viewer.setGrayChecked(element, true);

                }

            }

        }

        if (checkedValues != null && checkedValues != "") {

            String[] listChecked = checkedValues.split(",");

            ArrayList<DefinedTermBase<?>> listCheckedTerms = new ArrayList<DefinedTermBase<?>>();

            getTermsFromStringValues(listChecked, listCheckedTerms);

            for(Object element : listCheckedTerms){

                if(element != null){

                    viewer.setChecked(element, true);

                }

            }

        }

    }

    /**

     * @param split

     * @param termlist

     */

    private void getTermsFromStringValues(String[] split, ArrayList termlist) {

        List<String> listValue = Arrays.asList(split);

        for (String s : listValue) {

            UUID uuid = UUID.fromString(s);

            ITermService termService = CdmStore.getService(ITermService.class);

            DefinedTermBase definedTermBase = termService.load(uuid);

            if(definedTermBase != null){

                termlist.add(definedTermBase);

            }else{

                IVocabularyService vocabularyService = CdmStore.getService(IVocabularyService.class);

                TermVocabulary termVocabulary = vocabularyService.load(uuid);

                termlist.add(termVocabulary);

            }

        }

    }

    private boolean checkNoneChecked() {

        if (viewer.getCheckedElements().length == 0) {

            setMessage("Please check at least one item", WARNING);

            return true;

        } else {

            setMessage(null);

            return false;

        }

    }

    /**

     * @return the viewer

     */

    public CheckboxTreeViewer getViewer() {

        return viewer;

    }

    /**

     *

     * @param path

     */

    private void checkParents(final TreePath path) {

        if (path == null) {

            return;

        }

        TermContentProvider tcp = new TermContentProvider();

        Object treeElement = path.getLastSegment();

        if (treeElement != null) {

            boolean allChecked = true;

            boolean allUnchecked = true;

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

                allChecked = allChecked && this.viewer.getChecked(fieldElement);

                allUnchecked = allUnchecked && !this.viewer.getChecked(fieldElement);

            }

            if (allUnchecked) {

                this.viewer.setChecked(treeElement, false);

                this.viewer.setGrayed(treeElement, false);

            } else if (allChecked) {

                this.viewer.setChecked(treeElement, true);

                this.viewer.setGrayed(treeElement, false);

            } else if (!allUnchecked && !allChecked) {

                this.viewer.setChecked(treeElement, true);

                this.viewer.setGrayed(treeElement, true);

            }

        }

        checkParents(path.getParentPath());

    }

//    /**

//     *

//     * @param path

//     * @param checked

//     */

//    private void checkChildren(final TreePath path, final boolean checked) {

//        TermContentProvider tcp = new TermContentProvider();

//

//        if (path == null) {

//            return;

//        }

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

//        if (element != null) {

//            this.viewer.setChecked(element, checked);

//            this.viewer.setGrayed(element, false);

//            int length = tcp.getChildren(element).length;

//            if(length>1){

//                this.viewer.expandToLevel(element, CheckboxTreeViewer.ALL_LEVELS);

//                this.viewer.setSubtreeChecked(element, checked);

//            }

//        }

//    }

    /**

    *

    * @param path

    * @param checked

    */

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

       TermContentProvider tcp = new TermContentProvider();

       if (element == null) {

           return;

       }

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

       if (element != null) {

           this.viewer.setChecked(element, checked);

           this.viewer.setGrayed(element, false);

           int length = tcp.getChildren(element).length;

           if(length>1){

               this.viewer.expandToLevel(element, CheckboxTreeViewer.ALL_LEVELS);

               this.viewer.setSubtreeChecked(element, 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"

        });

        propertyPathsMap.put("includes", termsPropertyPaths);

        propertyPathsMap.put("terms", termsPropertyPaths);

        return propertyPathsMap;

    }

}