D-Net

package eu.dnetlib.msro.workflows.nodes.functions;

import java.net.URLEncoder;

import java.util.function.Function;

import org.antlr.stringtemplate.StringTemplate;

import org.apache.commons.io.IOUtils;

import org.apache.commons.lang3.StringUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

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

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

import org.springframework.context.annotation.Scope;

import org.springframework.core.io.Resource;

import org.springframework.stereotype.Component;

import eu.dnetlib.clients.is.InformationServiceClient;

import eu.dnetlib.exceptions.InformationServiceException;

import eu.dnetlib.miscutils.functional.xml.ApplyXslt;

import eu.dnetlib.msro.annotations.ProcessNode;

import eu.dnetlib.msro.exceptions.MSROException;

import eu.dnetlib.msro.workflows.nodes.SimpleParallelProcessNode;

@Component

@Scope("prototype")

@ProcessNode("MdBuilder")

public class MdBuilderJobNode extends AbstractApplyFunctionJobNode<String, String> {

	private String datasourceId;

	private String datasourceInterface;

	@Value("${msro.worker.mdstore.mdbuilder.xslt.template}")

	private Resource mdBuilderTemplateXslt;

	@Autowired

	private InformationServiceClient isClient;

	private static final Log log = LogFactory.getLog(SimpleParallelProcessNode.class);

	@Override

	protected Function<String, String> getFunction() throws MSROException {

		if (StringUtils.isBlank(datasourceId)) { throw new MSROException("Missing datasourceId"); }

		if (StringUtils.isBlank(datasourceInterface)) { throw new MSROException("Missing datasourceInterface"); }

		final String xq = "for $x in doc('/db/DRIVER/" + datasourceId + "') "

				+ "return concat("

				+ "$x//BASE_URL, ' @@@ ', "

				+ "$x//EXTRA_FIELDS/FIELD/value[../key='NamespacePrefix'], ' @@@ ', "

				+ "$x//INTERFACE[@id='" + datasourceInterface + "']/INTERFACE_EXTRA_FIELD[@name='metadata_identifier_path'], ' @@@ ', "

				+ "$x//INTERFACE[@id='" + datasourceInterface + "']/ACCESS_PROTOCOL/@format)";

		try {

			final String[] arr = isClient.findOne(xq).split("@@@");

			final StringTemplate st = new StringTemplate(IOUtils.toString(mdBuilderTemplateXslt.getInputStream()));

			st.setAttribute("datasourceId", datasourceId);

			st.setAttribute("baseurl", URLEncoder.encode(arr[0].trim(), "UTF-8"));

			st.setAttribute("namespacePrefix", arr[1].trim());

			st.setAttribute("xpath", arr[2].trim());

			st.setAttribute("metadatanamespace", getMetadataNamespace(arr[3].trim()));

			return new ApplyXslt(st.toString());

		} catch (final Exception e) {

			log.error("Error preparing MDBuilder function", e);

			throw new MSROException("Error preparing MDBuilder function", e);

		}

	}

	private String getMetadataNamespace(final String format) {

		final String xQuery = "collection('/db/DRIVER/conf/mdFormats')//METADATAFORMAT[@name='" + format + "' or @prefix='" + format + "']/@namespace/string()";

		try {

			return isClient.findOne(xQuery);

		} catch (final InformationServiceException e) {

			log.warn("Format " + format + " not registered on IS");

			return "";

		}

	}

	public String getDatasourceId() {

		return datasourceId;

	}

	public void setDatasourceId(final String datasourceId) {

		this.datasourceId = datasourceId;

	}

	public String getDatasourceInterface() {

		return datasourceInterface;

	}

	public void setDatasourceInterface(final String datasourceInterface) {

		this.datasourceInterface = datasourceInterface;

	}

}