EDIT

/**

* Copyright (C) 2017 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.editor.workingSet;

import java.util.Collection;

import java.util.HashSet;

import java.util.Set;

import javax.annotation.PostConstruct;

import javax.annotation.PreDestroy;

import javax.inject.Inject;

import javax.inject.Named;

import org.eclipse.core.runtime.IProgressMonitor;

import org.eclipse.e4.core.contexts.ContextInjectionFactory;

import org.eclipse.e4.core.contexts.IEclipseContext;

import org.eclipse.e4.ui.di.Focus;

import org.eclipse.e4.ui.di.Persist;

import org.eclipse.e4.ui.model.application.ui.MDirtyable;

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

import org.eclipse.swt.SWT;

import org.eclipse.swt.events.SelectionEvent;

import org.eclipse.swt.events.SelectionListener;

import org.eclipse.swt.widgets.Composite;

import org.eclipse.swt.widgets.Display;

import org.eclipse.swt.widgets.Shell;

import eu.etaxonomy.cdm.api.conversation.ConversationHolder;

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

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

import eu.etaxonomy.cdm.hibernate.HibernateProxyHelper;

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

import eu.etaxonomy.cdm.model.description.FeatureTree;

import eu.etaxonomy.cdm.model.description.SpecimenDescription;

import eu.etaxonomy.cdm.model.description.WorkingSet;

import eu.etaxonomy.cdm.model.location.NamedArea;

import eu.etaxonomy.cdm.model.name.Rank;

import eu.etaxonomy.cdm.model.occurrence.SpecimenOrObservationBase;

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

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

import eu.etaxonomy.taxeditor.store.CdmStore;

import eu.etaxonomy.taxeditor.ui.dialog.selection.WorkingSetSelectionDialog;

import eu.etaxonomy.taxeditor.ui.element.CdmFormFactory;

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

/**

 * @author pplitzner

 * @since Nov 21, 2017

 *

 */

public class WorkingSetEditor implements IE4SavablePart{

    private WorkingSetComposite composite;

    private ConversationHolder conversation;

    private WorkingSet workingSet;

    @Inject

    private MDirtyable dirty;

    @PostConstruct

    public void create(Composite parent, IEclipseContext context, @Named(IServiceConstants.ACTIVE_SHELL)Shell shell){

        if(CdmStore.isActive() && conversation==null){

            conversation = CdmStore.createConversation();

        }

        else{

            return;

        }

        CdmFormFactory cdmFormFactory = new CdmFormFactory(Display.getCurrent());

        ContextInjectionFactory.inject(cdmFormFactory, context);

        composite = new WorkingSetComposite(parent, SWT.NONE);

        //TODO remove with correct dirty handling

        dirty.setDirty(true);

        composite.getBtnOpenWorkingSet().addSelectionListener(new SelectionListener() {

            @Override

            public void widgetSelected(SelectionEvent e) {

                WorkingSetEditor.this.workingSet = WorkingSetSelectionDialog.select(shell, null);

                init();

            }

            @Override

            public void widgetDefaultSelected(SelectionEvent e) {

            }

        });

    }

    public void init() {

        if(workingSet.getTitleCache()!=null){

            composite.getTxt_label().setText(workingSet.getTitleCache());

        }

        if(workingSet.getDescriptiveSystem()!=null){

            composite.setCharacters(workingSet.getDescriptiveSystem());

        }

        Rank maxRank = workingSet.getMaxRank();

        if(maxRank!=null){

            composite.setRankMax(maxRank);

        }

        Rank minRank = workingSet.getMinRank();

        if(minRank!=null){

            composite.setRankMin(minRank);

        }

        Set<NamedArea> geoFilter = workingSet.getGeoFilter();

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

            composite.setArea(geoFilter.iterator().next());

        }

    }

    /**

     * {@inheritDoc}

     */

    @Persist

    @Override

    public void save(IProgressMonitor monitor) {

        NamedArea area = composite.getArea();

        Set<NamedArea> areas = new HashSet<>();

        if(area!=null){

            areas.add(area);

        }

        TaxonNode taxonNode = composite.getTaxonNode();

        Set<NamedArea> descriptions;

        if(taxonNode!=null && taxonNode.getTaxon()!=null){

            Taxon taxon = HibernateProxyHelper.deproxy(taxonNode.getTaxon(), Taxon.class);

            Collection<SpecimenOrObservationBase> specimens = CdmStore.getService(IOccurrenceService.class).listByAssociatedTaxon(SpecimenOrObservationBase.class, null, taxon, null, null, null, null, null);

            for (SpecimenOrObservationBase specimen : specimens) {

                Set<SpecimenDescription> specimenDescriptions = specimen.getSpecimenDescriptions();

                for (SpecimenDescription specimenDescription : specimenDescriptions) {

                    workingSet.addDescription(specimenDescription);

                }

            }

        }

        FeatureTree characters = composite.getCharacters();

        DefinedTermBase rankMaxSelection = composite.getRankMax().getSelection();

        Rank rankMax = null;

        if(rankMaxSelection instanceof Rank){

            rankMax = (Rank) rankMaxSelection;

        }

        DefinedTermBase rankMinSelection = composite.getRankMin().getSelection();

        Rank rankMin = null;

        if(rankMinSelection instanceof Rank){

            rankMin = (Rank) rankMinSelection;

        }

        workingSet.setMaxRank(rankMax);

        workingSet.setMinRank(rankMin);

        workingSet.setDescriptiveSystem(characters);

        workingSet.addTaxonSubtree(taxonNode);

        workingSet.setGeoFilter(areas);

        conversation.commit();

        CdmStore.getService(IWorkingSetService.class).merge(workingSet, true);

        dirty.setDirty(false);

    }

    @Focus

    public void setFocus(){

        if(conversation!=null){

            conversation.bind();

        }

    }

    @PreDestroy

    public void dispose(){

        if(conversation!=null){

            conversation.close();

            conversation = null;

        }

    }

}