EDIT

/**

* 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.Set;

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

 * @since 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<>();

    private Map<String, Taxon> taxonIdMap;

    @Override

    protected String getWorksheetName() {

        return "NombresComunes";

    }

    @Override

    protected void firstPass(SimpleExcelTaxonImportState<CONFIG> state) {

        initLanguageVocabulary(state);

        initTaxa();

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

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

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

        Taxon taxon = taxonIdMap.get(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 = getValueNd(record, "Lengua");

            Language language = languagesMap.get(langStr);

            if (language == null && langStr != null){

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

            }

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

            CommonTaxonName commonName = CommonTaxonName.NewInstance(nomComunStr,

                    language, null);

            desc.addElement(commonName);

            Reference ref = getReference(state, refStr);

            if (ref != null){

                commonName.addSource(OriginalSourceType.PrimaryTaxonomicSource,

                        null, null, ref, null);

            }

            getTaxonService().save(taxon);

        }

    }

    @SuppressWarnings("unchecked")

    private void initTaxa() {

        if (taxonIdMap == null){

            Set<String> existingKeys = MexicoConabioTaxonImport.taxonIdMap.keySet();

            taxonIdMap = (Map<String, Taxon>)getCommonService().getSourcedObjectsByIdInSource(Taxon.class,

                    existingKeys, MexicoConabioTaxonImport.TAXON_NAMESPACE);

        }

    }

    /**

     * @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 languages as used by the CONABIO Rubiaceae database";

        languagesVoc = TermVocabulary.NewInstance(TermType.Language,

                description, label, null, termSourceUri);

        languagesVoc.setUuid(MexicoConabioTransformer.uuidMexicanLanguagesVoc);

        addLanguage(state, "Chontal", MexicoConabioTransformer.uuidChontal);

        addLanguage(state, "Chinanteco", MexicoConabioTransformer.uuidChinanteco);

        addLanguage(state, "Chiapaneca", MexicoConabioTransformer.uuidChiapaneca);

        addLanguage(state, "Huasteco", MexicoConabioTransformer.uuidHuasteco);

        addLanguage(state, "Español-Maya", MexicoConabioTransformer.uuidEspanol_Maya);

        addLanguage(state, "Guarijío", MexicoConabioTransformer.uuidGuarijio);

        addLanguage(state, "Huave", MexicoConabioTransformer.uuidHuave);

        addLanguage(state, "Español", MexicoConabioTransformer.uuidEspanol);

        addLanguage(state, "Maya", MexicoConabioTransformer.uuidMaya);

        addLanguage(state, "Lacandón", MexicoConabioTransformer.uuidLacandon);

        addLanguage(state, "Inglés", MexicoConabioTransformer.uuidIngles);

        addLanguage(state, "Itzmal", MexicoConabioTransformer.uuidItzmal);

        addLanguage(state, "Náhuatl", MexicoConabioTransformer.uuidNahuatl);

        addLanguage(state, "Tarahumara", MexicoConabioTransformer.uuidTarahumara);

        addLanguage(state, "Otomí", MexicoConabioTransformer.uuidOtomi);

        addLanguage(state, "Mixe", MexicoConabioTransformer.uuidMixe);

        addLanguage(state, "Tseltal", MexicoConabioTransformer.uuidTseltal);

        addLanguage(state, "Zapoteco", MexicoConabioTransformer.uuidZapoteco);

        addLanguage(state, "Totonaco", MexicoConabioTransformer.uuidTotonaco);

        addLanguage(state, "Tarasco", MexicoConabioTransformer.uuidTarasco);

        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;

        }

    }

    private String getValueNd(HashMap<String, String> record, String string) {

        String value = getValue(record, string);

        if ("ND".equals(value)){

            return null;

        }else{

            return value;

        }

    }

    @Override

    protected boolean isIgnore(SimpleExcelTaxonImportState<CONFIG> state) {

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

    }

}