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 java.util.stream.Collectors;

import org.apache.commons.lang3.StringUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

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

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

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

public class EnrichProjectsPlugin extends MdRecordPlugin {

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

	@Value("${plugin.enrich.projects.openaire.url}")

	private String baseUrl;

	private Map<String, Project> mapProjects = new HashMap<>();

	@Override

	protected void reconfigure() {

		log.info("Cleaning projects cache");

		mapProjects.clear();

	}

	@Override

	protected boolean updateRecord(final MdRecord record) {

		log.debug("Found " + record.getProjects().size() + " projects");

		if (record.getProjects().isEmpty()) { return false; }

		final List<Project> list = record.getProjects()

				.stream()

				.map(Project::getInfoId)

				.filter(StringUtils::isNotBlank)

				.map(Project::newInstance)

				.map(this::findEnrichedProject)

				.collect(Collectors.toList());

		log.debug("Processed " + list.size() + " projects");

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

		record.getProjects().clear();

		record.getProjects().addAll(list);

		return true;

	}

	private Project findEnrichedProject(final Project simpleProject) {

		final String s = simpleProject.getFunder() + "/" + simpleProject.getProgram() + "/" + simpleProject.getCode();

		if (!mapProjects.containsKey(s)) {

			final Project project = resolveProject(s);

			mapProjects.put(s, project != null ? project : simpleProject);

		}

		return mapProjects.get(s);

	}

	private Project resolveProject(final String s) {

		try {

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

			final String url = String.format(baseUrl, arr[0], arr[1], arr[2]);

			return Project.newInstance(new URI(url));

		} catch (final URISyntaxException e) {

			log.error("Error resolving project: " + s, e);

			throw new RuntimeException("Error resolving project: " + s, e);

		}

	}

}