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.ProcessNode;

import eu.dnetlib.msro.workflows.Arc;

import eu.dnetlib.msro.workflows.WorkflowStatus;

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

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

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

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

@Component

@Scope("prototype")

@ProcessNode("RegisterWorkflowFromTemplate")

public class RegisterWorkflowFromTemplateJobNode extends AbstractProcessNode 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;

	private WorkflowProcess process;

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

	@Override

	public void execute(final Token token) {

		try {

			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) -> {

					if (token.getEnv().hasAttribute(v)) {

						final Object rv = token.getEnv().getAttribute(v);

						if (rv != null) {

							log.info(String.format("%s -> env(%s) = %s", k, v, rv));

							map.put(k, StringEscapeUtils.escapeXml11(rv.toString()));

						} else {

							log.warn(String.format("Env param %s is NULL", v));

						}

					} else {

						log.warn(String.format("Env param %s is missing, valid params are %s", v, token.getEnv().getAttributes().keySet()));

					}

				});

			}

			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());

			}

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

			token.getEnv().setAttribute("wfId", wfId);

			token.release(Arc.DEFAULT_ARC);

		} catch (final Throwable e) {

			token.releaseAsFailed(e);

		}

	}

	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;

	}

}