EDIT

package eu.etaxonomy.cdm.vaadin.presenter.dbstatus;

import java.sql.SQLException;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.Collection;

import java.util.Collections;

import java.util.HashMap;

import java.util.HashSet;

import java.util.List;

import java.util.Set;

import java.util.UUID;

import com.vaadin.data.util.BeanItemContainer;

import com.vaadin.server.VaadinSession;

import com.vaadin.ui.ComboBox;

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

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

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

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

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

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

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

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

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

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

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

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

import eu.etaxonomy.cdm.vaadin.container.CdmSQLContainer;

import eu.etaxonomy.cdm.vaadin.model.CdmTaxonTableCollection;

import eu.etaxonomy.cdm.vaadin.model.DbTableDTO;

import eu.etaxonomy.cdm.vaadin.model.DbTableDTOS;

import eu.etaxonomy.cdm.vaadin.model.LazyLoadedContainer;

import eu.etaxonomy.cdm.vaadin.util.CdmQueryFactory;

import eu.etaxonomy.cdm.vaadin.util.CdmSpringContextHelper;

import eu.etaxonomy.cdm.vaadin.view.dbstatus.DistributionTableView;

import eu.etaxonomy.cdm.vaadin.view.dbstatus.IDistributionTableComponent;

public class DistributionTablePresenter implements IDistributionTableComponent.DistributionTableComponentListener{

	private final IClassificationService classificationService;

	private final IVocabularyService vocabularyService;

	private final IDescriptionService descriptionService;

	private final ITaxonNodeService taxonNodeService;

	private final ITermService termService;

	private final DistributionTableView view;

	private final ITaxonService taxonService;

	public DistributionTablePresenter(DistributionTableView dtv) throws SQLException{

	    this.view = dtv;

	    view.addListener(this);

	    taxonService = CdmSpringContextHelper.getTaxonService();

	    classificationService = CdmSpringContextHelper.getClassificationService();

	    taxonNodeService = CdmSpringContextHelper.getTaxonNodeService();

		vocabularyService = CdmSpringContextHelper.getVocabularyService();

		descriptionService = CdmSpringContextHelper.getDescriptionService();

		termService = CdmSpringContextHelper.getTermService();

		view.dataBinding();

	}

	@Override

    public ComboBox updateDistributionField(DescriptionElementBase deb,

			Distribution db,

			BeanItemContainer<PresenceAbsenceTerm> termContainer, ComboBox box,

			Taxon taxon) {

		// TODO Auto-generated method stub

		return null;

	}

	@Override

	public Set<DefinedTermBase> getChosenTerms() {

		VaadinSession session = VaadinSession.getCurrent();

		UUID termUUID = (UUID) session.getAttribute("selectedTerm");

		TermVocabulary<DefinedTermBase> term = vocabularyService.load(termUUID);

		term = CdmBase.deproxy(term, TermVocabulary.class);

		return term.getTerms();

	}

	public List<String> getTermList() {

		VaadinSession session = VaadinSession.getCurrent();

		UUID termUUID = (UUID) session.getAttribute("selectedTerm");

		TermVocabulary<DefinedTermBase> term = vocabularyService.load(termUUID);

		term = CdmBase.deproxy(term, TermVocabulary.class);

		Set<DefinedTermBase> terms = term.getTerms();

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

		for(DefinedTermBase dtb: terms){

			list.add(dtb.getTitleCache());

		}

		Collections.sort(list);

		return list;

	}

	@Override

	public HashMap<DescriptionElementBase, Distribution> getDistribution(DefinedTermBase dt, Taxon taxon) {

		Set<Feature> setFeature = new HashSet<Feature>(Arrays.asList(Feature.DISTRIBUTION()));

		List<DescriptionElementBase> listTaxonDescription = descriptionService.listDescriptionElementsForTaxon(taxon, setFeature, null, null, null, DESCRIPTION_INIT_STRATEGY);

		HashMap<DescriptionElementBase, Distribution> map = null;

		for(DescriptionElementBase deb : listTaxonDescription){

			if(deb instanceof Distribution){

				Distribution db = (Distribution)deb;

				String titleCache = dt.getTitleCache();

				if(db.getArea().getTitleCache().equalsIgnoreCase(titleCache)){

					map = new HashMap<DescriptionElementBase, Distribution>();

					map.put(deb, db);

				}

			}

		}

		return map;

	}

