D-Net

package eu.dnetlib.administration.uis.modules.datasources;

import java.io.StringReader;

import java.util.ArrayList;

import java.util.List;

import java.util.Objects;

import java.util.Set;

import java.util.stream.Collectors;

import javax.servlet.http.HttpServletRequest;

import org.apache.commons.io.IOUtils;

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 org.springframework.beans.factory.annotation.Autowired;

import org.springframework.ui.ModelMap;

import org.springframework.web.bind.annotation.PathVariable;

import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RequestParam;

import org.springframework.web.bind.annotation.RestController;

import com.google.common.base.Splitter;

import com.google.common.collect.Sets;

import com.google.gson.Gson;

import eu.dnetlib.administration.uis.annotations.Authorization;

import eu.dnetlib.administration.uis.annotations.MenuEntry;

import eu.dnetlib.administration.uis.annotations.MenuGroup;

import eu.dnetlib.administration.uis.annotations.PermissionLevel;

import eu.dnetlib.clients.dsManager.BrowseTerm;

import eu.dnetlib.clients.dsManager.DatasourceDesc;

import eu.dnetlib.clients.dsManager.DsManagerClient;

import eu.dnetlib.clients.dsManager.SearchInterfacesEntry;

import eu.dnetlib.clients.is.InformationServiceClient;

import eu.dnetlib.clients.locators.ServiceClientFactory;

import eu.dnetlib.clients.msro.MsroClient;

import eu.dnetlib.exceptions.DnetGenericException;

import eu.dnetlib.miscutils.collections.Pair;

import eu.dnetlib.msro.workflows.WorkflowRef;

@RestController

@RequestMapping("/ajax/apis")

@MenuEntry(value = "Overview", order = 1, urlSection = "apis", group = MenuGroup.DS_MANAGEMENT)

@Authorization(PermissionLevel.DS_ADMIN)

public class ApisModule extends AbstractDsModule {

	@Autowired

	private InformationServiceClient isClient;

	@Autowired

	private DsManagerClient dsManager;

	@Autowired

	private ServiceClientFactory clientFactory;

	@Autowired

	private DatasourceVocabularies vocs;

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

	@Override

	public void populateModelMap(final ModelMap map, final HttpServletRequest request) throws Exception {

		final Gson gson = new Gson();

		map.addAttribute("compatibilityLevels", fetchVocabularyTermsAsJson(vocs.getCompatibilityLevels()));

		map.addAttribute("browseFields", gson.toJson(dsManager.listBrowsableFields()));

	}

	@RequestMapping(value = "ds")

	public DatasourceDesc getDS(@RequestParam final String id) throws Exception {

		return dsManager.getDatasource(id);

	}

	@RequestMapping(value = "browse/{field}")

	public List<BrowseTerm> browseField(@PathVariable final String field) throws Exception {

		return dsManager.browseField(field);

	}

	@RequestMapping(value = "search")

	public List<SearchInterfacesEntry> listApis(@RequestParam final String param,

			@RequestParam final String value,

			@RequestParam(required = false, defaultValue = "false") final boolean refresh) throws Exception {

		if (refresh) {

			dsManager.emptyCache();

		}

		return dsManager.searchInterface(param, value);

	}

	@RequestMapping(value = "repohi/wfs")

	public List<RepoHIWorkflow> listRepoHiWfs(@RequestParam final String compliance, @RequestParam final String type) throws Exception {

		try {

			final String query =

					IOUtils.toString(getClass().getResourceAsStream("/xquery/find_repohi.xquery"));

			final SAXReader reader = new SAXReader();

			return isClient.find(query)

					.stream().map(s -> {

						try {

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

							final Set<String> ifcTypes = Sets.newHashSet(Splitter.on(",").omitEmptyStrings().trimResults().split(doc.valueOf("//TYPES")));

							final Set<String> compliances =

									Sets.newHashSet(Splitter.on(",").omitEmptyStrings().trimResults().split(doc.valueOf("//COMPLIANCES")));

							final String id = doc.valueOf("//ID");

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

							final String desc = doc.valueOf("//DESC");

							final List<Pair<String, String>> fields = new ArrayList<Pair<String, String>>();

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

								final Node currentNode = (Node) o;

								final String key = currentNode.valueOf("@name");

								final String value = currentNode.getText();

								fields.add(new Pair<String, String>(key, value));

							}

							return new RepoHIWorkflow(id, name, desc, ifcTypes, compliances, fields);

						} catch (final Exception ex) {

							log.warn("Error parsing xml: " + s);

							return null;

						}

					})

					.filter(Objects::nonNull)

					.filter(wf -> wf.isValidFor(type, compliance))

					.sorted()

					.collect(Collectors.toList());

		} catch (final Exception e) {

			log.error("Error finding repo-hi wfs", e);

			throw new DnetGenericException("Error finding repo-hi wfs", e);

		}

	}

	@RequestMapping(value = "repohi/start")

	public WorkflowRef startRepoHiWf(@RequestParam final String dsId,

			@RequestParam final String iface,

			@RequestParam final String wfId,

			@RequestParam final String workerId) throws Exception {

		return clientFactory.getClient(MsroClient.class).startRepoHiWorkflow(wfId, workerId, dsId, iface);

	}

	@RequestMapping(value = "repobye")

	public WorkflowRef startRepoHiWf(@RequestParam final String wfId) throws Exception {

		return clientFactory.getClient(MsroClient.class).startRepoByeForWorkflow(wfId);

	}

}