D-Net

package eu.dnetlib.data.mdstore.plugins;

import java.net.URI;

import java.net.URISyntaxException;

import java.util.ArrayList;

import java.util.List;

import java.util.Map;

import org.apache.commons.lang3.StringUtils;

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

import eu.dnetlib.data.mdstore.plugins.objects.MdRecord;

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 Gson gson = new Gson();

	@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 void reconfigure(final Map<String, String> params) {}

	@Override

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

		doc.getDatasets().clear();

		final List<Dataset> datasets = new ArrayList<>();

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

		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.add(new Dataset(id.getIdentifier(), title));

				}

			}

		}

		if (datasets.isEmpty()) {

			return false;

		} else {

			doc.getDatasets().addAll(datasets);

			return true;

		}

	}

}