D-Net

package eu.dnetlib.administration.uis.modules.repos;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.stream.Collectors;

import org.springframework.beans.factory.annotation.Autowired;

import com.google.gson.Gson;

import eu.dnetlib.administration.uis.modules.UIModule;

import eu.dnetlib.clients.is.InformationServiceClient;

import eu.dnetlib.clients.msro.MsroClient;

import eu.dnetlib.clients.msro.ProtocolParameter;

import eu.dnetlib.exceptions.InformationServiceException;

public abstract class AbstractRepoModule extends UIModule {

	private final Map<String, List<ProtocolParameter>> parametersMap = new HashMap<>();

	@Autowired

	private InformationServiceClient isClient;

	@Autowired

	private MsroClient msroClient;

	protected List<VocabularyEntry> fetchVocabularyTerms(final String voc) throws InformationServiceException {

		final String xquery = "for $x in collection('/db/DRIVER/VocabularyDSResources/VocabularyDSResourceType')[.//VOCABULARY_NAME/@code = '"

				+ voc.trim() + "']//TERM return concat($x/@code, ' @@@ ', $x/@english_name)";

		return isClient.find(xquery)

				.stream()

				.map(s -> s.split("@@@"))

				.map(arr -> new VocabularyEntry(arr[0].trim(), arr[1].trim()))

				.sorted()

				.collect(Collectors.toList());

	}

	protected String fetchVocabularyTermsAsJson(final String voc) throws InformationServiceException {

		return (new Gson()).toJson(fetchVocabularyTerms(voc));

	}

	protected List<ProtocolParameter> listParametersForProtocol(final String protocol) {

		return getMapProtocolParameters().get(protocol.toLowerCase());

	}

	protected Map<String, List<ProtocolParameter>> getMapProtocolParameters() {

		if (parametersMap.isEmpty()) {

			msroClient.listProtocols().forEach(d -> parametersMap.put(d.getName().toLowerCase(), d.getParams()));

		}

		return parametersMap;

	}

}