D-Net

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

import java.util.List;

import java.util.stream.Collectors;

import org.springframework.beans.BeansException;

import org.springframework.context.ApplicationContext;

import org.springframework.context.ApplicationContextAware;

import org.springframework.stereotype.Component;

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

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

@Component

public class CollectorPluginEnumerator implements ApplicationContextAware {

	private ApplicationContext applicationContext;

	public List<CollectorPlugin> getAll() {

		return applicationContext.getBeansOfType(CollectorPlugin.class)

				.values()

				.stream()

				.filter(o -> o.getClass().isAnnotationPresent(DnetCollectorPlugin.class))

				.collect(Collectors.toList());

	}

	public CollectorPlugin get(final String protocol) throws CollectException {

		for (final CollectorPlugin cp : getAll()) {

			if (protocol.equalsIgnoreCase(cp.getProtocol())) { return cp; }

		}

		throw new CollectException("plugin not found for protocol: " + protocol);

	}

	@Override

	public void setApplicationContext(final ApplicationContext applicationContext) throws BeansException {

		this.applicationContext = applicationContext;

	}

}