D-Net

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

import java.io.StringReader;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.Set;

import javax.annotation.PostConstruct;

import javax.annotation.Resource;

import javax.xml.transform.stream.StreamSource;

import com.google.common.base.Function;

import com.google.common.collect.Lists;

import com.google.common.collect.Maps;

import com.google.common.collect.Sets;

import com.google.gson.Gson;

import com.google.gson.reflect.TypeToken;

import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpException;

import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpService;

import eu.dnetlib.enabling.locators.UniqueServiceLocator;

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

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

import eu.dnetlib.miscutils.functional.xml.ApplyXslt;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.dom4j.DocumentException;

import org.dom4j.Node;

import org.dom4j.io.SAXReader;

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

public class ISLookupClient {

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

	private static final ApplyXslt workflowToHTML = new ApplyXslt(new StreamSource(

			ISLookupClient.class.getResourceAsStream("/eu/dnetlib/functionality/modular/workflows/xslt/wf_profile2html.xslt")));

	@Resource

	private UniqueServiceLocator serviceLocator;

	private String categoryUisJson;

	private Map<String, List<WorkflowParamUI.UI>> categoryUis;

	@PostConstruct

	private void init() {

		log.info("Initialing categoryUis map using JSON: " + getCategoryUisJson());

		this.categoryUis = new Gson().fromJson(getCategoryUisJson(), new TypeToken<Map<String, List<WorkflowParamUI.UI>>>() {

		}.getType());

	}

	public List<String> listSimpleWorflowSections() {

		final String xquery = "distinct-values(//WORKFLOW_NAME/@menuSection/string())";

		try {

			return serviceLocator.getService(ISLookUpService.class).quickSearchProfile(xquery);

		} catch (ISLookUpException e) {

			log.error("Error obtaining worflowSections", e);

			return Lists.newArrayList();

		}

	}

	public List<WorkflowItem> listWorflowsForSection(final String name) {

		return listWorflowsByCondition("//WORKFLOW_NAME/@menuSection='" + name + "'");

	}

	public List<WorkflowItem> listWorflowsForDatasource(final String dsId) {

		return listWorflowsByCondition("//DATASOURCE/@id='" + dsId + "'");

	}

	private List<WorkflowItem> listWorflowsByCondition(final String cond) {

		final String query =

				"for $x in collection('/db/DRIVER/WorkflowDSResources/WorkflowDSResourceType') where $x" + cond

						+ " order by $x//WORKFLOW_NAME return concat($x//RESOURCE_IDENTIFIER/@value, ' ==@== ', $x//WORKFLOW_NAME, ' ==@== ', $x//WORKFLOW_DESCRIPTION)";

		try {

			return Lists.transform(serviceLocator.getService(ISLookUpService.class).quickSearchProfile(query), new Function<String, WorkflowItem>() {

				@Override

				public WorkflowItem apply(final String s) {

					final String[] arr = s.split("==@==");

					return new WorkflowItem(arr[0].trim(), arr[1].trim(), arr[2].trim());

				}

			});

		} catch (ISLookUpException e) {

			log.error("Error obtaining wfs using query: " + query, e);

			return Lists.newArrayList();

		}

	}

	public String getProfile(final String id) {

		try {

			return serviceLocator.getService(ISLookUpService.class).getResourceProfile(id);

		} catch (ISLookUpException e) {

			log.error("Error finding profile: " + id, e);

			return null;

		}

	}

	public Set<String> getNotConfiguredNodes(final String id) {

		final String query = "for $x in (/*[.//RESOURCE_IDENTIFIER/@value='" + id + "']//NODE) "

				+ "where count($x//PARAM[@required='true' and string-length(normalize-space(.)) = 0]) > 0 " + "return $x/@name/string()";

		try {

			final List<String> list = serviceLocator.getService(ISLookUpService.class).quickSearchProfile(query);

			return Sets.newHashSet(list);

		} catch (Exception e) {

			log.error("Error executing xquery: " + query, e);

			return Sets.newHashSet();

		}

	}

