D-Net

package eu.dnetlib.data.mdstore.plugins;

import java.net.URI;

import java.net.URISyntaxException;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import org.apache.commons.lang3.StringUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.dom4j.Document;

import org.dom4j.Element;

import org.dom4j.Namespace;

import org.dom4j.QName;

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

import com.google.gson.Gson;

import com.google.gson.reflect.TypeToken;

import eu.dnetlib.data.mdstore.plugins.objects.dli.DliIdentifier;

import eu.dnetlib.data.mdstore.plugins.objects.dli.DliRelation;

public class EnrichDatasetsPlugin extends GenericDoiMdstorePlugin {

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

	@Value("${plugin.enrich.dataset.dli.url}")

	private String baseUrl;

	@Override

	protected URI prepareURI(final String doi) throws URISyntaxException {

		return new URI(String.format(baseUrl, doi));

	}

	@Override

	protected boolean updateDocument(final Document doc, final String response) {

		final Gson gson = new Gson();

		final List<DliRelation> rels = gson.fromJson(response, new TypeToken<List<DliRelation>>() {}.getType());

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

		for (final DliRelation rel : rels) {

			final String title = rel.getTarget().getTitle();

			for (final DliIdentifier id : rel.getTarget().getIdentifiers()) {

				if (id.getSchema().equalsIgnoreCase("doi") && StringUtils.isNoneBlank(id.getIdentifier()) && StringUtils.isNotBlank(title)) {

					datasets.put(id.getIdentifier(), title);

				}

			}

		}

		if (datasets.isEmpty()) { return false; }

		final Element node = (Element) doc.selectSingleNode("//*[local-name() = 'datasets']");

		datasets.entrySet().forEach(e -> {

			final Element ds = node.addElement(new QName("dataset", new Namespace("isti", "http://www.isti.cnr.it/")));

			ds.addAttribute("doi", e.getKey());

			ds.addAttribute("url", "https://dx.doi.org/" + e.getKey());

			ds.setText(e.getValue());

			log.info("Found dataset: " + e.getKey() + " -> " + e.getValue());

		});

		return true;

	}

}