EDIT

/**

* Copyright (C) 2018 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.List;

import java.util.UUID;

import org.apache.commons.lang.StringUtils;

import org.eclipse.jface.viewers.CheckboxTreeViewer;

import org.eclipse.jface.wizard.WizardPage;

import org.eclipse.swt.SWT;

import org.eclipse.swt.layout.GridData;

import org.eclipse.swt.layout.GridLayout;

import org.eclipse.swt.widgets.Composite;

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

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

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

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

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

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

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

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

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

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

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

import eu.etaxonomy.taxeditor.store.CdmStore;

/**

 * @author k.luther

 * @since 04.06.2018

 *

 */

public abstract class AbstractTermSelectionWizardPage extends WizardPage {

    protected CheckBoxTreeComposite treeComposite;

    private List<TermVocabularyDto> vocabularies = new ArrayList<>();

    boolean localPref;

    protected TermType type;

    protected AbstractTermSelectionWizardPage(String pageName, TermType type) {

        super(pageName);

        // TODO check if configuration exists

       // CdmStore.getCurrentSessionManager().bindNullSession();

        this.type = type;

    }

    @Override

    public void createControl(Composite parent){

        parent.setLayout(new GridLayout());

        treeComposite = new CheckBoxTreeComposite(parent, new TermDtoContentProvider(), new TermDtoLabelProvider(), SWT.NONE);

        treeComposite.getViewer().setComparator(new DefinedTermSorter());

        rememberCheckedValues(getCheckedValuesFromPreferences());

        treeComposite.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true, 1, 1));

        setControl(treeComposite);

    }

    protected abstract String getCheckedValuesFromPreferences();

    protected CheckboxTreeViewer getViewer(){

        return treeComposite.getViewer();

    }

    public List<TermVocabularyDto> getVocabularies() {

        return vocabularies;

    }

    public void addVocabularies(TermVocabularyDto vocabulary) {

        this.vocabularies.add(vocabulary);

    }

    protected void setVocabularies(List<TermVocabularyDto> vocs) {

        vocabularies = vocs;

    }

    protected void rememberCheckedValues(String checkedValues) {

        initialiseVocabularies();

        treeComposite.getViewer().setInput(getVocabularies());

        if (checkedValues != null && checkedValues != "") { //$NON-NLS-1$

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

            String[] listCheckedComma = checkedValues.split(","); //$NON-NLS-1$

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

            if (listChecked != null ){

                checked = Arrays.asList(listChecked);

            }

            if (listCheckedComma != null && checkedValues.contains(",")){ //$NON-NLS-1$

                checked = Arrays.asList(listCheckedComma);

            }

            List<AbstractTermDto> termsFromStringValues = getTermsFromStringValues(checked);

            treeComposite.setCheckedElements(termsFromStringValues.toArray());

        }else{

            List<AbstractTermDto> termsFromStringValues = getTermsFromStringValues(new ArrayList<String>());

            treeComposite.setCheckedElements(termsFromStringValues.toArray());

        }

    }

    protected List<AbstractTermDto> getTermsFromStringValues(List<String> listValue) {

        List<AbstractTermDto> termlist = new ArrayList<>();

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

        for (String s : listValue) {

            if (!StringUtils.isBlank(s)){

                UUID uuid = UUID.fromString(s);

                DefinedTermBase definedTermBase = termService.load(uuid);

                if(definedTermBase != null){

                    termlist.add(TermDto.fromTerm(definedTermBase, true));

                }else{

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

                    TermVocabulary termVocabulary = vocabularyService.load(uuid);

                    if (termVocabulary != null){

                        termlist.add(new TermVocabularyDto(uuid, termVocabulary.getRepresentations(), termVocabulary.getTermType()));

                    }

                }

            }

        }

        if (listValue.isEmpty()){

            List<DefinedTermBase> terms = CdmStore.getTermManager().getAllTerms(type, null);

            for (DefinedTermBase term: terms){

                termlist.add(TermDto.fromTerm(term, true));

            }

        }

        return termlist;

    }

    protected void initialiseVocabularies() {

        if (getVocabularies() != null) {

            getVocabularies().clear();

        }

        List<TermVocabularyDto> vocs = new ArrayList<>();

        vocs = CdmStore.getService(IVocabularyService.class).findVocabularyDtoByTermType(type);

        setVocabularies(vocs);

    }

    protected abstract List<TermVocabularyDto> getVocabulariesFromPreference();

}