package org.kuali.student.r2.core.class1.state.dao;

 import java.util.List;

 import javax.persistence.Query;

 import org.kuali.student.enrollment.dao.GenericEntityDao;

 import org.kuali.student.r2.core.class1.state.model.StateEntity;

 import org.kuali.student.r2.core.class1.state.model.StateLifecycleRelationEntity;

 public class StateLifecycleRelationDao extends GenericEntityDao<StateLifecycleRelationEntity> {

     @SuppressWarnings("unchecked")

     public List<StateEntity> getInitialValidStates(String processKey) {

         Query query = em.createNamedQuery("StateProcess.getInitialValidStates");

         query.setParameter("processKey", processKey);

         List<StateEntity> initStates = query.getResultList();

         return initStates;

     }

     public StateEntity getNextHappyState(String processKey, String currentStateKey) {

         Query query = em.createNamedQuery("StateProcess.getNextHappyState");

         query.setParameter("processKey", processKey);

         query.setParameter("stateKey", currentStateKey);

         StateEntity nextState = (StateEntity) query.getSingleResult();

         return nextState;

     }

     @SuppressWarnings("unchecked")

     public List<StateLifecycleRelationEntity> getStateLifecycleRelationsByState(String stateKey) {

         Query query = em.createNamedQuery("StateProcess.getStateLifecycleRelationsByState");

         query.setParameter("stateKey", stateKey);

         List<StateLifecycleRelationEntity> relStates = query.getResultList();

         return relStates;

     }

     @SuppressWarnings("unchecked")

     public List<StateLifecycleRelationEntity> getStateLifecycleRelationsByLifecycle(String lifecycleKey) {

         Query query = em.createNamedQuery("StateProcess.getStateLifecycleRelationsByLifecycle");

         query.setParameter("lifecycleKey", lifecycleKey);

         List<StateLifecycleRelationEntity> relStates = query.getResultList();

         return relStates;

     }

 }