D-Net

package eu.dnetlib.data.mdstore.plugins;

import java.io.StringReader;

import java.net.URISyntaxException;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.Objects;

import java.util.regex.Matcher;

import java.util.regex.Pattern;

import java.util.stream.Collectors;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.dom4j.Document;

import org.dom4j.DocumentException;

import org.dom4j.Node;

import org.dom4j.io.SAXReader;

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

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

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

import eu.dnetlib.data.utils.HttpFetcher;

import eu.dnetlib.enabling.locators.UniqueServiceLocator;

import eu.dnetlib.rmi.enabling.ISLookUpService;

public class EnrichCollectionPlugin extends MdRecordPlugin {

	private Map<String, CnrCollection> colls = new HashMap<>();

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

	@Autowired

	private UniqueServiceLocator serviceLocator;

	@Override

	protected void reconfigure(final Map<String, String> params) {

		params.entrySet().forEach(e -> log.info(String.format("******************* %s -> %s", e.getKey(), e.getValue())));

		try {

			final String profile = serviceLocator.getService(ISLookUpService.class).getResourceProfile(params.get("dsId"));

			final Document doc = new SAXReader().read(new StringReader(profile));

			final String baseUrl = doc.valueOf("//INTERFACE[@id='" + params.get("dsInterface") + "']/BASE_URL").trim();

			colls.clear();

			colls.putAll(listOaiCollections(baseUrl + "?verb=ListSets").stream()

					.map(this::createCollection)

					.filter(Objects::nonNull)

					.distinct()

					.collect(Collectors.toMap(CnrCollection::getCode, o -> o)));

		} catch (final Exception e) {

			log.error("Error evaluating ListSets", e);

			throw new RuntimeException("Error evaluating ListSets", e);

		}

	}

	@Override

	protected boolean updateRecord(final String recordId, final MdRecord record) {

		for (final CnrCollection c : record.getInCollections()) {

			if (colls.containsKey(c.getCode())) {

				c.setName(colls.get(c.getCode()).getName());

				c.setAcronym(colls.get(c.getCode()).getAcronym());

			}

		}

		return true;

	}

	@SuppressWarnings("unchecked")

	private List<Node> listOaiCollections(final String listSetsUrl) {

		try {

			final SAXReader reader = new SAXReader();

			final String s = HttpFetcher.fetch(listSetsUrl, null);

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

			return doc.selectNodes("//*[local-name() = 'ListSets']/*[local-name() = 'set']");

		} catch (final DocumentException | URISyntaxException e) {

			log.error("Error listing sets from url: " + listSetsUrl, e);

			return new ArrayList<>();

		}

	}

	private CnrCollection createCollection(final Node n) {

		final CnrCollection c = new CnrCollection();

		final String code = n.valueOf("./*[local-name() = 'setSpec']").trim();

		c.setCode(n.valueOf("./*[local-name() = 'setSpec']").trim());

		if (code.equalsIgnoreCase("openaire")) {

			c.setAcronym("openaire");

			c.setName("Openaire Collection");

		} else {

			final String desc = n.valueOf("./*[local-name() = 'setDescription']").trim();

			final Matcher m = Pattern.compile("^Prodotti della ricerca di (.+) \\- (.+)$").matcher(desc);

			if (m.matches()) {

				c.setAcronym(m.group(1));

				c.setName(m.group(2));

			} else {

				c.setAcronym(desc);

				c.setName(desc);

			}

		}

		return c;

	}

	@Override

	protected void resetConfiguration() {

		colls.clear();

	}

}