D-Net

package eu.dnetlib.msro.workflows.nodes.collect;

import java.util.Set;

import java.util.stream.Stream;

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

import org.springframework.stereotype.Component;

import com.google.common.collect.Sets;

import eu.dnetlib.msro.workers.aggregation.collect.plugins.CollectorPlugin;

import eu.dnetlib.msro.workers.aggregation.objects.InterfaceDescriptor;

@Component

public class MetadataCollector {

	@Autowired

	private CollectorPluginEnumerator collectorPluginEnumerator;

	public Stream<String> collect(final InterfaceDescriptor ifDescriptor) throws CollectException {

		return dateRangeCollect(ifDescriptor, null, null);

	}

	public Stream<String> dateRangeCollect(

			final InterfaceDescriptor ifDescriptor, final String from, final String until)

			throws CollectException {

		final CollectorPlugin plugin = collectorPluginEnumerator.get(ifDescriptor.getProtocol());

		if (!verifyParams(ifDescriptor.getParams().keySet(), Sets.newHashSet(plugin.listNameParameters()))) { throw new CollectException(

				"Invalid parameters, valid: " + plugin.listNameParameters() + ", current: " + ifDescriptor.getParams().keySet()); }

		return plugin.collect(ifDescriptor, from, until);

	}

	private boolean verifyParams(final Set<String> curr, final Set<String> valid) {

		return valid.containsAll(curr);

	}

}