/src/main/java/org/isatools/isacreator/ontologymanager/BioPortal4Client.java

package org.isatools.isacreator.ontologymanager;

import org.apache.log4j.Logger;
import org.isatools.isacreator.configuration.Ontology;
import org.isatools.isacreator.configuration.RecommendedOntology;
import org.isatools.isacreator.ontologymanager.bioportal.io.AcceptedOntologies;
import org.isatools.isacreator.ontologymanager.bioportal.jsonresulthandlers.BioPortalSearchResultHandler;
import org.isatools.isacreator.ontologymanager.common.OntologyTerm;

import java.util.*;

/**
 * Created by eamonnmaguire on 17/12/2013.
 */
public class BioPortal4Client implements OntologyService {

    private static final Logger log = Logger.getLogger(BioPortal4Client.class);

    public static final String DIRECT_ONTOLOGY_URL = "http://bioportal.bioontology.org/ontologies/";
    public static final String REST_URL = "http://data.bioontology.org/";

    private BioPortalSearchResultHandler handler;

    private Map<String, Map<String, OntologyTerm>> cachedNodeChildrenQueries;

    public BioPortal4Client() {
        this.handler = new BioPortalSearchResultHandler();
        this.cachedNodeChildrenQueries = new HashMap<String, Map<String, OntologyTerm>>();
    }

    public Map<String, String> getOntologyNames() {
        return AcceptedOntologies.getOntologySourceToNames();
    }

    /**
     * Retrieves an OntologyTerm given the URI of a term and the ontology bioPortal URI
     * @param termId
     * @param ontology
     * @return
     */
    public OntologyTerm getTerm(String termId, String ontology) {
        return handler.getTermMetadata(termId, ontology);
    }

    public Map<String, List<OntologyTerm>> exactSearch(String term, String ontology) {
        return handler.getSearchResults(term, ontology, null, true);
    }

    public Map<String, String> getTermMetadata(String termId, String ontology) {
        return handler.getTermMetadata(termId, ontology).getComments();
    }

    public Map<OntologySourceRefObject, List<OntologyTerm>> getTermsByPartialNameFromSource(String term, String source, boolean reverseOrder) {

        term = correctTermForHTTPTransport(term);

        Map<String, List<OntologyTerm>> searchResult = handler.getSearchResults(term, source, null);

        return convertStringKeyMapToOntologySourceRefKeyMap(searchResult);
    }

    public Map<OntologySourceRefObject, List<OntologyTerm>> getTermsByPartialNameFromSource(String term, List<RecommendedOntology> recommendedOntology) {
        term = correctTermForHTTPTransport(term);

        // need to accommodate more complicated search strings in the case where the recommended source contains the branch
        // to search under as well!
        Map<OntologySourceRefObject, List<OntologyTerm>> result = new HashMap<OntologySourceRefObject, List<OntologyTerm>>();

        // need to do a loop over all branches and do a single query on those recommended ontologies only defined
        // by the source, not the branch.
        for (RecommendedOntology ro : recommendedOntology) {

            if (ro.getOntology() != null) {

                String subtree = null;
                if (ro.getBranchToSearchUnder() != null && !ro.getBranchToSearchUnder().getBranchIdentifier().equals("")) {
                    subtree = ro.getBranchToSearchUnder().getBranchIdentifier();
                }

                Map<String, List<OntologyTerm>> searchResult = handler.getSearchResults(term, ro.getOntology().getOntologyAbbreviation(), subtree);

                if (searchResult != null) {
                    result.putAll(convertStringKeyMapToOntologySourceRefKeyMap(searchResult));
                }
            }
        }
        return result;
    }

    public Map<String, String> getOntologyVersions() {

        return AcceptedOntologies.getOntologySourceToVersion();
    }

    public Map<String, OntologyTerm> getOntologyRoots(String ontologyAbbreviation) {
//        http://data.bioontology.org/ontologies/EFO/classes/roots
        if (!cachedNodeChildrenQueries.containsKey(ontologyAbbreviation)) {
            cachedNodeChildrenQueries.put(ontologyAbbreviation, handler.getOntologyRoots(ontologyAbbreviation));
        }
        return cachedNodeChildrenQueries.get(ontologyAbbreviation);
    }

    public Map<String, OntologyTerm> getTermParent(String termAccession, String ontology) {

        return null;
    }

    public Map<String, OntologyTerm> getTermChildren(String termAccession, String ontologyAbbreviation) {
        String uniqueReferenceId = ontologyAbbreviation + "-" + termAccession + "-children";
        if (!cachedNodeChildrenQueries.containsKey(uniqueReferenceId)) {
            cachedNodeChildrenQueries.put(uniqueReferenceId, handler.getTermChildren(termAccession, ontologyAbbreviation));
        }
        return cachedNodeChildrenQueries.get(uniqueReferenceId);
    }

    public Map<String, OntologyTerm> getAllTermParents(String termAccession, String ontologyAbbreviation) {

        String uniqueReferenceId = ontologyAbbreviation + "-" + termAccession + "-parents";
        if (!cachedNodeChildrenQueries.containsKey(uniqueReferenceId)) {
            cachedNodeChildrenQueries.put(uniqueReferenceId, handler.getTermParents(termAccession, ontologyAbbreviation));
        }
        return cachedNodeChildrenQueries.get(uniqueReferenceId);

    }

    public Collection<Ontology> getAllOntologies() {
        return handler.getAllOntologies().values();
    }

    private Map<OntologySourceRefObject, List<OntologyTerm>> convertStringKeyMapToOntologySourceRefKeyMap(Map<String, List<OntologyTerm>> toConvert) {

        Map<OntologySourceRefObject, List<OntologyTerm>> convertedMap = new HashMap<OntologySourceRefObject, List<OntologyTerm>>();
        for (String ontologyId : toConvert.keySet()) {
            Ontology ontology = AcceptedOntologies.getAcceptedOntologies().get(ontologyId);
            if (ontology != null) {
                OntologySourceRefObject osro = new OntologySourceRefObject(ontology.getOntologyAbbreviation(),
                        ontologyId, ontology.getOntologyVersion(), ontology.getOntologyDisplayLabel());

                convertedMap.put(osro, new ArrayList<OntologyTerm>());

                for (OntologyTerm ontologyTerm : toConvert.get(ontologyId)) {
                    ontologyTerm.setOntologySourceInformation(osro);
                    convertedMap.get(osro).add(ontologyTerm);
                }

            }
        }
        return convertedMap;
    }

    private String correctTermForHTTPTransport(String term) {
        return term;
//        try {
//            return URLEncoder.encode(term, "UTF-8");
//        } catch (UnsupportedEncodingException e) {
//            return term;
//        }
    }
}