D-Net

package eu.dnetlib.functionality.modular.ui.workflows.util;

import java.io.StringReader;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.dom4j.Document;

import org.dom4j.Node;

import org.dom4j.io.SAXReader;

import com.google.common.base.Function;

import com.google.common.base.Splitter;

import com.google.common.collect.Lists;

import eu.dnetlib.functionality.modular.ui.workflows.objects.AtomicWorkflowDescriptor;

import eu.dnetlib.functionality.modular.ui.workflows.objects.MetaWorkflowDescriptor;

import eu.dnetlib.functionality.modular.ui.workflows.objects.NodeInfo;

import eu.dnetlib.miscutils.datetime.DateUtils;

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

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

public class TransformerUtils {

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

	public static final Function<String, MetaWorkflowDescriptor> toMetaWorkflow = new Function<String, MetaWorkflowDescriptor>() {

		@Override

		public MetaWorkflowDescriptor apply(final String s) {

			Iterator<String> iter = Splitter.on("@@@").trimResults().split(s).iterator();

			String id = iter.next();

			String name = iter.next();

			String family = iter.next();

			WorkflowStatus status = WorkflowStatus.valueOf(iter.next());

			return new MetaWorkflowDescriptor(id, name, status, family);

		}

	};

	public static final Function<String, NodeInfo> toNodeInfo = new Function<String, NodeInfo>() {

		@Override

		public NodeInfo apply(final String xml) {

			try {

				final SAXReader reader = new SAXReader();

				final Document doc = reader.read(new StringReader(xml));

				final List<WorkflowParam> params = Lists.newArrayList();

				for (Object o : doc.selectNodes("//PARAM")) {

					Node n = (Node) o;

					String key = n.valueOf("@name");

					boolean required = "true".equalsIgnoreCase(n.valueOf("@required"));

					String type = n.valueOf("@type");

					String function = n.valueOf("@function");

					String value = n.getText();

					boolean userParam = "user".equals(n.valueOf("@managedBy"));

					params.add(new WorkflowParam(key, value, required, userParam, type, function));

				}

				String name = doc.valueOf("/NODE/@name");

				String desc = doc.valueOf("//DESCRIPTION");

				return new NodeInfo(name, desc, params);

			} catch (Exception e) {

				log.error("Error parsing xml: " + xml, e);

			}

			return new NodeInfo("", "", new ArrayList<WorkflowParam>());

		}

	};

	public static final Function<String, AtomicWorkflowDescriptor> toAtomicWorkflow= new Function<String, AtomicWorkflowDescriptor>() {

		@Override

		public AtomicWorkflowDescriptor apply(final String xml) {

			try {

				final SAXReader reader = new SAXReader();

				final Document doc = reader.read(new StringReader(xml));

				final String id = doc.valueOf("//RESOURCE_IDENTIFIER/@value");

				final String name = doc.valueOf("//WORKFLOW_NAME");

				final String imageUrl = "wf_atomic_workflow.img?id=" + id + "&t=" + DateUtils.now();

				final String lastExecutionDate = doc.valueOf("//STATUS/LAST_EXECUTION_DATE");

				final String start = doc.valueOf("//CONFIGURATION/@start");

				final boolean ready = doc.selectNodes("//PARAM[@required='true' and string-length(normalize-space(.)) = 0]").isEmpty();

				return new AtomicWorkflowDescriptor(id, name, "", null, imageUrl, ready, start, lastExecutionDate, 0, 0);

			} catch (Exception e) {

				log.error("Error parsing xml: " + xml, e);

			}

			return new AtomicWorkflowDescriptor("", "", "", null, "", false, "disabled", "", 0, 0);

		}

	};

}