information-system-model/src/main/java/org/gcube/informationsystem/discovery/Discovery.java

/**
 * 
 */
package org.gcube.informationsystem.discovery;

import java.lang.reflect.Method;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.gcube.informationsystem.base.reference.Element;
import org.gcube.informationsystem.base.reference.properties.PropertyElement;
import org.gcube.informationsystem.types.annotations.ISProperty;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * @author Luca Frosini (ISTI - CNR)
 */
public class Discovery<E extends Element> {
	
	public static Logger logger = LoggerFactory.getLogger(Discovery.class);
	
	protected final Class<E> root;
	protected final List<Package> packages;
	protected final List<Class<? extends E>> discovered;
	protected final boolean includeRootPackage;
	
	public List<Class<? extends E>> getDiscovered() {
		return discovered;
	}
	
	public Discovery(Class<E> root) {
		this(root, true);
	}
	
	public Discovery(Class<E> root, boolean includeRootPackage) {
		this.root = root;
		this.packages = new ArrayList<>();
		this.includeRootPackage = includeRootPackage;
		if(includeRootPackage) {
			addPackage(root.getPackage());
		}
		this.discovered = new ArrayList<>();
		add(root);
	}
	
	public void addPackage(Package p) {
		if(!includeRootPackage && p==root.getPackage()) {
			return;
		}
		packages.add(p);
	}
	
	protected void add(Class<? extends E> clz) {
		if(!discovered.contains(clz)) {
			discovered.add(clz);
			logger.info("+ Added {}.", clz);
		}
	}
	
	protected void analizeISM(Class<? extends E> clz) {
		logger.trace("Analizyng {}", clz);
		
		if(!clz.isInterface()) {
			logger.trace("- Discarding {} that is not an interface", clz);
			return;
		}
		
		if(!root.isAssignableFrom(clz)) {
			logger.trace("- Discarding {} because is not a {}", clz, root.getSimpleName());
			return;
		}
		
		if(discovered.contains(clz)) {
			logger.trace("- Discarding {} because was already managed", clz);
			return;
		}
		
		Class<?>[] interfaces = clz.getInterfaces();
		
		for(Class<?> interfaceClass : interfaces) {
			@SuppressWarnings("unchecked")
			Class<E> parent = (Class<E>) interfaceClass;
			analizeISM(parent);
		}
		
		if(root == PropertyElement.class) {
			
			for(Method m : clz.getDeclaredMethods()) {
				m.setAccessible(true);
				if(m.isAnnotationPresent(ISProperty.class)) {
					if(Map.class.isAssignableFrom(m.getReturnType()) || Set.class.isAssignableFrom(m.getReturnType())
							|| List.class.isAssignableFrom(m.getReturnType())) {
						
						Type[] typeArguments = ((ParameterizedType) m.getGenericReturnType()).getActualTypeArguments();
						for(Type t : typeArguments) {
							@SuppressWarnings("unchecked")
							Class<? extends PropertyElement> tClass = (Class<? extends PropertyElement>) t;
							if(root.isAssignableFrom(tClass)) {
								@SuppressWarnings("unchecked")
								Class<E> type = (Class<E>) tClass;
								analizeISM(type);
							}
						}
						
					} else if(root.isAssignableFrom(m.getReturnType())) {
						@SuppressWarnings("unchecked")
						Class<E> type = (Class<E>) m.getReturnType();
						analizeISM(type);
					}
				}
				
			}
		}
		add(clz);
	}
	
	public void discover() throws Exception {
		for(Package p : packages) {
			List<Class<?>> classes = ReflectionUtility.getClassesForPackage(p);
			for(Class<?> clz : classes) {
				@SuppressWarnings("unchecked")
				Class<E> ism = (Class<E>) clz;
				analizeISM(ism);
			}
		}
		
	}
	
}