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.util.Arrays;

import java.util.List;

import java.util.Map;

import java.util.Set;

import org.apache.log4j.Logger;

import eu.etaxonomy.cdm.io.excel.common.ExcelImportBase;

import eu.etaxonomy.cdm.io.excel.common.ExcelImportConfiguratorBase;

import eu.etaxonomy.cdm.io.excel.common.ExcelRowBase;

import eu.etaxonomy.cdm.model.agent.AgentBase;

import eu.etaxonomy.cdm.model.agent.Person;

import eu.etaxonomy.cdm.model.agent.Team;

import eu.etaxonomy.cdm.model.agent.TeamOrPersonBase;

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

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

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

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

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

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

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

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

import eu.etaxonomy.cdm.strategy.parser.INonViralNameParser;

import eu.etaxonomy.cdm.strategy.parser.NonViralNameParserImpl;

/**

 * Simple Excel import class that works without default state class

 * {@link SimpleExcelTaxonImportState}

 * @author a.mueller

 * @since 16.06.2016

 */

public abstract class SimpleExcelTaxonImport<CONFIG extends ExcelImportConfiguratorBase>

        extends ExcelImportBase<SimpleExcelTaxonImportState<CONFIG>, CONFIG, ExcelRowBase>{

    private static final long serialVersionUID = -4345647703312616421L;

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

    protected static INonViralNameParser<?> nameParser = NonViralNameParserImpl.NewInstance();

    @Override

    protected void analyzeRecord(Map<String, String> record, SimpleExcelTaxonImportState<CONFIG> state) {

        //override only if needed

    }

    @Override

    protected void secondPass(SimpleExcelTaxonImportState<CONFIG> state) {

        //override only if needed

    }

    @Override

    protected boolean isIgnore(SimpleExcelTaxonImportState<CONFIG> state) {

        return false;

    }

//***************************** METHODS *********************************/

    protected IdentifiableSource makeOriginalSource(SimpleExcelTaxonImportState<CONFIG> state) {

        return IdentifiableSource.NewDataImportInstance("line: " + state.getCurrentLine(), null, state.getConfig().getSourceReference());

    }

    protected void makeGenus(SimpleExcelTaxonImportState<CONFIG> state,

            IBotanicalName speciesName,

            Reference sec,

            Taxon taxon,

            TaxonNode familyNode) {

        TaxonNode higherNode;

        if (speciesName.isProtectedTitleCache()){

            higherNode = familyNode;

        }else{

            String genusStr = speciesName.getGenusOrUninomial();

            Taxon genus = state.getHigherTaxon(genusStr);

            if (genus != null){

                higherNode = genus.getTaxonNodes().iterator().next();

            }else{

                IBotanicalName genusName = TaxonNameFactory.NewBotanicalInstance(Rank.GENUS());

                genusName.addSource(makeOriginalSource(state));

                genusName.setGenusOrUninomial(genusStr);

                genus = Taxon.NewInstance(genusName, sec);

                genus.addSource(makeOriginalSource(state));

                higherNode = familyNode.addChildTaxon(genus, null, null);

                state.putHigherTaxon(genusStr, genus);

            }

        }

        higherNode.addChildTaxon(taxon, null, null);

    }

    protected void checkAllKeysExist(String line, Set<String> keys, List<String> expectedKeys) {

        for (String key: keys) {

            if (! expectedKeys.contains(key)){

                logger.warn(line + "Unexpected Key: " + key);

            }

        }

    }

    protected void replaceAuthorNamesAndNomRef(SimpleExcelTaxonImportState<CONFIG> state,

            IBotanicalName name) {

        TeamOrPersonBase<?> combAuthor = name.getCombinationAuthorship();

        name.setCombinationAuthorship(getExistingAuthor(state, combAuthor));

        TeamOrPersonBase<?> exAuthor = name.getExCombinationAuthorship();

        name.setExCombinationAuthorship(getExistingAuthor(state, exAuthor));

        TeamOrPersonBase<?> basioAuthor = name.getBasionymAuthorship();

        name.setBasionymAuthorship(getExistingAuthor(state, basioAuthor));

        TeamOrPersonBase<?> exBasioAuthor = name.getExBasionymAuthorship();

        name.setExBasionymAuthorship(getExistingAuthor(state, exBasioAuthor));

        INomenclaturalReference nomRef = name.getNomenclaturalReference();

        if (nomRef != null){

            TeamOrPersonBase<?> refAuthor = nomRef.getAuthorship();

            nomRef.setAuthorship(getExistingAuthor(state, refAuthor));

            Reference existingRef = getExistingReference(state, (Reference)nomRef);

            if (existingRef != null){

                name.setNomenclaturalReference(existingRef);

            }

        }

    }

    private Reference getExistingReference(SimpleExcelTaxonImportState<CONFIG> state, Reference ref) {

        if (ref == null){

            return null;

        }else{

            initRerenceMap(state);

            Reference result = state.getReference(ref.getTitleCache());

            if (result == null){

                result = ref;

                Reference inRef = result.getInReference();

                if (inRef != null){

                    result.setInReference(getExistingReference(state, result.getInReference()));

                }

                state.putReference(result.getTitleCache(), result);

            }

            return result;

        }

    }

    boolean referenceMapIsInitialized;

    private void initRerenceMap(SimpleExcelTaxonImportState<CONFIG> state) {

        if (!referenceMapIsInitialized){

            List<String> propertyPaths = Arrays.asList("");

            List<Reference> existingReferences = this.getReferenceService().list(null, null, null, null, propertyPaths);

            for (Reference ref : existingReferences){

                state.putReference(ref.getTitleCache(), ref);

            }

            referenceMapIsInitialized = true;

        }

    }

    boolean agentMapIsInitialized = false;

    @SuppressWarnings("rawtypes")

    private void initAgentMap(SimpleExcelTaxonImportState<CONFIG> state) {

        if (!agentMapIsInitialized){

            List<String> propertyPaths = Arrays.asList("");

            List<AgentBase<?>> existingAgents = this.getAgentService().list(null, null, null, null, propertyPaths);

            for (AgentBase<?> agent : existingAgents){

                if (agent.isInstanceOf(TeamOrPersonBase.class)){

                    state.putAgentBase(agent.getTitleCache(), (TeamOrPersonBase)agent);

                }

            }

            agentMapIsInitialized = true;

        }

    }

    protected TeamOrPersonBase<?> getExistingAuthor(SimpleExcelTaxonImportState<CONFIG> state,

            TeamOrPersonBase<?> author) {

        if (author == null){

            return null;

        }else{

            initAgentMap(state);

            TeamOrPersonBase<?> result = state.getAgentBase(author.getTitleCache());

            if (result == null){

                state.putAgentBase(author.getTitleCache(), author);

                if (author instanceof Team){

                    handleTeam(state, (Team)author);

                }

                result = author;

            }

            return result;

        }

    }

    private void handleTeam(SimpleExcelTaxonImportState<CONFIG> state, Team team) {

        List<Person> members = team.getTeamMembers();

        for (int i =0; i< members.size(); i++){

            Person person = members.get(i);

            //FIXME cast find a better way to guarantee that only persons are returned

            Person existingPerson = (Person)state.getAgentBase(person.getTitleCache());

            if (existingPerson != null){

                members.set(i, existingPerson);

            }else{

                state.putAgentBase(person.getTitleCache(), person);

            }

        }

    }

    /**

     * Returns the line string and prints it on syso for each count-th line

     */

    protected String getLine(SimpleExcelTaxonImportState<CONFIG> state, Integer count) {

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

        if (count != null && (state.getCurrentLine() % count) == 0){

            System.out.println(line);

        }

        return line;

    }

}