D-Net

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

import java.io.IOException;

import java.util.List;

import javax.xml.ws.wsaddressing.W3CEndpointReference;

import org.antlr.stringtemplate.StringTemplate;

import org.apache.commons.io.IOUtils;

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

import org.springframework.core.io.Resource;

import com.googlecode.sarasvati.Arc;

import com.googlecode.sarasvati.NodeToken;

import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpDocumentNotFoundException;

import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpException;

import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpService;

import eu.dnetlib.enabling.resultset.XSLTMappedResultSetFactory;

import eu.dnetlib.enabling.resultset.client.utils.EPRUtils;

import eu.dnetlib.enabling.tools.ServiceLocator;

import eu.dnetlib.msro.rmi.MSROException;

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

public class MdBuilderJobNode extends SimpleJobNode {

	private Resource mdBuilderTemplateXslt;

	private String inputEprParam;

	private String outputEprParam;

	private String datasourceId;

	private String datasourceInterface;

	private XSLTMappedResultSetFactory xsltMappedResultSetFactory;

	@javax.annotation.Resource(name = "lookupLocator")

	private ServiceLocator<ISLookUpService> lookupLocator;

	@Override

	protected String execute(final NodeToken token) throws MSROException {

		final String inputEpr = token.getEnv().getAttribute(inputEprParam);

		if ((inputEpr == null) || inputEpr.isEmpty()) throw new MSROException("InputEprParam (" + inputEprParam + ") not found in ENV");

		String xQuery = "for $x in collection('/db/DRIVER/RepositoryServiceResources/RepositoryServiceResourceType') "

				+ "return $x//INTERFACE[@id='{interfaceId}']/INTERFACE_EXTRA_FIELD[@name='metadata_identifier_path']/string()";

		xQuery = xQuery.replace("{interfaceId}", datasourceInterface);

		try {

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

			st.setAttribute("xpath", lookupLocator.getService().getResourceProfileByQuery(xQuery));

			st.setAttribute("datasourceId", datasourceId);

			/* If namespacePrefix has been already pushed to env by some custom JobNode e.g. ObtainOpenaireDataSourceParamsJobNode

			   then push it to ST. Else:

			   a) try to get it from EXTRAFIELDS of the datasource

			   b) try to get it from DATASOURCE_ORIGINAL_ID of the datasource

			   c) if any of the is present, then push to ST the datasourceId */

			if (token.getEnv().hasAttribute("namespacePrefix")) {

				st.setAttribute("namespacePrefix", token.getEnv().getAttribute("namespacePrefix"));

			} else {

				List<String> namespacePrefix;

				xQuery = "/*[.//RESOURCE_IDENTIFIER/@value='" + datasourceId + "']//EXTRA_FIELDS/FIELD/value[../key='NamespacePrefix']/string()";

				namespacePrefix = lookupLocator.getService().quickSearchProfile(xQuery);

				if (namespacePrefix.size() != 0){

					st.setAttribute("namespacePrefix", namespacePrefix.get(0));

				} else {

					xQuery = "/*[.//RESOURCE_IDENTIFIER/@value='" + datasourceId + "']//DATASOURCE_ORIGINAL_ID/string()";

					namespacePrefix = lookupLocator.getService().quickSearchProfile(xQuery);

					if (namespacePrefix.size() != 0) {

						st.setAttribute("namespacePrefix", namespacePrefix.get(0));

					} else {

						st.setAttribute("namespacePrefix", datasourceId);

					}

				}

			}

			final W3CEndpointReference epr = xsltMappedResultSetFactory.createMappedResultSet((new EPRUtils()).getEpr(inputEpr), st.toString());

			token.getEnv().setAttribute(outputEprParam, epr.toString());

			return Arc.DEFAULT_ARC;

		} catch (ISLookUpDocumentNotFoundException e) {

			throw new MSROException("Missing metadata_identifier_path extra field for ds: " + datasourceId, e);

		} catch (ISLookUpException e) {

			throw new MSROException("Error searching metadata_identifier_path extra field for ds: " + datasourceId, e);

		} catch (IOException e) {

			throw new MSROException("Error parsing template: " + getMdBuilderTemplateXslt().getFilename(), e);

		}

	}

	public String getInputEprParam() {

		return inputEprParam;

	}

	public void setInputEprParam(final String inputEprParam) {

		this.inputEprParam = inputEprParam;

	}

	public String getOutputEprParam() {

		return outputEprParam;

	}

	public void setOutputEprParam(final String outputEprParam) {

		this.outputEprParam = outputEprParam;

	}

	public XSLTMappedResultSetFactory getXsltMappedResultSetFactory() {

		return xsltMappedResultSetFactory;

	}

	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;

	}

	@Required

	public void setXsltMappedResultSetFactory(final XSLTMappedResultSetFactory xsltMappedResultSetFactory) {

		this.xsltMappedResultSetFactory = xsltMappedResultSetFactory;

	}

	public Resource getMdBuilderTemplateXslt() {

		return mdBuilderTemplateXslt;

	}

	@Required

	public void setMdBuilderTemplateXslt(final Resource mdBuilderTemplateXslt) {

		this.mdBuilderTemplateXslt = mdBuilderTemplateXslt;

	}

}