	public Map<String, String> listRepoHiWorkflows() {

		final Map<String, String> map = Maps.newHashMap();

		final String query = "for $x in collection('/db/DRIVER/WorkflowDSResources/WorkflowDSResourceType') " + "where $x//WORKFLOW_TYPE='REPO_HI' "

				+ "return concat($x//RESOURCE_IDENTIFIER/@value, ' @@@ ', $x//WORKFLOW_NAME)";

		try {

			for (String s : serviceLocator.getService(ISLookUpService.class).quickSearchProfile(query)) {

				String[] arr = s.split("@@@");

				map.put(arr[0].trim(), arr[1].trim());

			}

		} catch (Exception e) {

			log.error("Error executing xquery: " + query, e);

		}

		return map;

	}

	public List<String> listWfFamilies() {

		final String query = "distinct-values(for $x in collection('/db/DRIVER/WorkflowDSResources/WorkflowDSResourceType') "

				+ "where string-length($x//DATASOURCE/@id) > 0 return $x//WORKFLOW_TYPE/@text())";

		try {

			return serviceLocator.getService(ISLookUpService.class).quickSearchProfile(query);

		} catch (ISLookUpException e) {

			log.error("Error executing xquery: " + query, e);

			return Lists.newArrayList();

		}

	}

	public String getDatasourceName(final String dsId) {

		final String query = "/*[.//RESOURCE_IDENTIFIER/@value='" + dsId + "']//OFFICIAL_NAME/text()";

		try {

			return serviceLocator.getService(ISLookUpService.class).getResourceProfileByQuery(query);

		} catch (ISLookUpException e) {

			log.error("Error executing xquery: " + query, e);

			return "UNKNOWN";

		}

	}

	public Map<String, List<WorkflowParamUI.UI>> getCategoryUis() {

		return categoryUis;

	}

	public String getCategoryUisJson() {

		return categoryUisJson;

	}

	@Required

	public void setCategoryUisJson(final String categoryUisJson) {

		this.categoryUisJson = categoryUisJson;

	}

	public List<Map<String, String>> obtainSubWorkflows(final String wfId) {

		final String query = "for $x in collection('/db/DRIVER/WorkflowDSResources/WorkflowDSResourceType')[.//RESOURCE_IDENTIFIER/@value='" + wfId

				+ "']//NODE[@type='LaunchWorkflow']\n"

				+ "for $y in collection('/db/DRIVER/WorkflowDSResources/WorkflowDSResourceType')[.//RESOURCE_IDENTIFIER/@value = $x//PARAM[@name='wfId']/@value]\n"

				+ "return <res>\n"

				+ "<id>{$x//PARAM[@name='wfId']/@value/string()}</id>\n"

				+ "<name>{$x/@name/string()}</name>\n"

				+ "<description>{$x/DESCRIPTION/text()}</description>\n"

				+ "<lastDate>{$y//LAST_EXECUTION_DATE/string()}</lastDate>\n"

				+ "<lastStatus>{$y//LAST_EXECUTION_STATUS/string()}</lastStatus>\n"

				+ "<disabled>{$y//CONFIGURATION/@start=\"DISABLED\" or $y//CONFIGURATION/@status !=\"EXECUTABLE\"}</disabled>\n"

				+ "</res>";

		try {

			final List<String> list = serviceLocator.getService(ISLookUpService.class).quickSearchProfile(query);

			return Lists.transform(list, new Function<String, Map<String, String>>() {

				private SAXReader reader = new SAXReader();

				@Override

				public Map<String, String> apply(final String s) {

					try {

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

						for (Object o : reader.read(new StringReader(s)).selectNodes("/res/*")) {

							map.put(((Node) o).getName(), ((Node) o).getText());

						}

						return map;

					} catch (DocumentException e) {

						throw new RuntimeException(e);

					}

				}

			});

		} catch (ISLookUpException e) {

			log.error("Error executing xquery: " + query, e);

			return Lists.newArrayList();

		}

	}

}