D-Net

package eu.dnetlib.msro.workers.aggregation.collect.functions;

import java.io.StringReader;

import java.util.Iterator;

import java.util.List;

import java.util.Map;

import org.dom4j.Document;

import org.dom4j.DocumentException;

import org.dom4j.io.SAXReader;

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

import org.springframework.stereotype.Component;

import com.google.common.collect.Iterators;

import com.google.common.collect.Lists;

import eu.dnetlib.msro.workers.aggregation.collect.plugins.oai.engine.HttpConnector;

import eu.dnetlib.msro.workers.aggregation.collect.plugins.oaisets.OaiSetsIterator;

import eu.dnetlib.rmi.data.ProtocolParameterValue;

import eu.dnetlib.rmi.data.functions.ParamValuesFunction;

@Component

public class ListOaiSetsFunction implements ParamValuesFunction {

	@Autowired

	private HttpConnector httpConnector;

	@Override

	public List<ProtocolParameterValue> findValues(final String baseUrl, final Map<String, String> params) {

		final SAXReader reader = new SAXReader();

		final Iterator<ProtocolParameterValue> iter = Iterators.transform(new OaiSetsIterator(baseUrl, httpConnector),

				s -> {

					try {

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

						final String id = doc.valueOf("//*[local-name()='setSpec']");

						final String name = doc.valueOf("//*[local-name()='setName']");

						return new ProtocolParameterValue(id, name);

					} catch (final DocumentException e) {

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

					}

				});

		return Lists.newArrayList(iter);

	}

}