br.com.caelum.vraptor.ioc.cdi.ScopesUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of vraptor-cdi-provider Show documentation
VRaptor CDI Provider
The newest version!
package br.com.caelum.vraptor.ioc.cdi;

import java.lang.annotation.Annotation;
import java.util.Arrays;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import javax.enterprise.context.Dependent;
import javax.enterprise.context.NormalScope;
import javax.enterprise.context.RequestScoped;

import br.com.caelum.vraptor.ioc.ApplicationScoped;
import br.com.caelum.vraptor.ioc.PrototypeScoped;
import br.com.caelum.vraptor.ioc.SessionScoped;

/**
 * This class should be used for bean registration at startup time
 * 
 * @author Alberto Souza
 * 
 */
// TODO create unit tests
@SuppressWarnings({ "unchecked", "rawtypes" })
public class ScopesUtil {

	private final List cdiScopes = Arrays.asList(
			javax.enterprise.context.ApplicationScoped.class,
			javax.enterprise.context.SessionScoped.class, Dependent.class,
			RequestScoped.class);

	private final List vraptorScopes = Arrays.asList(ApplicationScoped.class,
			SessionScoped.class, br.com.caelum.vraptor.ioc.RequestScoped.class,
			PrototypeScoped.class);

	private final Map, Class> cdiToVraptorScopes = 
			new HashMap, Class>();
	
	{
		cdiToVraptorScopes.put(javax.enterprise.context.ApplicationScoped.class,ApplicationScoped.class);
		cdiToVraptorScopes.put(javax.enterprise.context.SessionScoped.class,SessionScoped.class);
		cdiToVraptorScopes.put(RequestScoped.class,br.com.caelum.vraptor.ioc.RequestScoped.class);
		cdiToVraptorScopes.put(javax.enterprise.context.Dependent.class,PrototypeScoped.class);
	}

	private HashSet> findAnnotations(final Class componentType,
			final List scopesAnnotation) {
		Annotation[] annotations = componentType.getAnnotations();
		HashSet> result = new HashSet>();
		for (Annotation componentAnnotation : annotations) {
			for (Class scopeAnnotation : (List) scopesAnnotation) {
				if (componentAnnotation.annotationType().equals(scopeAnnotation)) {
					result.add(componentAnnotation.annotationType());
				}
			}
		}
		return result;
	}

	private HashSet> cdiScopes(final Class componentType) {
		return findAnnotations(componentType, cdiScopes);
	}

	public ScopeInfo isScoped(Class clazz) {
		ScopeInfo scopedInfo = new ScopeInfo();
		Iterator iterator = cdiScopes(clazz).iterator();
		if(iterator.hasNext()){
			Class cdiScope = (Class) iterator.next();
			scopedInfo.setScope(cdiToVraptorScopes.get(cdiScope));
			return scopedInfo;
		}
		iterator = vraptorScopes(clazz).iterator();
		if(iterator.hasNext()){
			Class vraptorScope = (Class) iterator.next();
			scopedInfo.setScope(vraptorScope);			
		}
		return scopedInfo;
	}

	private HashSet> vraptorScopes(final Class componentType) {
		return findAnnotations(componentType, vraptorScopes);
	}

}