D-Net

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

import java.util.HashMap;

import java.util.Map;

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

import org.springframework.core.io.ClassPathResource;

import com.googlecode.sarasvati.Arc;

import com.googlecode.sarasvati.NodeToken;

import eu.dnetlib.data.resultSet.ResultSetFactory;

import eu.dnetlib.miscutils.ApplyXslt;

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

import eu.dnetlib.rmi.objects.resultSet.ResultSet;

import eu.dnetlib.rmi.soap.exceptions.ManagerServiceException;

public class ApplyXsltJobNode extends SimpleJobNode {

	private String inputResultSetParam;

	private String outputResultSetParam;

	private String xsltClasspath;

	@Autowired

	private ResultSetFactory resultSetFactory;

	@Override

	protected String execute(final NodeToken token) throws Exception {

		@SuppressWarnings("unchecked")

		final ResultSet<String> inputRs = (ResultSet<String>) token.getEnv().getTransientAttribute(inputResultSetParam);

		if (inputRs == null) { throw new ManagerServiceException("InputResultSetParam (" + inputResultSetParam + ") not found in ENV"); }

		final Map<String, String> xsltParams = new HashMap<String, String>();

		for (String name : token.getFullEnv().getAttributeNames()) {

			xsltParams.put(name.replaceAll(":", "_"), token.getFullEnv().getAttribute(name));

		}

		for (String name : token.getEnv().getAttributeNames()) {

			xsltParams.put(name.replaceAll(":", "_"), token.getEnv().getAttribute(name));

		}

		xsltParams.putAll(parseJsonParameters(token));

		final ResultSet<String> outputRs = resultSetFactory.transform(inputRs, new ApplyXslt(new ClassPathResource(xsltClasspath), xsltParams));

		token.getEnv().setAttribute(outputResultSetParam, outputRs.toString());

		return Arc.DEFAULT_ARC;

	}

	public String getXsltClasspath() {

		return xsltClasspath;

	}

	public void setXsltClasspath(final String xsltClasspath) {

		this.xsltClasspath = xsltClasspath;

	}

	public String getInputResultSetParam() {

		return inputResultSetParam;

	}

	public void setInputResultSetParam(final String inputResultSetParam) {

		this.inputResultSetParam = inputResultSetParam;

	}

	public String getOutputResultSetParam() {

		return outputResultSetParam;

	}

	public void setOutputResultSetParam(final String outputResultSetParam) {

		this.outputResultSetParam = outputResultSetParam;

	}

}