	@Override

	public List<DescriptionElementBase> listDescriptionElementsForTaxon(Taxon taxon, Set<Feature> setFeature){

		return descriptionService.listDescriptionElementsForTaxon(taxon, setFeature, null, null, null, DESCRIPTION_INIT_STRATEGY);

	}

	@Override

	public List<Distribution> getDistribution(Taxon taxon) {

		Set<Feature> setFeature = new HashSet<Feature>(Arrays.asList(Feature.DISTRIBUTION()));

		List<Distribution> listTaxonDescription = descriptionService.listDescriptionElementsForTaxon(taxon, setFeature, null, null, null, DESCRIPTION_INIT_STRATEGY);

		return listTaxonDescription;

	}

	@Override

	public List<TaxonNode> getAllNodes(int start, int end){

		Classification classification = getChosenClassification();

		List<TaxonNode> nodesForClassification = taxonNodeService.listAllNodesForClassification(classification, start, end);

		return nodesForClassification;

	}

	@Override

	public Classification getChosenClassification() {

		VaadinSession session = VaadinSession.getCurrent();

		UUID classificationUUID = (UUID) session.getAttribute("classificationUUID");

		Classification classification = classificationService.load(classificationUUID);

		return classification;

	}

	@Override

	public int getSizeOfClassification(){

		Classification classification = getChosenClassification();

		return taxonNodeService.countAllNodesForClassification(classification);

	}

	@Override

	public DbTableDTOS getDataList(int start, int end){

		List<TaxonNode> nodes = getAllNodes(start, end);

		DbTableDTOS items = new DbTableDTOS();

		for(TaxonNode tn: nodes){

			Taxon taxon = tn.getTaxon();

			DbTableDTO dbTableDTO = new DbTableDTO(taxon);

			Set<DefinedTermBase> terms = getChosenTerms();

			List<Distribution> distribution = getDistribution(taxon);

			for(DefinedTermBase dt: terms){

				for(Distribution db : distribution){

					if(dt.getTitleCache().equalsIgnoreCase(db.getArea().getTitleCache())){

//						DistributionDTO distributionDTO = new DistributionDTO(db.getStatus().getTitleCache());

//						dbTableDTO.setdDTO(distributionDTO);

					}

				}

			}

			items.add(dbTableDTO);

		}

		return items;

	}

	@Override

	public CdmSQLContainer getSQLContainer() throws SQLException{

		Classification classification = getChosenClassification();

		int classificationId = classification.getId();

		List<String> termList = getTermList();

		CdmSQLContainer container = new CdmSQLContainer(CdmQueryFactory.generateTaxonDistributionQuery(termList, classificationId));

		return container;

	}

	@Override

	public LazyLoadedContainer getLazyLoadedContainer(){

		LazyLoadedContainer lz = new LazyLoadedContainer(CdmTaxonTableCollection.class);

	    lz.addListener(this);

		return lz;

	}

	@Override

	public List<PresenceAbsenceTerm> getPresenceAbsenceTerms() {

		//TODO Better to use TermType instead of class to get the list

		return termService.list(PresenceAbsenceTerm.class, null, null, null, DESCRIPTION_INIT_STRATEGY);

	}

	protected static final List<String> DESCRIPTION_INIT_STRATEGY = Arrays.asList(new String []{

            "$",

            "elements.*",

            "elements.sources.citation.authorship.$",

            "elements.sources.nameUsedInSource.originalNameString",

            "elements.area.level",

            "elements.modifyingText",

            "elements.states.*",

            "elements.media",

            "elements.multilanguageText",

            "multilanguageText",

            "stateData.$"

    });

	@Override

	public IClassificationService getClassificationService() {

		return classificationService;

	}

	@Override

	public IVocabularyService getVocabularyService() {

		return vocabularyService;

	}

	@Override

	public IDescriptionService getDescriptionService() {

		return descriptionService;

	}

	@Override

	public ITaxonNodeService getTaxonNodeService() {

		return taxonNodeService;

	}

	@Override

	public ITermService getTermService() {

		return termService;

	}

	@Override

	public LazyLoadedContainer getTableContainer() {

		// TODO Auto-generated method stub

		return null;

	}

	@Override

	public ITaxonService getTaxonService() {

		return taxonService;

	}

}