/*
    * Copyright 2008 The Kuali Foundation
    *
    * Licensed under the Educational Community License, Version 2.0 (the "License");
    * you may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    *
    * http://www.opensource.org/licenses/ecl2.php
    *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  package org.kuali.rice.kim.dao.impl;
  
  import java.util.ArrayList;
  import java.util.Collection;
  import java.util.List;
  
  import org.apache.ojb.broker.query.Criteria;
  import org.apache.ojb.broker.query.Query;
  import org.apache.ojb.broker.query.QueryFactory;
  import org.kuali.rice.kim.bo.role.KimResponsibility;
  import org.kuali.rice.kim.bo.role.impl.RoleResponsibilityActionImpl;
  import org.kuali.rice.kim.bo.role.impl.RoleResponsibilityImpl;
  import org.kuali.rice.kim.dao.KimResponsibilityDao;
  import org.kuali.rice.kim.util.KIMPropertyConstants;
  import org.kuali.rice.kim.util.KimConstants;
  import org.kuali.rice.kns.dao.impl.PlatformAwareDaoBaseOjb;
  import org.kuali.rice.kns.util.KNSPropertyConstants;
  
  /**
   * This is a description of what this class does - kellerj don't forget to fill this in. 
   * 
   * @author Kuali Rice Team (rice.collab@kuali.org)
   *
   */
  public class KimResponsibilityDaoOjb extends PlatformAwareDaoBaseOjb implements KimResponsibilityDao {
  
  	/**
  	 * @see org.kuali.rice.kim.dao.KimResponsibilityDao#getRoleIdsForResponsibilities(java.util.Collection)
  	 */
  	@SuppressWarnings("unchecked")
  	public List<String> getRoleIdsForResponsibilities(Collection<? extends KimResponsibility> responsibilities) {
  		List<String> responsibilityIds = new ArrayList<String>( responsibilities.size() );
  		for ( KimResponsibility kp : responsibilities ) {
  			responsibilityIds.add( kp.getResponsibilityId() );
  		}
  		Criteria c = new Criteria();
  		c.addIn( KimConstants.PrimaryKeyConstants.RESPONSIBILITY_ID, responsibilityIds );
  		c.addEqualTo( KNSPropertyConstants.ACTIVE, true );
  		
  		Query query = QueryFactory.newQuery( RoleResponsibilityImpl.class, c, true );
  		Collection<RoleResponsibilityImpl> coll = getPersistenceBrokerTemplate().getCollectionByQuery(query);
  		List<String> roleIds = new ArrayList<String>( coll.size() );
  		for ( RoleResponsibilityImpl rp : coll ) {
  			roleIds.add( rp.getRoleId() );
  		}
  		return roleIds;
  	}
  	
  	/**
  	 * @see org.kuali.rice.kim.dao.KimResponsibilityDao#getRoleIdsForResponsibility(org.kuali.rice.kim.bo.role.impl.KimResponsibilityImpl)
  	 */
  	@SuppressWarnings("unchecked")
  	public List<String> getRoleIdsForResponsibility( KimResponsibility responsibility) {
  		Criteria c = new Criteria();
  		c.addEqualTo( KimConstants.PrimaryKeyConstants.RESPONSIBILITY_ID, responsibility.getResponsibilityId() );
  		c.addEqualTo( KNSPropertyConstants.ACTIVE, true );
  		
  		Query query = QueryFactory.newQuery( RoleResponsibilityImpl.class, c, true );
  		Collection<RoleResponsibilityImpl> coll = getPersistenceBrokerTemplate().getCollectionByQuery(query);
  		List<String> roleIds = new ArrayList<String>( coll.size() );
  		for ( RoleResponsibilityImpl rp : coll ) {
  			roleIds.add( rp.getRoleId() );
  		}
  		return roleIds;
  	}
  
  	/**
  	 * @see org.kuali.rice.kim.dao.KimResponsibilityDao#getResponsibilityAction(String, String)
  	 */
  	@SuppressWarnings("unchecked")
  	public RoleResponsibilityActionImpl getResponsibilityAction(String roleId, String responsibilityId, String roleMemberId ) {
  		if ( roleMemberId == null || responsibilityId == null ) {
  			return null;
  		}
  		
  		Criteria c = new Criteria();
  		c.addEqualTo( "roleResponsibility.responsibilityId", responsibilityId );
  		c.addEqualTo( "roleResponsibility.roleId", roleId );
  		c.addEqualTo( "roleResponsibility.active", true );
  		Criteria idCriteria = new Criteria();
  		idCriteria.addEqualTo( KIMPropertyConstants.RoleMember.ROLE_MEMBER_ID, roleMemberId );
  		// also handle when roleMemberId is "*" in table
  		Criteria starCrit = new Criteria();
  		starCrit.addEqualTo( KIMPropertyConstants.RoleMember.ROLE_MEMBER_ID, "*" );
 		idCriteria.addOrCriteria( starCrit );
 		c.addAndCriteria( idCriteria );
 		
 		Query query = QueryFactory.newQuery( RoleResponsibilityActionImpl.class, c );
 		Collection<RoleResponsibilityActionImpl> coll = getPersistenceBrokerTemplate().getCollectionByQuery(query);
 		RoleResponsibilityActionImpl result = null;
 		if ( coll.size() == 0 ) {
 			// ok, an exact match failed, attempt to pull where the role_rsp_id is "*"
 			// but, at this point, we would require an exact match on the role member ID
 			c = new Criteria();
 			c.addEqualTo( "roleResponsibilityId", "*" );
 			c.addEqualTo( KIMPropertyConstants.RoleMember.ROLE_MEMBER_ID, roleMemberId );
 			query = QueryFactory.newQuery( RoleResponsibilityActionImpl.class, c );
 			coll = getPersistenceBrokerTemplate().getCollectionByQuery(query);
 			if ( coll.size() != 0 ) {
 				result = coll.iterator().next();
 			}
 		} else {
 			result = coll.iterator().next();
 		}
 		return result;
 	}
 }