D-Net

package eu.dnetlib.msro.workers.aggregation.repohi;

import java.io.StringReader;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.stream.Collectors;

import org.apache.commons.lang3.StringEscapeUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.dom4j.Document;

import org.dom4j.Element;

import org.dom4j.Node;

import org.dom4j.io.SAXReader;

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

import org.springframework.context.annotation.Scope;

import org.springframework.stereotype.Component;

import eu.dnetlib.clients.is.InformationServiceClient;

import eu.dnetlib.miscutils.templates.TemplateUtil;

import eu.dnetlib.msro.annotations.Direction;

import eu.dnetlib.msro.annotations.EnvParam;

import eu.dnetlib.msro.annotations.ProcessNode;

import eu.dnetlib.msro.workflows.Arc;

import eu.dnetlib.msro.workflows.WorkflowStatus;

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

import eu.dnetlib.msro.workflows.procs.ProcessAware;

import eu.dnetlib.msro.workflows.procs.WorkflowProcess;

@Component

@Scope("prototype")

@ProcessNode("RegisterWorkflowFromTemplate")

public class RegisterWorkflowFromTemplateJobNode extends AbstractParallelProcessNode implements ProcessAware {

	private String wfName;

	private String wfDescription;

	private String wfTemplate;

	private Map<String, String> envParams;

	private Map<String, String> simpleParams;

	@Autowired

	private TemplateUtil templateUtil;

	@Autowired

	private InformationServiceClient isClient;

	@EnvParam(value = "wfId", direction = Direction.OUT)

	private String wfId;

	private WorkflowProcess process;

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

	@Override

	protected String execute() throws Exception {

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

		if (simpleParams != null) {

			envParams.forEach((k, v) -> map.put(k, StringEscapeUtils.escapeXml11(v)));

		}

		if (envParams != null) {

			envParams.forEach((k, v) -> map.put(k, StringEscapeUtils.escapeXml11(process.getEnv().getAttribute(v).toString())));

		}

		map.put("name", StringEscapeUtils.escapeXml11(wfName));

		map.put("desc", StringEscapeUtils.escapeXml11(wfDescription));

		map.put("worker", StringEscapeUtils.escapeXml11(process.getWorkerId()));

		map.put("dsId", StringEscapeUtils.escapeXml11(process.getDsId()));

		map.put("iface", StringEscapeUtils.escapeXml11(process.getDsInterface()));

		final String wfProfile = templateUtil.processTemplateProfile(wfTemplate, map).get("wf");

		final Document doc = new SAXReader().read(new StringReader(wfProfile));

		final Node paramsNode = doc.selectSingleNode("//CONFIGURATION/PARAMETERS");

		final List<?> emptySysParams = paramsNode.selectNodes("//PARAM[@managedBy='system' and @required='true' and not(text())]");

		if (emptySysParams.size() > 0) {

			((Element) doc.selectSingleNode("//CONFIGURATION")).addAttribute("status", WorkflowStatus.WAIT_SYS_SETTINGS.toString());

			log.warn("The following system parameters are empty: "

					+ emptySysParams.stream().map(p -> ((Node) p).valueOf("@name")).collect(Collectors.joining()));

		} else if (paramsNode.selectNodes("//PARAM[@managedBy='user' and @required='true' and not(text())]").size() > 0) {

			((Element) doc.selectSingleNode("//CONFIGURATION")).addAttribute("status", WorkflowStatus.WAIT_USER_SETTINGS.toString());

		} else {

			((Element) doc.selectSingleNode("//CONFIGURATION")).addAttribute("status", WorkflowStatus.EXECUTABLE.toString());

		}

		wfId = isClient.register(doc.asXML());

		return Arc.DEFAULT_ARC;

	}

	public String getWfName() {

		return wfName;

	}

	public void setWfName(final String wfName) {

		this.wfName = wfName;

	}

	public String getWfDescription() {

		return wfDescription;

	}

	public void setWfDescription(final String wfDescription) {

		this.wfDescription = wfDescription;

	}

	public String getWfTemplate() {

		return wfTemplate;

	}

	public void setWfTemplate(final String wfTemplate) {

		this.wfTemplate = wfTemplate;

	}

	public Map<String, String> getEnvParams() {

		return envParams;

	}

	public void setEnvParams(final Map<String, String> envParams) {

		this.envParams = envParams;

	}

	public Map<String, String> getSimpleParams() {

		return simpleParams;

	}

	public void setSimpleParams(final Map<String, String> simpleParams) {

		this.simpleParams = simpleParams;

	}

	@Override

	public void setProcess(final WorkflowProcess process) {

		this.process = process;

	}

	public String getWfId() {

		return wfId;

	}

	public void setWfId(final String wfId) {

		this.wfId = wfId;

	}

}