RoleFacade.java 2.46 KB
package fi.codecrew.moya.facade;

import java.util.ArrayList;
import java.util.List;

import javax.ejb.EJB;
import javax.ejb.LocalBean;
import javax.ejb.Stateless;
import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Root;

import fi.codecrew.moya.beans.EventBeanLocal;
import fi.codecrew.moya.model.EventUser;
import fi.codecrew.moya.model.LanEvent;
import fi.codecrew.moya.model.Role;
import fi.codecrew.moya.model.Role_;

@Stateless
@LocalBean
public class RoleFacade extends IntegerPkGenericFacade<Role> {

	@EJB
	private EventBeanLocal eventbean;

	public RoleFacade() {

		super(Role.class);
	}

	// public Role findByName(String name, LanEvent event) {
	// TypedQuery<Role> q = em.createNamedQuery("Role.findByRoleName",
	// Role.class);
	// q.setParameter("name", name);
	// q.setParameter("event", event);
	// return getSingleNullableResult(q);
	// }

	public List<Role> findForUser(EventUser user) {
		CriteriaBuilder cb = getEm().getCriteriaBuilder();
		CriteriaQuery<Role> cq = cb.createQuery(Role.class);
		Root<Role> root = cq.from(Role.class);
		cq.where(cb.equal(root.get(Role_.event), eventbean.getCurrentEvent()), cb.isMember(user, root.get(Role_.users)));

		return getEm().createQuery(cq).getResultList();
	}

	public Role createRole(LanEvent event, String rolename) {
		Role ret = new Role(event);
		ret.setName(rolename);
		create(ret);
		return ret;
	}

	public List<Role> findAll() {
		CriteriaBuilder cb = getEm().getCriteriaBuilder();
		CriteriaQuery<Role> cq = cb.createQuery(Role.class);
		Root<Role> root = cq.from(Role.class);
		cq.where(cb.equal(root.get(Role_.event), eventbean.getCurrentEvent()));

		return getEm().createQuery(cq).getResultList();
	}

	public List<Role> findUserSelectableRoles() {
		CriteriaBuilder cb = getEm().getCriteriaBuilder();
		CriteriaQuery<Role> cq = cb.createQuery(Role.class);
		Root<Role> root = cq.from(Role.class);
		cq.where(cb.equal(root.get(Role_.event), eventbean.getCurrentEvent()),
				cb.greaterThan(root.get(Role_.selectChildrensCount), new Integer(0)));
		
		List<Role> roles = getEm().createQuery(cq).getResultList();
		ArrayList<Role> returnRoles = new ArrayList<Role>();
		
		// sorry for uqly loop
		for(Role role : roles) {
			if(role.getSelectChildrensCount() > 0) { // just in case
				for(Role subRole : role.getChildren()) {
					if(!returnRoles.contains(subRole)) {
						returnRoles.add(subRole);
					}
				}
			}
		}
		
		return returnRoles;
	}

}