EDIT

// $Id$

/**

* Copyright (C) 2016 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.cdm.io.mexico;

import java.net.URI;

import java.util.HashMap;

import java.util.Map;

import java.util.UUID;

import org.apache.commons.lang3.StringUtils;

import org.apache.log4j.Logger;

import org.springframework.stereotype.Component;

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

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

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

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

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

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

import eu.etaxonomy.cdm.model.reference.Reference;

import eu.etaxonomy.cdm.model.reference.ReferenceFactory;

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

/**

 * @author a.mueller

 * @date 16.06.2016

 *

 */

@Component

public class MexicoConabioCommonNamesImport<CONFIG extends MexicoConabioImportConfigurator>

            extends SimpleExcelTaxonImport<CONFIG>{

    private static final long serialVersionUID = 3579868489510261569L;

    private static final Logger logger = Logger.getLogger(MexicoConabioCommonNamesImport.class);

    private TermVocabulary<Language> languagesVoc;

    private Map<String, Language> languagesMap = new HashMap<>();

    @Override

    protected void firstPass(SimpleExcelTaxonImportState<CONFIG> state) {

        initLanguageVocabulary(state);

        String line = state.getCurrentLine() + ": ";

        HashMap<String, String> record = state.getOriginalRecord();

        String idCat = getValue(record, "IdCAT");

        Taxon taxon = state.getHigherTaxon(idCat);

        if (taxon == null){

            logger.warn(line + "Taxon could not be found: " + idCat);

        }else{

            TaxonDescription desc = getTaxonDescription(taxon);

            String nomComunStr = getValue(record, "NomComun");

            String langStr = getValue(record, "Lengua");

            Language language = languagesMap.get(langStr);

            if (language == null){

                logger.warn("Language not found: " + langStr);

            }

            String refStr = getValue(record, "ReferenciasNombreComun");

            CommonTaxonName commonName = CommonTaxonName.NewInstance(nomComunStr,

                    language, null);

            Reference ref = getReference(state, refStr);

            if (ref != null){

                commonName.addSource(OriginalSourceType.PrimaryTaxonomicSource,

                        null, null, ref, null);

            }

            getTaxonService().save(taxon);

        }

    }

    /**

     * @param state

     */

    @SuppressWarnings("unchecked")

    private void initLanguageVocabulary(SimpleExcelTaxonImportState<CONFIG> state) {

        if (languagesVoc == null){

            languagesVoc = this.getVocabularyService().find(MexicoConabioTransformer.uuidMexicanLanguagesVoc);

            if (languagesVoc == null){

                createLanguagesVoc(state);

            }

        }

    }

    /**

     * @param state

     * @return

     */

    private void createLanguagesVoc(SimpleExcelTaxonImportState<CONFIG> state) {

        URI termSourceUri = null;

        String label = "Mexican States";

        String description = "Mexican States as used by the CONABIO Rubiaceae database";

        languagesVoc = TermVocabulary.NewInstance(TermType.Language,

                description, label, null, termSourceUri);

        languagesVoc.setUuid(MexicoConabioTransformer.uuidMexicanLanguagesVoc);

        addLanguage(state, "Aguascalientes", MexicoConabioTransformer.uuidAguascalientes);

        this.getVocabularyService().save(languagesVoc);

        return;

    }

    /**

     * @param state

     * @param string

     * @param uuidaguascalientes

     */

    private void addLanguage(SimpleExcelTaxonImportState<CONFIG> state, String label, UUID uuid) {

        String abbrev = null;

        Language language = Language.NewInstance(

                label, label, abbrev);

        language.setUuid(uuid);

        languagesVoc.addTerm(language);

        languagesMap.put(label, language);

    }

    /**

     * @param state

     * @param refStr

     * @return

     */

    private Reference getReference(SimpleExcelTaxonImportState<CONFIG> state, String refStr) {

        if (StringUtils.isNoneBlank(refStr)){

            return null;

        }

        Reference ref = state.getReference(refStr);

        if (ref == null){

            ref = ReferenceFactory.newBook();

            ref.setTitleCache(refStr, true);

            state.putReference(refStr, ref);

        }

        return ref;

    }

    /**

     * @param taxon

     * @return

     */

    private TaxonDescription getTaxonDescription(Taxon taxon) {

        if (!taxon.getDescriptions().isEmpty()){

            return taxon.getDescriptions().iterator().next();

        }else{

            TaxonDescription desc = TaxonDescription.NewInstance(taxon);

            return desc;

        }

    }

    @Override

    protected boolean isIgnore(SimpleExcelTaxonImportState<CONFIG> state) {

        return ! state.getConfig().isDoCommonNames();

    }

}