EDIT

/**

* Copyright (C) 2019 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.descriptive.owl.in;

import java.util.UUID;

import org.apache.logging.log4j.LogManager;

import org.apache.logging.log4j.Logger;

import org.springframework.stereotype.Component;

import com.hp.hpl.jena.rdf.model.Model;

import com.hp.hpl.jena.rdf.model.ResIterator;

import com.hp.hpl.jena.rdf.model.Resource;

import com.hp.hpl.jena.rdf.model.Statement;

import com.hp.hpl.jena.rdf.model.StmtIterator;

import eu.etaxonomy.cdm.common.URI;

import eu.etaxonomy.cdm.io.common.CdmImportBase;

import eu.etaxonomy.cdm.io.descriptive.owl.OwlUtil;

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

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

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

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

import eu.etaxonomy.cdm.model.term.TermNode;

import eu.etaxonomy.cdm.model.term.TermTree;

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

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

/**

 * @author pplitzner

 * @since Apr 24, 2019

 */

@Component("structureTreeOwlImport")

public class StructureTreeOwlImport extends CdmImportBase<StructureTreeOwlImportConfigurator, StructureTreeOwlImportState> {

    private static final long serialVersionUID = -3659780404413458511L;

    private final static Logger logger = LogManager.getLogger();

    @Override

    protected boolean doCheck(StructureTreeOwlImportState state) {

        logger.warn("Checking not yet implemented for " + this.getClass().getSimpleName());

        return true;

    }

    @Override

    public void doInvoke(StructureTreeOwlImportState state) {

        URI source = state.getConfig().getSource();

        state.getModel().read(source.toString());

        //get all trees

        ResIterator iterator = state.getModel().listResourcesWithProperty(OwlUtil.propHasRootNode);

        while(iterator.hasNext()){

            Resource treeResource = iterator.next();

            UUID treeUuid = UUID.fromString(treeResource.getProperty(OwlUtil.propUuid).getString());

            TermTree<?> termTree = getTermTreeService().find(treeUuid);

            if(termTree==null){

                String type = treeResource.getProperty(OwlUtil.propType).getString();

                TermTree<Feature> featureTree = TermTree.NewInstance(TermType.getByKey(type));

                featureTree.setTitleCache(treeResource.getProperty(OwlUtil.propLabel).getString(), true);

                featureTree.setUuid(treeUuid);

                Resource rootNode = treeResource.getProperty(OwlUtil.propHasRootNode).getResource();

                rootNode.listProperties(OwlUtil.propHasSubStructure).forEachRemaining(prop->createNode(featureTree.getRoot(), prop, featureTree.getTitleCache(), state.getModel(), state));

                getTermTreeService().saveOrUpdate(featureTree);

            }

        }

    }

    private <T extends DefinedTermBase> void createNode(TermNode<T> parent, Statement nodeStatement, String treeLabel, Model model, StructureTreeOwlImportState state) {

        if(state.getConfig().getProgressMonitor().isCanceled()){

            return;

        }

        Resource nodeResource = model.createResource(nodeStatement.getObject().toString());

        UUID nodeUuid = UUID.fromString(nodeResource.getProperty(OwlUtil.propUuid).getString());

        Resource termResource = nodeResource.getPropertyResourceValue(OwlUtil.propHasTerm);

        TermNode<?> termNode = getTermNodeService().load(nodeUuid);

        if(termNode==null){

            //import term vocabulary

            Resource vocabularyResource = termResource.getPropertyResourceValue(OwlUtil.propHasVocabulary);

            UUID vocUuid = UUID.fromString(vocabularyResource.getProperty(OwlUtil.propUuid).getString());

            TermVocabulary vocabulary = getVocabularyService().load(vocUuid);

            if(vocabulary==null){

                vocabulary = OwlImportUtil.createVocabulary(vocabularyResource, this, model, state);

                vocabulary = getVocabularyService().save(vocabulary);

            }

            // import term

            UUID termUuid = UUID.fromString(termResource.getProperty(OwlUtil.propUuid).getString());

            T term = (T)getTermService().find(termUuid);

            if(term == null){

                term = (T)OwlImportUtil.createTerm(termResource, this, model, state);

                term = getTermService().save(term);

                vocabulary.addTerm(term); // only add term if it does not already exist

            }

            getVocabularyService().saveOrUpdate(vocabulary);

            termNode = parent.addChild(term);

            termNode.setUuid(nodeUuid);

            // inapplicable if

            StmtIterator inapplicableIterator = nodeResource.listProperties(OwlUtil.propNodeIsInapplicableIf);

            while(inapplicableIterator.hasNext()){

                Statement statement = inapplicableIterator.next();

                FeatureState featureState = createFeatureState(statement, model, state);

                termNode.addInapplicableState(featureState);

            }

            // only applicable if

            StmtIterator onlyApplicableIterator = nodeResource.listProperties(OwlUtil.propNodeIsOnlyApplicableIf);

            while(onlyApplicableIterator.hasNext()){

                Statement statement = onlyApplicableIterator.next();

                FeatureState featureState = createFeatureState(statement, model, state);

                termNode.addApplicableState(featureState);

            }

        }

        state.getConfig().getProgressMonitor().worked(1);

        StmtIterator listProperties = nodeResource.listProperties(OwlUtil.propHasSubStructure);

        while(listProperties.hasNext()){

            Statement prop = listProperties.next();

            createNode(termNode, prop, treeLabel, model, state);

        }

    }

    private FeatureState createFeatureState(Statement statement, Model model, StructureTreeOwlImportState state) {

        Resource featureStateResource = model.createResource(statement.getObject().toString());

        Resource featureResouce = featureStateResource.getPropertyResourceValue(OwlUtil.propFeatureStateHasFeature);

        Resource stateResouce = featureStateResource.getPropertyResourceValue(OwlUtil.propFeatureStateHasState);

        UUID featureUuid = UUID.fromString(featureResouce.getProperty(OwlUtil.propUuid).getString());

        Feature feature = (Feature)getTermService().find(featureUuid);

        if(feature==null){

            feature = (Feature) OwlImportUtil.createTerm(featureResouce, this, model, state);

            getTermService().saveOrUpdate(feature);

        }

        UUID stateUuid = UUID.fromString(stateResouce.getProperty(OwlUtil.propUuid).getString());

        State stateTerm = (State)getTermService().find(stateUuid);

        if(stateTerm==null){

            stateTerm = (State) OwlImportUtil.createTerm(stateResouce, this, model, state);

            getTermService().saveOrUpdate(stateTerm);

        }

        FeatureState featureState = FeatureState.NewInstance(feature, stateTerm);

        return featureState;

    }

    @Override

    protected boolean isIgnore(StructureTreeOwlImportState state) {

        return false;

    }

}