D-Net

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

import java.io.IOException;

import java.io.StringWriter;

import javax.annotation.Resource;

import org.antlr.stringtemplate.StringTemplate;

import org.apache.commons.io.IOUtils;

import com.googlecode.sarasvati.Arc;

import com.googlecode.sarasvati.NodeToken;

import eu.dnetlib.enabling.is.registry.rmi.ISRegistryException;

import eu.dnetlib.enabling.is.registry.rmi.ISRegistryService;

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

import eu.dnetlib.msro.workflows.util.WorkflowConstants;

import eu.dnetlib.msro.workflows.util.WorkflowConstants.WorkflowStatus;

public class RegisterMetaWfJobNode extends SimpleJobNode {

	@Resource

	private DnetServiceLocator serviceLocator;

	private String wfName;

	@Override

	protected String execute(final NodeToken token) throws Exception {

		final String dsId = token.getFullEnv().getAttribute(WorkflowConstants.DATAPROVIDER_ID);

		final String dsName = token.getFullEnv().getAttribute(WorkflowConstants.DATAPROVIDER_NAME);

		final String ifaceId = token.getFullEnv().getAttribute(WorkflowConstants.DATAPROVIDER_INTERFACE);

		final String metaWfId = registerDatasourceWorkflow(dsId, dsName, ifaceId);

		token.getFullEnv().setAttribute("META_WORKFLOW_ID", metaWfId);

		token.getFullEnv().setAttribute("META_WORKFLOW_STATUS", WorkflowStatus.ASSIGNED.toString());

		return Arc.DEFAULT_ARC;

	}

	public String registerDatasourceWorkflow(final String dsId, final String dsName, final String ifaceId) throws ISRegistryException, IOException {

		final StringWriter sw = new StringWriter();

		IOUtils.copy(getClass().getResourceAsStream("/eu/dnetlib/msro/workflows/templates/meta-workflow.xml.st"), sw);

		final StringTemplate st = new StringTemplate(sw.toString());

		st.setAttribute("dsId", dsId);

		st.setAttribute("ifaceId", ifaceId);

		st.setAttribute("dsName", dsName);

		st.setAttribute("section", "dataproviders");

		st.setAttribute("wfName", getWfName());

		st.setAttribute("wfFamily", getWfName());

		st.setAttribute("status", WorkflowStatus.ASSIGNED.toString());

		return serviceLocator.getService(ISRegistryService.class).registerProfile(st.toString());

	}

	public String getWfName() {

		return wfName;

	}

	public void setWfName(final String wfName) {

		this.wfName = wfName;

	}

}